From 70e681b079a51863358a185982499f34de0ac929 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@digitalocean.com>
Date: Fri, 15 Feb 2019 15:53:04 -0500
Subject: [PATCH] Added view tests for tenancy

---
 netbox/tenancy/tests/test_views.py | 58 ++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 netbox/tenancy/tests/test_views.py

diff --git a/netbox/tenancy/tests/test_views.py b/netbox/tenancy/tests/test_views.py
new file mode 100644
index 000000000..a6ca8f358
--- /dev/null
+++ b/netbox/tenancy/tests/test_views.py
@@ -0,0 +1,58 @@
+import urllib.parse
+
+from django.test import Client, TestCase
+from django.urls import reverse
+
+from tenancy.models import Tenant, TenantGroup
+
+
+class TenantGroupTestCase(TestCase):
+
+    def setUp(self):
+
+        self.client = Client()
+
+        TenantGroup.objects.bulk_create([
+            TenantGroup(name='Tenant Group 1', slug='tenant-group-1'),
+            TenantGroup(name='Tenant Group 2', slug='tenant-group-2'),
+            TenantGroup(name='Tenant Group 3', slug='tenant-group-3'),
+        ])
+
+    def test_tenantgroup_list(self):
+
+        url = reverse('tenancy:tenantgroup_list')
+
+        response = self.client.get(url, follow=True)
+        self.assertEqual(response.status_code, 200)
+
+
+class TenantTestCase(TestCase):
+
+    def setUp(self):
+
+        self.client = Client()
+
+        tenantgroup = TenantGroup(name='Tenant Group 1', slug='tenant-group-1')
+        tenantgroup.save()
+
+        Tenant.objects.bulk_create([
+            Tenant(name='Tenant 1', slug='tenant-1', group=tenantgroup),
+            Tenant(name='Tenant 2', slug='tenant-2', group=tenantgroup),
+            Tenant(name='Tenant 3', slug='tenant-3', group=tenantgroup),
+        ])
+
+    def test_tenant_list(self):
+
+        url = reverse('tenancy:tenant_list')
+        params = {
+            "group": TenantGroup.objects.first().slug,
+        }
+
+        response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)), follow=True)
+        self.assertEqual(response.status_code, 200)
+
+    def test_tenant(self):
+
+        tenant = Tenant.objects.first()
+        response = self.client.get(tenant.get_absolute_url(), follow=True)
+        self.assertEqual(response.status_code, 200)