def setUp(self):
super(SerializerTestCase, self).setUp()
self.s = Serializer()
self.ms = ModelSerializer()
self.d = Deserializer()
self.md = ModelDeserializer()
class SerializerTestCase(FlaskPeeweeTestCase):
def setUp(self):
super(SerializerTestCase, self).setUp()
self.s = Serializer()
self.ms = ModelSerializer()
self.d = Deserializer()
self.md = ModelDeserializer()
def test_serializer(self):
users = self.create_users()
serialized = self.s.serialize_object(self.admin)
self.assertEqual(
serialized,
{
"id": self.admin.id,
"username": "******",
"password": self.admin.password,
"join_date": self.admin.join_date.strftime("%Y-%m-%d %H:%M:%S"),
"active": True,
"admin": True,
"email": "",
},
)
serialized = self.s.serialize_object(self.admin, fields=("id", "username"))
self.assertEqual(serialized, {"id": self.admin.id, "username": "******"})
serialized = self.s.serialize_object(self.admin, exclude=("password", "join_date"))
self.assertEqual(
serialized, {"id": self.admin.id, "username": "******", "active": True, "admin": True, "email": ""}
)
def test_deserializer(self):
users = self.create_users()
deserialized = self.d.deserialize_object(
{
"id": self.admin.id,
"username": "******",
"password": self.admin.password,
"join_date": self.admin.join_date.strftime("%Y-%m-%d %H:%M:%S"),
"active": True,
"admin": True,
},
User(),
)
for attr in ["id", "username", "password", "active", "admin"]:
self.assertEqual(getattr(deserialized, attr), getattr(self.admin, attr))
self.assertEqual(
deserialized.join_date.strftime("%Y-%m-%d %H:%M:%S"), self.admin.join_date.strftime("%Y-%m-%d %H:%M:%S")
)
admin_pk = self.admin.id
deserialized = self.d.deserialize_object({"username": "******", "active": False, "admin": False}, self.admin)
self.assertEqual(deserialized.username, "edited")
self.assertEqual(deserialized.admin, False)
self.assertEqual(deserialized.active, False)
self.assertEqual(deserialized.id, admin_pk)
deserialized.save()
self.assertEqual(User.select().count(), 3)
edited = User.get(username="******")
self.assertEqual(edited.id, admin_pk)
def test_model_serializer(self):
users = self.create_users()
serialized = self.ms.serialize_object(self.admin)
self.assertEqual(
serialized,
{
"__model__": "User",
"__module__": "flask_peewee.tests.test_app",
"id": self.admin.id,
"username": "******",
"password": self.admin.password,
"join_date": self.admin.join_date.strftime("%Y-%m-%d %H:%M:%S"),
"active": True,
"admin": True,
"email": "",
},
)
serialized = self.ms.serialize_object(self.admin, fields=("id", "username"))
self.assertEqual(
serialized,
{
"__model__": "User",
"__module__": "flask_peewee.tests.test_app",
"id": self.admin.id,
"username": "******",
},
)
def test_model_deserializer(self):
users = self.create_users()
deserialized = self.md.deserialize_object(
{
"__model__": "User",
"__module__": "flask_peewee.tests.test_app",
"id": self.admin.id,
"username": "******",
"password": self.admin.password,
"join_date": self.admin.join_date.strftime("%Y-%m-%d %H:%M:%S"),
"active": True,
"admin": True,
}
)
for attr in ["id", "username", "password", "active", "admin"]:
self.assertEqual(getattr(deserialized, attr), getattr(self.admin, attr))
self.assertEqual(
deserialized.join_date.strftime("%Y-%m-%d %H:%M:%S"), self.admin.join_date.strftime("%Y-%m-%d %H:%M:%S")
)
admin_pk = self.admin.id
deserialized = self.md.deserialize_object(
{
"__model__": "User",
"__module__": "flask_peewee.tests.test_app",
"username": "******",
"active": False,
"admin": False,
},
self.admin,
)
self.assertEqual(deserialized.username, "edited")
self.assertEqual(deserialized.admin, False)
self.assertEqual(deserialized.active, False)
self.assertEqual(deserialized.id, admin_pk)
deserialized.save()
self.assertEqual(User.select().count(), 3)
edited = User.get(username="******")
self.assertEqual(edited.id, admin_pk)
def test_s_and_d(self):
self.create_users()
s = self.s.serialize_object(self.admin)
d = self.d.deserialize_object(s, User())
self.assertEqual(d, self.admin)
s = self.ms.serialize_object(self.admin)
d = self.md.deserialize_object(s)
self.assertEqual(d, self.admin)
