def test_serialize_list(self):
fake_item1 = test_worker.FakeMapping()
fake_item2 = test_worker.FakeMapping()
fake_item2.a_prop = 13
def times_2(item):
return item.a_prop * 2
serialized = list(self.a_worker.serialize(
(fake_item1, fake_item2), times_2=times_2))
self.assertEqual(serialized[0]['a_prop'], fake_item1.a_prop)
self.assertEqual(serialized[1]['a_prop'], fake_item2.a_prop)
self.assertEqual(serialized[0]['times_2'], fake_item1.a_prop * 2)
self.assertEqual(serialized[1]['times_2'], fake_item2.a_prop * 2)
def test_invalid_worker(self):
foreman = test_worker.FakeForeman()
a_worker = InvalidWorker(
foreman, managed_sqla_map=test_worker.FakeMapping, managed_sqla_map_name="fake"
)
with self.assertRaises(NotImplementedError):
a_worker.serialize(test_worker.FakeMapping())
def test_serialize_one_with_func(self):
fake_item = test_worker.FakeMapping()
def times_3(item):
return item.a_prop * 3
serialized = self.a_worker.serialize_one(fake_item, times_3=times_3)
self.assertEqual(serialized['times_3'], fake_item.a_prop * 3)
def test(self):
an_object = test_worker.FakeMapping()
self.assertEqual(an_object.a_prop, 12)
update = mapping.update(sqla_obj=an_object, schema=a_schema, a_prop=13)
self.assertEqual(an_object.a_prop, 13)
self.assertTrue(update)
update = mapping.update(sqla_obj=an_object, schema=a_schema, a_prop=13)
self.assertFalse(update)
update = mapping.update(sqla_obj=an_object, schema=a_schema)
self.assertFalse(update)
update = mapping.update(sqla_obj=an_object, schema=a_schema, a_req_prop="bb")
self.assertEqual(an_object.a_req_prop, "bb")
self.assertTrue(update)
def test_serialize(self):
fake_item = test_worker.FakeMapping()
serialized = self.a_worker.serialize(fake_item)
self.assertTrue(isinstance(serialized, dict))
self.assertEqual(serialized["a_prop"], fake_item.a_prop)