X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=multivimbroker%2Fmultivimbroker%2Ftests%2Ftest_syscomm.py;h=1a5017cf3dfd07fb2bf030c94524fd89896fbda3;hb=refs%2Fchanges%2F49%2F46949%2F1;hp=ac4387780983e45f0ed99875f74c364c2e1c3d11;hpb=a88f4070e869f1f7d9733275d1d142bd0ca2bf93;p=multicloud%2Fframework.git

diff --git a/multivimbroker/multivimbroker/tests/test_syscomm.py b/multivimbroker/multivimbroker/tests/test_syscomm.py
index ac43877..1a5017c 100644
--- a/multivimbroker/multivimbroker/tests/test_syscomm.py
+++ b/multivimbroker/multivimbroker/tests/test_syscomm.py
@@ -11,6 +11,7 @@
 import mock
 import unittest
 
+from multivimbroker.pub import exceptions
 from multivimbroker.pub.msapi import extsys
 from multivimbroker.pub.utils import syscomm
 
@@ -37,3 +38,45 @@ class TestSyscomm(unittest.TestCase):
         expect_path = "multicloud-ocata/v0/openstack_regionone/identity"
         ret_path = syscomm.getMultivimDriver("openstack_regionone", full_path)
         self.assertEqual(expect_path, ret_path)
+
+    def test_findMultivimDriver_no_type(self):
+        vim = {"type": "wrong type"}
+        self.assertRaises(exceptions.NotFound, syscomm.findMultivimDriver, vim)
+
+    def test_findMultivimDriver_no_version(self):
+        vim = {"type": "openstack"}
+        ret = syscomm.findMultivimDriver(vim)
+        self.assertEqual("multicloud-ocata", ret)
+
+    def test_findMultivimDriver_with_version(self):
+        vim = {"type": "openstack", "version": "titanium_cloud"}
+        ret = syscomm.findMultivimDriver(vim)
+        self.assertEqual("multicloud-titanium_cloud", ret)
+
+    def test_originHeaders(self):
+        req = mock.Mock()
+        req.META = {
+            "HTTP_X_AUTH_TOKEN": "token_1",
+            "NOT_STARTSWITH_HTTP": "value_1",
+            "CONTENT_TYPE": "application/json"
+        }
+        expect_headers = {
+            "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))