mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code 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 1bdfcd1dbe
commit bd5e860be0
10 changed files with 153 additions and 279 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
netbox/extras/tests/test_customfields.py
View File

@@ -2,18 +2,15 @@ from __future__ import unicode_literals
from datetime import date
from django.contrib.auth.models import User
from django.contrib.contenttypes.models import ContentType
from django.test import TestCase
from django.urls import reverse
from rest_framework import status
from rest_framework.test import APITestCase
from dcim.models import Site
from extras.constants import CF_TYPE_TEXT, CF_TYPE_INTEGER, CF_TYPE_BOOLEAN, CF_TYPE_DATE, CF_TYPE_SELECT, CF_TYPE_URL
from extras.models import CustomField, CustomFieldValue, CustomFieldChoice
from users.models import Token
from utilities.testing import HttpStatusMixin
from utilities.testing import APITestCase
class CustomFieldTest(TestCase):
@@ -102,13 +99,11 @@ class CustomFieldTest(TestCase):
cf.delete()
class CustomFieldAPITest(HttpStatusMixin, APITestCase):
class CustomFieldAPITest(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(CustomFieldAPITest, self).setUp()
content_type = ContentType.objects.get_for_model(Site)