mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
|
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)
|