44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
from django.test import TestCase, Client
|
|
from django.contrib.auth.models import Permission
|
|
from knox.models import AuthToken
|
|
|
|
from authentication.models import ExtendedUser
|
|
|
|
|
|
class ApiTest(TestCase):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.user = ExtendedUser.objects.create_user('testuser', 'test', 'test')
|
|
self.user.user_permissions.add(*Permission.objects.all())
|
|
self.user.save()
|
|
self.token = AuthToken.objects.create(user=self.user)
|
|
self.client = Client(headers={'Authorization': 'Token ' + self.token[1]})
|
|
|
|
def test_root(self):
|
|
from core.settings import SYSTEM3_VERSION
|
|
response = self.client.get('/api/')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json()["framework_version"], SYSTEM3_VERSION)
|
|
|
|
def test_events(self):
|
|
response = self.client.get('/api/2/events/')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json(), [])
|
|
|
|
def test_containers(self):
|
|
response = self.client.get('/api/2/boxes/')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json(), [])
|
|
|
|
def test_files(self):
|
|
response = self.client.get('/api/2/files/')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json(), [])
|
|
|
|
def test_items(self):
|
|
from inventory.models import Event
|
|
Event.objects.create(slug='TEST1', name='Event')
|
|
response = self.client.get('/api/2/TEST1/items/')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json(), [])
|