From 48120d383ffb4b925862f015d4fe132967d16d5a Mon Sep 17 00:00:00 2001 From: Yun Huang Date: Wed, 4 Jul 2018 18:36:07 +0800 Subject: Add ocata UT for vesagent tasks_processBacklogs Change-Id: I721ac5ba404550a076496e5bf51c697b79e1283d Issue-ID: MULTICLOUD-230 Signed-off-by: Yun Huang --- ocata/ocata/vesagent/tests_tasks.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/ocata/ocata/vesagent/tests_tasks.py b/ocata/ocata/vesagent/tests_tasks.py index a139c51b..e1e3c5c8 100644 --- a/ocata/ocata/vesagent/tests_tasks.py +++ b/ocata/ocata/vesagent/tests_tasks.py @@ -67,4 +67,16 @@ class VesTaskTest(unittest.TestCase): self.assertEquals(None, result) pass + @mock.patch.object(tasks, 'processBacklogsOfOneVIM') + @mock.patch.object(cache, 'get') + def test_tasks_processBacklogs(self, mock_cache_get, mock_tasks_processBacklogsOfOneVIM): + mock_VesAgentBacklogs_vimlist = ["windriver-hudson-dc_RegionOne"] + COUNT_TIME_SLOT_ONE_VIM = (1, 1) + mock_tasks_processBacklogsOfOneVIM.return_value = COUNT_TIME_SLOT_ONE_VIM + mock_cache_get.side_effect= [ + json.dumps(mock_VesAgentBacklogs_vimlist), + ] + result = tasks.processBacklogs() + self.assertEquals(COUNT_TIME_SLOT_ONE_VIM, result) + pass -- cgit 1.2.3-korg