Add test_vim_types_success

Add unit tests for vim_types API.

Change-Id: I188f7a65f6573ae3fb5813a43c34245eebb2f4fa
Issue-ID: MULTICLOUD-184
Signed-off-by: Ethan Lynn <ethanlynnl@vmware.com>

1 files changed, 36 insertions, 0 deletions

diff --git a/multivimbroker/multivimbroker/tests/test_vim_types.py b/multivimbroker/multivimbroker/tests/test_vim_types.py
new file mode 100644
index 0000000..a3fe049
--- /dev/null
+++ b/multivimbroker/multivimbroker/tests/test_vim_types.py
@@ -0,0 +1,36 @@
+# 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 mock
+import unittest
+
+from rest_framework.views import status
+
+from multivimbroker.forwarder.views import VIMTypes
+
+
+class TestUrls(unittest.TestCase):
+    def setUp(self):
+        self.view = VIMTypes()
+
+    def test_vim_types_success(self):
+        resp = self.view.get(mock.Mock())
+        expect_body = [
+            {
+                'vim_type': u'openstack',
+                'versions': [u'ocata', u'titanium_cloud']
+            },
+            {
+                'vim_type': u'vmware',
+                'versions': [u'4.0']
+            }
+        ]
+        self.assertEqual(status.HTTP_200_OK, resp.status_code)
+        self.assertEqual(expect_body, resp.data)