add more tests to test_customer_device_table

network_inventory/inventory/tests/test_views/test_customer_device_table_view.py

@@ -1,6 +1,10 @@
 import pytest
 
 from django.test import Client
+from mixer.backend.django import mixer
+
+import helper
+from inventory.models import Customer
 
 pytestmark = pytest.mark.django_db
 
@@ -8,3 +12,35 @@ pytestmark = pytest.mark.django_db
 def test_customer_device_table_not_logged_in():
     response = Client().get('/customer/1/devices/')
     assert response.status_code == 302 and 'login' in response.url
+
+
+def test_customer_device_table(create_admin_user):
+    fixture = create_admin_user()
+    customer = fixture['customer']
+    client = Client()
+    client.login(username="novartis-admin", password="password")
+    device = mixer.blend('inventory.Device', customer=mixer.SELECT)
+    response = client.get('/customer/' + str(customer.id) + '/devices/')
+    assert (response.status_code == 200
+            and helper.in_content(response, device.name))
+
+
+def test_customer_device_table_no_device(create_admin_user):
+    fixture = create_admin_user()
+    customer = fixture['customer']
+    client = Client()
+    client.login(username="novartis-admin", password="password")
+    response = client.get('/customer/' + str(customer.id) + '/devices/')
+    assert (response.status_code == 200
+            and helper.not_in_content(response, "Novartis PC"))
+
+
+def test_customer_device_table_no_permission(create_admin_user):
+    create_admin_user()
+    customer = Customer.objects.create(name='Nestle')
+    client = Client()
+    client.login(username="novartis-admin", password="password")
+    mixer.blend('inventory.Device', customer=customer)
+    response = client.get('/customer/' + str(customer.id) + '/devices/')
+    assert response.status_code == 403
+