refactor test setup

InvenTree/stock/test_views.py

@@ -1,14 +1,13 @@
 """ Unit tests for Stock views (see views.py) """
 
-from django.test import TestCase
 from django.urls import reverse
-from django.contrib.auth import get_user_model
-from django.contrib.auth.models import Group
+
+from InvenTree.InvenTree.helpers import InvenTreeTestCate
 
 # from common.models import InvenTreeSetting
 
 
-class StockViewTestCase(TestCase):
+class StockViewTestCase(InvenTreeTestCate):
 
     fixtures = [
         'category',
@@ -19,36 +18,7 @@ class StockViewTestCase(TestCase):
         'stock',
     ]
 
-    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'
-        )
-
-        self.user.is_staff = True
-        self.user.save()
-
-        # 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 StockListTest(StockViewTestCase):
     """ Tests for Stock list views """

InvenTree/stock/tests.py

@@ -1,9 +1,8 @@
-from django.test import TestCase
 from django.db.models import Sum
-from django.contrib.auth import get_user_model
 from django.core.exceptions import ValidationError
 
 import datetime
+from InvenTree.InvenTree.helpers import InvenTreeTestCate
 
 from InvenTree.status_codes import StockHistoryCode
 
@@ -14,7 +13,7 @@ from part.models import Part
 from build.models import Build
 
 
-class StockTest(TestCase):
+class StockTest(InvenTreeTestCate):
     """
     Tests to ensure that the stock location tree functions correcly
     """
@@ -29,6 +28,8 @@ class StockTest(TestCase):
     ]
 
     def setUp(self):
+        super().setUp()
+
         # Extract some shortcuts from the fixtures
         self.home = StockLocation.objects.get(name='Home')
         self.bathroom = StockLocation.objects.get(name='Bathroom')
@@ -39,14 +40,6 @@ class StockTest(TestCase):
         self.drawer2 = StockLocation.objects.get(name='Drawer_2')
         self.drawer3 = StockLocation.objects.get(name='Drawer_3')
 
-        # Create a user
-        user = get_user_model()
-        user.objects.create_user('username', 'user@email.com', 'password')
-
-        self.client.login(username='username', password='password')
-
-        self.user = user.objects.get(username='username')
-
         # Ensure the MPTT objects are correctly rebuild
         Part.objects.rebuild()
         StockItem.objects.rebuild()