test_utils.py

"""
Test rest_framework_json_api's utils functions.
"""
fromrest_framework_json_apiimportutils

from ..serializersimportEntrySerializer
from ..testsimportTestBase


classGetRelatedResourceTests(TestBase):
"""
 Ensure the `get_related_resource_type` function returns correct types.
 """

deftest_reverse_relation(self):
"""
 Ensure reverse foreign keys have their types identified correctly.
 """
serializer=EntrySerializer()
field=serializer.fields['comments']

self.assertEqual(utils.get_related_resource_type(field), 'comments')

deftest_m2m_relation(self):
"""
 Ensure m2ms have their types identified correctly.
 """
serializer=EntrySerializer()
field=serializer.fields['authors']

self.assertEqual(utils.get_related_resource_type(field), 'authors')