X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=multivimbroker%2Fmultivimbroker%2Ftests%2Ftest_check_capacity.py;h=d7a7caef232b77e779cc29eba4063687af9ee21d;hb=9d536985089479573d5d292e20fc18a9adfd72f4;hp=0a852e83b125003ebb1e4fbd060a89126a0600ed;hpb=9a0d43b54519c8e08683e522255370ccba137902;p=multicloud%2Fframework.git

diff --git a/multivimbroker/multivimbroker/tests/test_check_capacity.py b/multivimbroker/multivimbroker/tests/test_check_capacity.py
index 0a852e8..d7a7cae 100644
--- a/multivimbroker/multivimbroker/tests/test_check_capacity.py
+++ b/multivimbroker/multivimbroker/tests/test_check_capacity.py
@@ -40,10 +40,10 @@ class CheckCapacityTest(unittest.TestCase):
                                           "/v0/check_vim_capacity")
         with mock.patch.object(self.view, "send") as send:
             plugin_resp = mock.Mock()
-            plugin_resp.body = """{
+            plugin_resp.content = """{
                 "result": true
             }"""
-            plugin_resp.status_code = status.HTTP_200_OK
+            plugin_resp.status_code = str(status.HTTP_200_OK)
             send.return_value = plugin_resp
 
             resp = self.view.post(req)
@@ -52,3 +52,42 @@ class CheckCapacityTest(unittest.TestCase):
             }
             self.assertEqual(status.HTTP_200_OK, resp.status_code)
             self.assertDictEqual(expect_body, resp.data)
+
+    def test_check_capacity_no_suitable_vim(self):
+        req = mock.Mock()
+        req.body = """
+        {
+            "vCPU": 1,
+            "Memory": 1,
+            "Storage": 500,
+            "VIMs": ["openstack_RegionOne"]
+        }"""
+        req.get_full_path.return_value = ("http://msb.onap.org/api/multicloud"
+                                          "/v0/check_vim_capacity")
+        with mock.patch.object(self.view, "send") as send:
+            plugin_resp = mock.Mock()
+            plugin_resp.content = """{
+                "result": false
+            }"""
+            plugin_resp.status_code = str(status.HTTP_200_OK)
+            send.return_value = plugin_resp
+
+            resp = self.view.post(req)
+            expect_body = {
+                "VIMs": []
+            }
+            self.assertEqual(status.HTTP_200_OK, resp.status_code)
+            self.assertDictEqual(expect_body, resp.data)
+
+    def test_check_capacity_invalid_input(self):
+        req = mock.Mock()
+        req.body = "hello world"
+        req.get_full_path.return_value = ("http://msb.onap.org/api/multicloud"
+                                          "/v0/check_vim_capacity")
+        expect_body = {
+            "error": ("Invalidate request body "
+                      "No JSON object could be decoded.")
+        }
+        resp = self.view.post(req)
+        self.assertEqual(status.HTTP_400_BAD_REQUEST, resp.status_code)
+        self.assertDictEqual(expect_body, resp.data)