finish the test for the customer list

This commit is contained in:
Andreas Zweili 2019-07-13 16:15:01 +02:00
parent 68e3bbeb4b
commit 66706c854e
1 changed files with 13 additions and 12 deletions

View File

@ -10,13 +10,13 @@ from inventory.models import Customer, Computer
def create_users(): def create_users():
User = get_user_model() User = get_user_model()
novartis = User.objects.create_user('novartis_admin', novartis = User.objects.create_user('novartis_admin',
'novartis_admin@novartis.com', 'novartis_admin@novartis.com',
'password', 'password',
is_staff=True) is_staff=True)
nestle = User.objects.create_user('nestle_admin', nestle = User.objects.create_user('nestle_admin',
'nestle_admin@nestle.com', 'nestle_admin@nestle.com',
'password', 'password',
is_staff=True) is_staff=True)
return novartis, nestle return novartis, nestle
@ -40,7 +40,7 @@ def create_computers():
@pytest.mark.django_db @pytest.mark.django_db
def test_something(): def test_customer_permissions():
novartis_admin_group, nestle_admin_group = create_groups() novartis_admin_group, nestle_admin_group = create_groups()
novartis_admin, nestle_admin = create_users() novartis_admin, nestle_admin = create_users()
novartis, nestle = create_customers() novartis, nestle = create_customers()
@ -52,9 +52,10 @@ def test_something():
novartis_client = Client() novartis_client = Client()
nestle_client = Client() nestle_client = Client()
response = novartis_client.post('/admin/', response = novartis_client.post('/admin/', follow=True)
{'username': 'novartis_admin', loginresponse = novartis_client.login(username='novartis_admin',
'password': 'password'}) password='password')
response = novartis_client.get('/') response = novartis_client.get('/')
print(response.content) print(response.content.decode('utf8'))
assert False assert ('Novartis' in response.content.decode('utf8') and
'Nestle' not in response.content.decode('utf8'))