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