diff options
| -rw-r--r-- | multivimbroker/multivimbroker/tests/test_vim_types.py | 36 |
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) |