Skip to content
Snippets Groups Projects
Commit cbebe04f authored by Mark Skipper's avatar Mark Skipper
Browse files

add filter to transport api

parent 6d08684c
Branches
No related tags found
No related merge requests found
...@@ -33,7 +33,7 @@ def test_get_items_returns_all_items(): ...@@ -33,7 +33,7 @@ def test_get_items_returns_all_items():
def test_filter_by_body(): def test_filter_by_body():
item1 = ItemFactory(body="one") item1 = ItemFactory(body="one")
item2 = ItemFactory(body="two") item2 = ItemFactory(body="two")
payload = get(ItemList, '/items', {'body': 'one'}).data payload = get(ItemList, '/items', data={'body': 'one'}).data
assert len(payload) == 1 assert len(payload) == 1
assert payload[0]['body'] == "one" assert payload[0]['body'] == "one"
...@@ -4,8 +4,8 @@ from data_layer.handlers import Message ...@@ -4,8 +4,8 @@ from data_layer.handlers import Message
from rest_api.views import ItemList from rest_api.views import ItemList
def get_messages(): # TODO rename get_items def get_messages(**kwargs): # TODO rename get_items
request = APIRequestFactory().get('/items') request = APIRequestFactory().get('/items', kwargs)
view = ItemList.as_view() view = ItemList.as_view()
return view(request).data return view(request).data
......
...@@ -16,5 +16,19 @@ def test_get_items_returns_items(): ...@@ -16,5 +16,19 @@ def test_get_items_returns_items():
item = ItemFactory(body="test") item = ItemFactory(body="test")
items = dl.get_messages() items = dl.get_messages()
assert len(items) == 1
[item] = items [item] = items
assert item['body'] == 'test' 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'
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment