From 46c1ee7f60a93fe474cdb57618c80cb96a239f09 Mon Sep 17 00:00:00 2001 From: Yun Huang Date: Fri, 15 Jun 2018 13:58:25 +0800 Subject: Add test case for vesagent_ctrl Change-Id: I733892bdb499d404761cf900ad44fc9ca0ad0133 Issue-ID: MULTICLOUD-230 Signed-off-by: Yun Huang --- windriver/titanium_cloud/vesagent/tests.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) (limited to 'windriver') diff --git a/windriver/titanium_cloud/vesagent/tests.py b/windriver/titanium_cloud/vesagent/tests.py index 7026d569..520c8ace 100644 --- a/windriver/titanium_cloud/vesagent/tests.py +++ b/windriver/titanium_cloud/vesagent/tests.py @@ -21,7 +21,7 @@ from rest_framework import status from django.core.cache import cache from common.msapi import extsys - +from titanium_cloud.vesagent import vesagent_ctrl MOCK_VIM_INFO = { @@ -43,9 +43,11 @@ MOCK_VIM_INFO = { 'insecure': 'True', } + class VesAgentCtrlTest(unittest.TestCase): def setUp(self): self.client = Client() + self.view = vesagent_ctrl.VesAgentCtrl() def tearDown(self): pass @@ -58,3 +60,17 @@ class VesAgentCtrlTest(unittest.TestCase): response = self.client.get("/api/multicloud-titanium_cloud/v0/windriver-hudson-dc_RegionOne/vesagent") self.assertEqual(status.HTTP_200_OK, response.status_code, response.content) + + @mock.patch.object(vesagent_ctrl.VesAgentCtrl, 'buildBacklogsOneVIM') + @mock.patch.object(extsys, 'get_vim_by_id') + def test_post(self, mock_get_vim_by_id, mock_buildBacklogsOneVIM): + mock_get_vim_by_id.return_value = MOCK_VIM_INFO + mock_buildBacklogsOneVIM.return_value = "mocked vesagent_backlogs" + mock_request = mock.Mock() + mock_request.META = {"testkey":"testvalue"} + mock_request.data = {"testdatakey":"testdatavalue"} + + response = self.view.post(request=mock_request, vimid="windriver-hudson-dc_RegionOne") + self.assertEquals(status.HTTP_201_CREATED, response.status_code) + + pass -- cgit 1.2.3-korg