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 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 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 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 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()
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()
