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

Test nested Terms within Items from api

parent aabab489
No related branches found
No related tags found
No related merge requests found
from __future__ import unicode_literals, absolute_import
import pytest
from rest_framework.test import APIRequestFactory
from data_layer.tests.factories import ItemFactory
from taxonomies.tests.factories import TermFactory
from rest_api.views import ItemViewSet
from ..views import ItemViewSet
def get(data=None):
......@@ -52,3 +50,18 @@ def test_filter_by_id_list():
payload = get(data={'ids': item_ids}).data
assert len(payload) == 10
@pytest.mark.django_db
def test_item_listed_with_associated_terms():
item = ItemFactory()
terms = [TermFactory() for i in range(3)]
for term in terms:
item.terms.add(term)
[api_item] = get().data
nested_terms = api_item['terms']
assert len(nested_terms) == 3
term_names = [term.name for term in terms]
assert all(t['name'] in term_names for t in nested_terms)
from factory.django import DjangoModelFactory
from factory import fuzzy, SubFactory
from ..models import Term, Taxonomy
class TaxonmyFactory(DjangoModelFactory):
class Meta:
model = Taxonomy
name = fuzzy.FuzzyText()
class TermFactory(DjangoModelFactory):
class Meta:
model = Term
name = fuzzy.FuzzyText()
taxonomy = SubFactory(TaxonmyFactory)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment