Add test_getVIMTypes

[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
new file mode 100644 (file)
index 0000000..bed857d
--- /dev/null
+++ b/multivimbroker/multivimbroker/tests/test_syscomm.py
@@ -0,0 +1,26 @@
+# Copyright (c) 2017-2018 VMware, Inc.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+import unittest
+
+from multivimbroker.pub.utils import syscomm
+
+
+class TestSyscomm(unittest.TestCase):
+
+    def test_getVIMTypes(self):
+        expected_body = {
+            "openstack": ["titanium_cloud", "ocata"],
+            "vmware": ["4.0"],
+        }
+        ret = syscomm.getVIMTypes()
+        for item in ret:
+            for v in item['versions']:
+                self.assertIn(v, expected_body[item['vim_type']])