mirror of
				https://github.com/inventree/InvenTree.git
				synced 2025-10-31 13:15:43 +00:00 
			
		
		
		
	refactor test setup
This commit is contained in:
		| @@ -1,12 +1,11 @@ | ||||
| """ Unit tests for Company 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 | ||||
|  | ||||
|  | ||||
| class CompanyViewTestBase(TestCase): | ||||
| class CompanyViewTestBase(InvenTreeTestCate): | ||||
|  | ||||
|     fixtures = [ | ||||
|         'category', | ||||
| @@ -17,33 +16,9 @@ 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') | ||||
|  | ||||
|     rules = [ | ||||
|         'all', | ||||
|     ] | ||||
|  | ||||
| class CompanyViewTest(CompanyViewTestBase): | ||||
|     """ | ||||
|   | ||||
		Reference in New Issue
	
	Block a user