def test_render_as_slug(self):
obj = Mock()
obj.attached = self.addon
field = SlugOrPrimaryKeyRelatedField(render_as='slug', read_only=True)
field.bind('attached', None)
assert field.to_representation(field.get_attribute(obj)) == self.addon.slug
def test_render_as_pks_many(self):
obj = Mock()
obj.attached = [self.addon]
field = SlugOrPrimaryKeyRelatedField(many=True, read_only=True)
field.bind('attached', None)
assert field.to_representation(field.get_attribute(obj)) == [self.addon.pk]
def test_render_as_pks_many(self):
obj = Mock()
obj.attached = [self.addon]
field = SlugOrPrimaryKeyRelatedField(many=True, read_only=True)
field.bind('attached', None)
assert (field.to_representation(field.get_attribute(obj)) ==
[self.addon.pk])
def test_render_as_slug(self):
obj = Mock()
obj.attached = self.addon
field = SlugOrPrimaryKeyRelatedField(render_as='slug',
read_only=True)
field.bind('attached', None)
assert (field.to_representation(field.get_attribute(obj)) ==
self.addon.slug)
