class RemoveAllTest(BaseTest):
def setUp(self):
super(RemoveAllTest, self).setUp()
self.emanager = EntityManager(SimpleClient)
@gen_test
def test_remove_all_documents(self):
yield self.emanager.remove_all()
for i in range(0, 100):
yield self.emanager.save(SimpleClient(name="name" + str(i), last_name="last_name" + str(i)))
count = yield self.emanager.find()
self.assertEqual(len(count), 100, "Ops... something went wrong on save or find ")
yield self.emanager.remove_all()
count = yield self.emanager.find()
self.assertEqual(len(count), 0, "Ops... something went wrong on remove or find ")
class UpdatedClientTest(BaseTest):
def setUp(self):
super(UpdatedClientTest, 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_updated(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)
self.client_to_update = yield self.emanager.find_one(_id=self.objectid_2)
self.client_to_update.name = "updated_name"
self.client_to_update.last_name = "updated_last_name"
yield self.emanager.update(self.client_to_update)
self.updated_client = yield self.emanager.find_one(_id=self.objectid_2)
self.return_client1 = yield self.emanager.find_one(_id=self.objectid_1)
self.return_client3 = yield self.emanager.find_one(_id=self.objectid_3)
# check if client2 was updated
self.assertEqual(self.client_to_update.name, self.updated_client.name)
self.assertEqual(self.client_to_update.last_name, self.updated_client.last_name)
# check if nothing changed in client1 and client3
self.assertTrue(compare_objs(self.client1, self.return_client1))
self.assertTrue(compare_objs(self.client3, self.return_client3))
class RemoveAllTest(BaseTest):
def setUp(self):
super(RemoveAllTest, self).setUp()
self.emanager = EntityManager(SimpleClient)
@gen_test
def test_remove_all_documents(self):
yield self.emanager.remove_all()
for i in range(0, 100):
yield self.emanager.save(
SimpleClient(name="name" + str(i),
last_name="last_name" + str(i)))
count = yield self.emanager.find()
self.assertEqual(len(count), 100,
"Ops... something went wrong on save or find ")
yield self.emanager.remove_all()
count = yield self.emanager.find()
self.assertEqual(len(count), 0,
"Ops... something went wrong on remove or find ")
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()
class UpdatedClientTest(BaseTest):
def setUp(self):
super(UpdatedClientTest, 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_updated(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)
self.client_to_update = yield self.emanager.find_one(
_id=self.objectid_2)
self.client_to_update.name = "updated_name"
self.client_to_update.last_name = "updated_last_name"
yield self.emanager.update(self.client_to_update)
self.updated_client = yield self.emanager.find_one(_id=self.objectid_2)
self.return_client1 = yield self.emanager.find_one(_id=self.objectid_1)
self.return_client3 = yield self.emanager.find_one(_id=self.objectid_3)
# check if client2 was updated
self.assertEqual(self.client_to_update.name, self.updated_client.name)
self.assertEqual(self.client_to_update.last_name,
self.updated_client.last_name)
# check if nothing changed in client1 and client3
self.assertTrue(compare_objs(self.client1, self.return_client1))
self.assertTrue(compare_objs(self.client3, self.return_client3))