From 57835be52913410a8661c55297b7101aae3a8bb7 Mon Sep 17 00:00:00 2001 From: gecong1973 Date: Mon, 19 Nov 2018 01:16:00 -0800 Subject: [PATCH] Add client search unit test for sql The patch add client search unit testcase for sqlalchemy Change-Id: I53bceb89e54ea3c60e3e088af3c65863b4fd35c3 --- freezer_api/tests/unit/common.py | 4 +++ .../tests/unit/sqlalchemy/test_client.py | 28 +++++++++++++++++++ 2 files changed, 32 insertions(+) diff --git a/freezer_api/tests/unit/common.py b/freezer_api/tests/unit/common.py index 6c180877..142fa428 100644 --- a/freezer_api/tests/unit/common.py +++ b/freezer_api/tests/unit/common.py @@ -420,6 +420,10 @@ def get_fake_client_0(): def get_fake_client_1(): return copy.deepcopy(fake_client_entry_1) + +def get_fake_client_id(): + return uuid.uuid4().hex + fake_action_0 = { "freezer_action": { diff --git a/freezer_api/tests/unit/sqlalchemy/test_client.py b/freezer_api/tests/unit/sqlalchemy/test_client.py index 1c23a20b..4205aee0 100644 --- a/freezer_api/tests/unit/sqlalchemy/test_client.py +++ b/freezer_api/tests/unit/sqlalchemy/test_client.py @@ -77,3 +77,31 @@ class DbClientTestCase(base.DbTestCase): client_id=client_id) self.assertEqual(len(result), 0) + + def test_add_and_search_client(self): + count = 0 + clientids = [] + while (count < 20): + client_doc = copy.deepcopy(self.fake_client_doc) + clientid = common.get_fake_client_id() + client_doc['client_id'] = clientid + client_id = self.dbapi.add_client(user_id=self.fake_user_id, + doc=client_doc, + project_id=self.fake_project_id) + self.assertIsNotNone(client_id) + self.assertEqual(clientid, client_id) + clientids.append(client_id) + count += 1 + + result = self.dbapi.get_client(project_id=self.fake_project_id, + user_id=self.fake_user_id, + limit=10, + offset=0) + + self.assertIsNotNone(result) + self.assertEqual(len(result), 10) + + for index in range(len(result)): + clientmap = result[index] + clientid = clientmap['client'].get('client_id') + self.assertEqual(clientids[index], clientid)