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