NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-20 02:06:42 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Refactor APITestCase to subclass TestCase

Browse Source
This commit is contained in:
Jeremy Stretch 2020-01-30 17:57:34 -05:00
parent 61ac7c44ba
commit 179abcc79d
1 changed files with 20 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
netbox/utilities/testing.py
View File

@ -3,7 +3,7 @@ from contextlib import contextmanager
from django.contrib.auth.models import Permission, User from django.contrib.auth.models import Permission, User
from django.test import Client, TestCase as _TestCase from django.test import Client, TestCase as _TestCase
from rest_framework.test import APITestCase as _APITestCase from rest_framework.test import APIClient
from users.models import Token from users.models import Token
@ -21,6 +21,10 @@ class TestCase(_TestCase):
self.client = Client() self.client = Client()
self.client.force_login(self.user) self.client.force_login(self.user)
#
# Permissions management
#
def add_permissions(self, *names): def add_permissions(self, *names):
""" """
Assign a set of permissions to the test user. Accepts permission names in the form <app>.<action>_<model>. Assign a set of permissions to the test user. Accepts permission names in the form <app>.<action>_<model>.
@ -39,8 +43,22 @@ class TestCase(_TestCase):
perm = Permission.objects.get(content_type__app_label=app, codename=codename) perm = Permission.objects.get(content_type__app_label=app, codename=codename)
self.user.user_permissions.remove(perm) self.user.user_permissions.remove(perm)
#
# Convenience methods
#
class APITestCase(_APITestCase): def assertHttpStatus(self, response, expected_status):
"""
TestCase method. Provide more detail in the event of an unexpected HTTP response.
"""
err_message = "Expected HTTP status {}; received {}: {}"
self.assertEqual(response.status_code, expected_status, err_message.format(
expected_status, response.status_code, getattr(response, 'data', 'No data')
))
class APITestCase(TestCase):
client_class = APIClient
def setUp(self): def setUp(self):
""" """
@ -50,15 +68,6 @@ class APITestCase(_APITestCase):
self.token = Token.objects.create(user=self.user) self.token = Token.objects.create(user=self.user)
self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(self.token.key)} self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(self.token.key)}
def assertHttpStatus(self, response, expected_status):
"""
Provide more detail in the event of an unexpected HTTP response.
"""
err_message = "Expected HTTP status {}; received {}: {}"
self.assertEqual(response.status_code, expected_status, err_message.format(
expected_status, response.status_code, getattr(response, 'data', 'No data')
))
def create_test_user(username='testuser', permissions=list()): def create_test_user(username='testuser', permissions=list()):
""" """