X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=multivimbroker%2Fmultivimbroker%2Ftests%2Ftest_syscomm.py;h=75eda1c9d6c5d8e4d8298a91e05190e2c89d4b65;hb=9d536985089479573d5d292e20fc18a9adfd72f4;hp=1e88aa2a5af060a387e924ac485164138a260464;hpb=0b2cfa524556bf4e1541706ecb94e807cf0995bb;p=multicloud%2Fframework.git

diff --git a/multivimbroker/multivimbroker/tests/test_syscomm.py b/multivimbroker/multivimbroker/tests/test_syscomm.py
index 1e88aa2..75eda1c 100644
--- a/multivimbroker/multivimbroker/tests/test_syscomm.py
+++ b/multivimbroker/multivimbroker/tests/test_syscomm.py
@@ -20,8 +20,10 @@ class TestSyscomm(unittest.TestCase):
 
     def test_getVIMTypes(self):
         expected_body = {
-            "openstack": ["titanium_cloud", "ocata"],
+            "openstack": ["titanium_cloud", "ocata", "pike", "starlingx"],
             "vmware": ["4.0"],
+            "vsphere": ["6.5"],
+            "azure": ["1.0"],
         }
         ret = syscomm.getVIMTypes()
         for item in ret:
@@ -51,16 +53,32 @@ class TestSyscomm(unittest.TestCase):
     def test_findMultivimDriver_with_version(self):
         vim = {"type": "openstack", "version": "titanium_cloud"}
         ret = syscomm.findMultivimDriver(vim)
-        self.assertEqual("multicloud-titanium_cloud", ret)
+        self.assertEqual("multicloud-titaniumcloud", ret)
 
     def test_originHeaders(self):
         req = mock.Mock()
         req.META = {
             "HTTP_X_AUTH_TOKEN": "token_1",
-            "NOT_STARTSWITH_HTTP": "value_1"
+            "NOT_STARTSWITH_HTTP": "value_1",
+            "CONTENT_TYPE": "application/json"
         }
         expect_headers = {
-            "X-AUTH-TOKEN": "token_1"
+            "X-AUTH-TOKEN": "token_1",
+            "CONTENT-TYPE": "application/json"
         }
         ret_headers = syscomm.originHeaders(req)
         self.assertDictEqual(expect_headers, ret_headers)
+
+    def test_getHeadersKeys_no_connection(self):
+        resp = {}
+        self.assertEqual([], syscomm.getHeadersKeys(resp))
+
+    def test_getHeadersKeys_with_connection(self):
+        resp = {
+            "connection": "aa,bb",
+            "keep-alive": "yes",
+            "aa": "vaa",
+            "bb": "vbb",
+            "cc": "vcc"
+        }
+        self.assertEqual(["cc"], syscomm.getHeadersKeys(resp))