Add test_findMultivimDriver_with_version

[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 ac43877..b80047d 100644 (file)
--- 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,17 @@ 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)