NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-25 18:08:38 -06:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Add tenancy CSV import tests

Browse Source
This commit is contained in:
Jeremy Stretch 2019-12-12 11:29:41 -05:00
parent 821cd58825
commit 7550d53d02
1 changed files with 40 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
netbox/tenancy/tests/test_views.py
View File

@ -10,7 +10,12 @@ from utilities.testing import create_test_user
class TenantGroupTestCase(TestCase): class TenantGroupTestCase(TestCase):
def setUp(self): def setUp(self):
user = create_test_user(permissions=['tenancy.view_tenantgroup']) user = create_test_user(
permissions=[
'tenancy.view_tenantgroup',
'tenancy.add_tenantgroup',
]
)
self.client = Client() self.client = Client()
self.client.force_login(user) self.client.force_login(user)
@ -27,11 +32,30 @@ class TenantGroupTestCase(TestCase):
response = self.client.get(url, follow=True) response = self.client.get(url, follow=True)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
def test_tenantgroup_import(self):
csv_data = (
"name,slug",
"Tenant Group 4,tenant-group-4",
"Tenant Group 5,tenant-group-5",
"Tenant Group 6,tenant-group-6",
)
response = self.client.post(reverse('tenancy:tenantgroup_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertEqual(TenantGroup.objects.count(), 6)
class TenantTestCase(TestCase): class TenantTestCase(TestCase):
def setUp(self): def setUp(self):
user = create_test_user(permissions=['tenancy.view_tenant']) user = create_test_user(
permissions=[
'tenancy.view_tenant',
'tenancy.add_tenant',
]
)
self.client = Client() self.client = Client()
self.client.force_login(user) self.client.force_login(user)
@ -59,3 +83,17 @@ class TenantTestCase(TestCase):
tenant = Tenant.objects.first() tenant = Tenant.objects.first()
response = self.client.get(tenant.get_absolute_url(), follow=True) response = self.client.get(tenant.get_absolute_url(), follow=True)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
def test_tenant_import(self):
csv_data = (
"name,slug",
"Tenant 4,tenant-4",
"Tenant 5,tenant-5",
"Tenant 6,tenant-6",
)
response = self.client.post(reverse('tenancy:tenant_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertEqual(Tenant.objects.count(), 6)