70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
# (c) Copyright 2015 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
import json
|
|
from unittest import mock
|
|
|
|
from heat_dashboard.api.rest import heat
|
|
from heat_dashboard.test import helpers as test
|
|
from openstack_dashboard import api
|
|
|
|
|
|
class ValidateRestTestCase(test.TestCase):
|
|
@mock.patch.object(heat.api, 'heat')
|
|
def test_validate_post(self, hc):
|
|
body = '''{"template_url":"http://localhost/template.yaml"}'''
|
|
request = self.mock_rest_request(body=body)
|
|
hc.template_validate.return_value = ({'Description': 'foo'})
|
|
response = heat.Validate().post(request)
|
|
self.assertStatusCode(response, 200)
|
|
self.assertEqual(response.json, {"Description": "foo"})
|
|
kwargs = json.loads(body)
|
|
hc.template_validate.assert_called_once_with(request, **kwargs)
|
|
|
|
|
|
class HeatRestTestCase(test.TestCase):
|
|
#
|
|
# Services
|
|
#
|
|
|
|
@test.create_mocks({api.base: ('is_service_enabled',)})
|
|
@mock.patch.object(heat.api, 'heat')
|
|
def test_services_get(self, hc):
|
|
request = self.mock_rest_request(GET={})
|
|
|
|
self.mock_is_service_enabled.return_value = True
|
|
|
|
hc.service_list.return_value = [
|
|
mock.Mock(**{'to_dict.return_value': {'id': '1'}}),
|
|
mock.Mock(**{'to_dict.return_value': {'id': '2'}})
|
|
]
|
|
|
|
response = heat.Services().get(request)
|
|
self.assertStatusCode(response, 200)
|
|
self.assertEqual(response.content.decode('utf-8'),
|
|
'{"items": [{"id": "1"}, {"id": "2"}]}')
|
|
hc.service_list.assert_called_once_with(request)
|
|
self.mock_is_service_enabled.assert_called_once_with(
|
|
request, 'orchestration')
|
|
|
|
@test.create_mocks({api.base: ('is_service_enabled',)})
|
|
def test_services_get_disabled(self):
|
|
request = self.mock_rest_request(GET={})
|
|
|
|
self.mock_is_service_enabled.return_value = False
|
|
|
|
response = heat.Services().get(request)
|
|
self.assertStatusCode(response, 501)
|
|
self.mock_is_service_enabled.assert_called_once_with(
|
|
request, 'orchestration')
|