class ClientWithManyProductsTest(BaseTest): def setUp(self): super(ClientWithManyProductsTest, self).setUp() self.emanager = EntityManager(ClientWithManyProducts) self.client = BaseTestFactory.create_clients_with_many_products() @gen_test def test_saved_client_with_many_products(self): self.object_id = yield self.emanager.save(self.client) self.client_result = yield self.emanager.find_one(_id=self.object_id) yield self.emanager.remove(_id=self.object_id) self.assertTrue(compare_objs(self.client, self.client_result), "client was not properly saved")
class StoreWithThreeClientsOnlyTest(BaseTest): def setUp(self): super(StoreWithThreeClientsOnlyTest, self).setUp() self.emanager = EntityManager(StoreWithThreeClientsOnly) self.client = BaseTestFactory.create_store_with_some_clients() @gen_test def test_saved_client_with_address(self): self.object_id = yield self.emanager.save(self.client) self.client_result = yield self.emanager.find_one(_id=self.object_id) yield self.emanager.remove(_id=self.object_id) self.assertTrue(compare_objs(self.client, self.client_result), "client with 3 addresses was not properly saved")
class ClientWithOneAddressTest(BaseTest): def setUp(self): super(ClientWithOneAddressTest, self).setUp() self.emanager = EntityManager(ClientWithOneAddress) self.client_with_address = ClientWithOneAddress('leo', 'sza') self.address = Address(city='MANILA', street='test', number=10) self.client_with_address.set_address(self.address) @gen_test def test_saved_client_with_one_address(self): object_id = yield self.emanager.save(self.client_with_address) saved_client = yield self.emanager.find_one(_id=object_id) yield self.emanager.remove(_id=object_id) self.assertTrue(compare_objs(saved_client.address, self.address), "Address was not mapped")
class DeletedClientTest(BaseTest): def setUp(self): super(DeletedClientTest, self).setUp() self.emanager = EntityManager(SimpleClient) self.client1 = BaseTestFactory.create_client_with_3_address() self.client2 = BaseTestFactory.create_client_with_3_address() self.client3 = BaseTestFactory.create_client_with_3_address() @gen_test def test_client2_was_successful_deleted(self): yield self.emanager.remove_all() self.objectid_1 = yield self.emanager.save(self.client1) self.objectid_2 = yield self.emanager.save(self.client2) self.objectid_3 = yield self.emanager.save(self.client3) yield self.emanager.remove(_id=self.objectid_2) self.clients = yield self.emanager.find() self.assertEqual(self.objectid_1, self.clients[0]._id) self.assertEqual(self.objectid_3, self.clients[1]._id) yield self.emanager.remove_all()