diff --git a/PDMI/api/tests.py b/PDMI/api/tests.py index 7ce503c..15093ff 100644 --- a/PDMI/api/tests.py +++ b/PDMI/api/tests.py @@ -1,3 +1,13 @@ from django.test import TestCase +from store.models import Module, Version + +class ApiModuleListTests(TestCase): + def setUp(self): + for name in ['module1', 'module2', 'module3']: + Module.objects.create(name=name) + def test_call_view_success(self): + response = self.client.get('/api/modules', follow=True) + self.assertEqual(response.status_code, 200) + self.assertEqual(response['content-type'], 'application/json') + self.assertJSONEqual(response.content, {'modules': ['module1', 'module2', 'module3']}) -# Create your tests here.