Newton UT for capacity check nova limits failed

Change-Id: Ied372034ada4cd417f8dccd982b09b3b2b489ee3
Issue-ID: MULTICLOUD-168
Signed-off-by: Yun Huang <yun.huang@windriver.com>

1 files changed, 28 insertions, 0 deletions

diff --git a/newton/newton/resource/tests/test_capacity.py b/newton/newton/resource/tests/test_capacity.py
index 605aed8b..071997e3 100644
--- a/newton/newton/resource/tests/test_capacity.py
+++ b/newton/newton/resource/tests/test_capacity.py
@@ -58,6 +58,12 @@ TEST_REQ_SUCCESS_SOURCE = {
     "Storage": "200"
 }
 
+TEST_REQ_FAILED_SOURCE = {
+    "vCPU": "17",
+    "Memory": "4096",
+    "Storage": "200"
+}
+
 class TestCapacity(test_base.TestRequest):
     def setUp(self):
         super(TestCapacity, self).setUp()
@@ -90,3 +96,25 @@ class TestCapacity(test_base.TestRequest):
         self.assertEquals(status.HTTP_200_OK, response.status_code)
         self.assertEqual({"result": True}, response.data)
 
+    @mock.patch.object(VimDriverUtils, 'get_session')
+    @mock.patch.object(VimDriverUtils, 'get_vim_info')
+    def test_capacity_check_nova_limits_failed(self, mock_get_vim_info, mock_get_session):
+        mock_get_vim_info.return_value = mock_info.MOCK_VIM_INFO
+        mock_get_session.return_value = test_base.get_mock_session(
+            ["get"], {
+                "side_effect": [
+                    self._get_mock_response(MOCK_GET_TENANT_LIMIT_RESPONSE),
+                    self._get_mock_response(MOCK_GET_HYPER_STATATICS_RESPONSE),
+                    self._get_mock_response(MOCK_GET_STORAGE_RESPONSE),
+                ]
+            })
+
+        response = self.client.post((
+            "/api/%s/v0/windriver-hudson-dc_RegionOne/"
+            "capacity_check" % test_base.MULTIVIM_VERSION),
+            TEST_REQ_FAILED_SOURCE,
+            HTTP_X_AUTH_TOKEN=mock_info.MOCK_TOKEN_ID)
+
+        self.assertEquals(status.HTTP_200_OK, response.status_code)
+        self.assertEqual({"result": False}, response.data)
+