Esempio n. 1
0
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 ")
Esempio n. 2
0
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))
Esempio n. 3
0
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 ")
Esempio n. 4
0
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()
Esempio n. 5
0
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()
Esempio n. 6
0
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))