Exemplo n.º 1
0
class TestSerializableModelEncoder:
    def setup(self):
        self.encoder = SerializableModelEncoder()

    def test_default_serialize(self):
        assert self.encoder.default(SerializableObject()) == 'test123'

    def test_default_queryset(self):
        User = get_user_model()
        assert self.encoder.default(User.objects.all()) == list(
            User.objects.all())

    def test_default_fallback(self):
        with pytest.raises(TypeError):
            self.encoder.default(object())
Exemplo n.º 2
0
 def setup(self):
     self.encoder = SerializableModelEncoder()