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