NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2026-01-25 12:59:59 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Refactored the tests to remove a lot of boilerplate

Browse Source
This commit is contained in:
Jeremy Stretch
2018-08-03 11:39:26 -04:00
parent 09f6b76cc9
commit d46cbcff5f
10 changed files with 153 additions and 279 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
netbox/tenancy/tests/test_api.py
View File

@@ -1,22 +1,17 @@
from __future__ import unicode_literals
from django.contrib.auth.models import User
from django.urls import reverse
from rest_framework import status
from rest_framework.test import APITestCase
from tenancy.models import Tenant, TenantGroup
from users.models import Token
from utilities.testing import HttpStatusMixin
from utilities.testing import APITestCase
class TenantGroupTest(HttpStatusMixin, APITestCase):
class TenantGroupTest(APITestCase):
def setUp(self):
user = User.objects.create(username='testuser', is_superuser=True)
token = Token.objects.create(user=user)
self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)}
super(TenantGroupTest, self).setUp()
self.tenantgroup1 = TenantGroup.objects.create(name='Test Tenant Group 1', slug='test-tenant-group-1')
self.tenantgroup2 = TenantGroup.objects.create(name='Test Tenant Group 2', slug='test-tenant-group-2')
@@ -103,13 +98,11 @@ class TenantGroupTest(HttpStatusMixin, APITestCase):
self.assertEqual(TenantGroup.objects.count(), 2)
class TenantTest(HttpStatusMixin, APITestCase):
class TenantTest(APITestCase):
def setUp(self):
user = User.objects.create(username='testuser', is_superuser=True)
token = Token.objects.create(user=user)
self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)}
super(TenantTest, self).setUp()
self.tenantgroup1 = TenantGroup.objects.create(name='Test Tenant Group 1', slug='test-tenant-group-1')
self.tenantgroup2 = TenantGroup.objects.create(name='Test Tenant Group 2', slug='test-tenant-group-2')