Merge branch 'master' of https://github.com/inventree/InvenTree into style-fixes

InvenTree/company/test_views.py

@@ -1,12 +1,11 @@
 """ Unit tests for Company views (see views.py) """
 
-from django.contrib.auth import get_user_model
-from django.contrib.auth.models import Group
-from django.test import TestCase
 from django.urls import reverse
 
+from InvenTree.helpers import InvenTreeTestCase
 
-class CompanyViewTestBase(TestCase):
+
+class CompanyViewTestBase(InvenTreeTestCase):
 
     fixtures = [
         'category',
@@ -17,32 +16,7 @@ class CompanyViewTestBase(TestCase):
         'supplier_part',
     ]
 
-    def setUp(self):
-        super().setUp()
-
-        # Create a user
-        user = get_user_model()
-
-        self.user = user.objects.create_user(
-            username='username',
-            email='user@email.com',
-            password='password'
-        )
-
-        # Put the user into a group with the correct permissions
-        group = Group.objects.create(name='mygroup')
-        self.user.groups.add(group)
-
-        # Give the group *all* the permissions!
-        for rule in group.rule_sets.all():
-            rule.can_view = True
-            rule.can_change = True
-            rule.can_add = True
-            rule.can_delete = True
-
-            rule.save()
-
-        self.client.login(username='username', password='password')
+    roles = 'all'
 
 
 class CompanyViewTest(CompanyViewTestBase):