diff --git a/django/website/rest_api/tests/item_view_tests.py b/django/website/rest_api/tests/item_view_tests.py index 14a9f9ee4904dcfd33805235cb4f7014bd9d3222..b9a42003fbf706364beece3f9897ffd38e6cd6b9 100644 --- a/django/website/rest_api/tests/item_view_tests.py +++ b/django/website/rest_api/tests/item_view_tests.py @@ -1,6 +1,7 @@ from __future__ import unicode_literals, absolute_import -from django.test import TestCase +import pytest + from rest_framework.test import APIRequestFactory from data_layer.tests.factories import ItemFactory @@ -8,24 +9,24 @@ from data_layer.tests.factories import ItemFactory from rest_api.views import ItemList -class ItemViewTests(TestCase): - - def get(self, view_class, url): - view = view_class.as_view() - request = APIRequestFactory().get(url) - return view(request) +def get(view_class, url): + view = view_class.as_view() + request = APIRequestFactory().get(url) + return view(request) - def test_get_items_returns_empty_if_no_items(self): - response = self.get(ItemList, '/items/') - self.assertEqual(response.data, []) +@pytest.mark.django_db +def test_get_items_returns_empty_if_no_items(): + response = get(ItemList, '/items/') + assert response.data == [] - def test_get_items_returns_all_items(self): - item = ItemFactory() +@pytest.mark.django_db +def test_get_items_returns_all_items(): + item = ItemFactory() - payload = self.get(ItemList, '/items/').data + payload = get(ItemList, '/items/').data - self.assertEqual(len(payload), 1) - self.assertEqual(payload[0]['body'], item.body) - # I don't much care for this test. - # It's testing too much of the stack, and its messy because the payload - # isn't the same kind data strcture even as ItemSerializer(item).data + assert len(payload) == 1 + assert payload[0]['body'] == item.body + # I don't much care for this test. + # It's testing too much of the stack, and its messy because the payload + # isn't the same kind data strcture even as ItemSerializer(item).data