Merge "Add test_originHeaders"

[multicloud/framework.git] / multivimbroker / multivimbroker / tests / test_syscomm.py

diff --git a/multivimbroker/multivimbroker/tests/test_syscomm.py b/multivimbroker/multivimbroker/tests/test_syscomm.py
index b5097d7..1e88aa2 100644 (file)
--- a/multivimbroker/multivimbroker/tests/test_syscomm.py
+++ b/multivimbroker/multivimbroker/tests/test_syscomm.py
@@ -47,3 +47,20 @@ class TestSyscomm(unittest.TestCase):
         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"
+        }
+        expect_headers = {
+            "X-AUTH-TOKEN": "token_1"
+        }
+        ret_headers = syscomm.originHeaders(req)
+        self.assertDictEqual(expect_headers, ret_headers)