diff --git a/django/website/rest_api/tests/item_list_view_tests.py b/django/website/rest_api/tests/item_list_view_tests.py index b7659a1813b9bf8218db725085e8229ac6544774..a98e5d9370872e5004983f7a3c49874183ec4eeb 100644 --- a/django/website/rest_api/tests/item_list_view_tests.py +++ b/django/website/rest_api/tests/item_list_view_tests.py @@ -33,7 +33,7 @@ def test_get_items_returns_all_items(): def test_filter_by_body(): item1 = ItemFactory(body="one") item2 = ItemFactory(body="two") - payload = get(ItemList, '/items', {'body': 'one'}).data + payload = get(ItemList, '/items', data={'body': 'one'}).data assert len(payload) == 1 assert payload[0]['body'] == "one" diff --git a/django/website/transport/data_layer_transport.py b/django/website/transport/data_layer_transport.py index 3ec650e86c82d19dfdbecc58a550cc99d00bd0d5..9d3def0dc23b65a94f45d26aac3210ab50022169 100644 --- a/django/website/transport/data_layer_transport.py +++ b/django/website/transport/data_layer_transport.py @@ -4,8 +4,8 @@ from data_layer.handlers import Message from rest_api.views import ItemList -def get_messages(): # TODO rename get_items - request = APIRequestFactory().get('/items') +def get_messages(**kwargs): # TODO rename get_items + request = APIRequestFactory().get('/items', kwargs) view = ItemList.as_view() return view(request).data diff --git a/django/website/transport/tests/item_list_tests.py b/django/website/transport/tests/item_list_tests.py index 9a385416db04c5007632c5410c2f578bca3dd623..42e9b6f7c6fe6b3fa604c4d8907d7e77ec463940 100644 --- a/django/website/transport/tests/item_list_tests.py +++ b/django/website/transport/tests/item_list_tests.py @@ -16,5 +16,19 @@ def test_get_items_returns_items(): item = ItemFactory(body="test") items = dl.get_messages() + + assert len(items) == 1 [item] = items assert item['body'] == 'test' + + +@pytest.mark.django_db +def test_get_items_filters_by_body(): + item1 = ItemFactory(body="one") + item2 = ItemFactory(body="two") + + items = dl.get_messages(body='one') + + assert len(items) == 1 + [item] = items + assert item['body'] == 'one'