diff --git a/django/website/rest_api/tests/taxonomy_and_term_create_tests.py b/django/website/rest_api/tests/taxonomy_and_term_create_tests.py index 409adebc8b77e4222266289bd855aab4c126e7ef..9e5a1bc059c7e5d00595bb480c387be4cbaa2525 100644 --- a/django/website/rest_api/tests/taxonomy_and_term_create_tests.py +++ b/django/website/rest_api/tests/taxonomy_and_term_create_tests.py @@ -25,17 +25,31 @@ def create_category(name): return view(request, pk=id) +def count_taxonomies(): + return len(get_taxonomies().data) + + +def get_taxonomies(): + url = reverse('taxonomy-list') + request = APIRequestFactory().get(url) + view = TaxonomyViewSet.as_view(actions={'get': 'list'}) + response = view(request) + assert status.is_success(response.status_code), response.data + + return response + + @pytest.mark.django_db def test_create_a_category(): category = "Test Ebola Questions" - old_count = Taxonomy.objects.count() + old_count = count_taxonomies() assert not Taxonomy.objects.filter(name=category).exists() response = create_category(category) assert status.is_success(response.status_code), response.data - new_count = Taxonomy.objects.count() + new_count = count_taxonomies() assert new_count - old_count == 1 assert Taxonomy.objects.filter(name=category).exists()