122 lines
4.9 KiB
Python
122 lines
4.9 KiB
Python
import pytest
|
|
|
|
from django.test import Client
|
|
from django.contrib.auth import get_user_model
|
|
from guardian.shortcuts import assign_perm
|
|
|
|
from mixer.backend.django import mixer
|
|
|
|
from core.tests import helper
|
|
|
|
pytestmark = pytest.mark.django_db
|
|
|
|
|
|
def test_customer_list_view_not_logged_in():
|
|
response = Client().get('/')
|
|
assert response.status_code == 302 and 'login' in response.url
|
|
|
|
|
|
def test_customer_list_view_no_customer():
|
|
User = get_user_model()
|
|
User.objects.create_user("pharma-admin", "admin@pharma.com",
|
|
"password", is_staff=True)
|
|
client = Client()
|
|
client.login(username="pharma-admin", password="password")
|
|
response = client.get('/')
|
|
assert response.status_code == 200
|
|
|
|
|
|
def test_customer_list_view(create_admin_user):
|
|
fixture = create_admin_user()
|
|
customer = fixture['customer']
|
|
client = Client()
|
|
client.login(username="pharma-admin", password="password")
|
|
response = client.get('/')
|
|
assert (response.status_code == 200
|
|
and helper.in_content(response, customer)
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer.id)
|
|
+ "/nets/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer.id)
|
|
+ "/computers/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer.id)
|
|
+ "/devices/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer.id)
|
|
+ "/backups/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer.id)
|
|
+ "/licenses/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer.id)
|
|
+ "/users/"))
|
|
|
|
|
|
def test_customer_list_view_multiple_customers(create_admin_user):
|
|
fixture = create_admin_user()
|
|
customer1 = fixture['customer']
|
|
customer2 = mixer.blend('customers.Customer')
|
|
assign_perm('view_customer', fixture['admin'], customer2)
|
|
client = Client()
|
|
client.login(username="pharma-admin", password="password")
|
|
response = client.get('/')
|
|
assert (response.status_code == 200
|
|
and helper.in_content(response, customer1)
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer1.id)
|
|
+ "/nets/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer1.id)
|
|
+ "/computers/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer1.id)
|
|
+ "/devices/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer1.id)
|
|
+ "/backups/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer1.id)
|
|
+ "/licenses/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer1.id)
|
|
+ "/users/")
|
|
and helper.in_content(response, customer2)
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer2.id)
|
|
+ "/nets/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer2.id)
|
|
+ "/computers/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer2.id)
|
|
+ "/devices/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer2.id)
|
|
+ "/backups/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer2.id)
|
|
+ "/licenses/")
|
|
and helper.in_content(response,
|
|
"/customer/"
|
|
+ str(customer1.id)
|
|
+ "/users/"))
|