100 lines
4.2 KiB
Python
100 lines
4.2 KiB
Python
# Copyright 2017 GoDaddy
|
|
#
|
|
# 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.
|
|
|
|
from oslo_config import cfg
|
|
from oslo_config import fixture as oslo_fixture
|
|
import pecan.testing
|
|
|
|
from octavia.api import config as pconfig
|
|
from octavia.common import constants
|
|
from octavia.tests.functional.db import base as base_db_test
|
|
|
|
|
|
class TestRootController(base_db_test.OctaviaDBTestBase):
|
|
|
|
def get(self, app, path, params=None, headers=None, status=200,
|
|
expect_errors=False):
|
|
response = app.get(
|
|
path, params=params, headers=headers, status=status,
|
|
expect_errors=expect_errors)
|
|
return response
|
|
|
|
def setUp(self):
|
|
super(TestRootController, self).setUp()
|
|
self.conf = self.useFixture(oslo_fixture.Config(cfg.CONF))
|
|
self.conf.config(group='api_settings', auth_strategy=constants.NOAUTH)
|
|
|
|
def _get_versions_with_config(self, api_v1_enabled, api_v2_enabled):
|
|
self.conf.config(group='api_settings', api_v1_enabled=api_v1_enabled)
|
|
self.conf.config(group='api_settings', api_v2_enabled=api_v2_enabled)
|
|
app = pecan.testing.load_test_app({'app': pconfig.app,
|
|
'wsme': pconfig.wsme})
|
|
return self.get(app=app, path='/').json.get('versions', None)
|
|
|
|
def test_api_versions(self):
|
|
versions = self._get_versions_with_config(
|
|
api_v1_enabled=True, api_v2_enabled=True)
|
|
version_ids = tuple(v.get('id') for v in versions)
|
|
self.assertEqual(10, len(version_ids))
|
|
self.assertIn('v1', version_ids)
|
|
self.assertIn('v2.0', version_ids)
|
|
self.assertIn('v2.1', version_ids)
|
|
self.assertIn('v2.2', version_ids)
|
|
self.assertIn('v2.3', version_ids)
|
|
self.assertIn('v2.4', version_ids)
|
|
self.assertIn('v2.5', version_ids)
|
|
self.assertIn('v2.6', version_ids)
|
|
self.assertIn('v2.7', version_ids)
|
|
self.assertIn('v2.8', version_ids)
|
|
|
|
# Each version should have a 'self' 'href' to the API version URL
|
|
# [{u'rel': u'self', u'href': u'http://localhost/v2'}]
|
|
# Validate that the URL exists in the response
|
|
for version in versions:
|
|
url_version = None
|
|
if version['id'].startswith('v2.'):
|
|
url_version = 'v2'
|
|
else:
|
|
url_version = version['id']
|
|
version_url = 'http://localhost/{}'.format(url_version)
|
|
links = version['links']
|
|
# Note, there may be other links present, this test is for 'self'
|
|
version_link = [link for link in links if link['rel'] == 'self']
|
|
self.assertEqual(version_url, version_link[0]['href'])
|
|
|
|
def test_api_v1_disabled(self):
|
|
versions = self._get_versions_with_config(
|
|
api_v1_enabled=False, api_v2_enabled=True)
|
|
self.assertEqual(9, len(versions))
|
|
self.assertEqual('v2.0', versions[0].get('id'))
|
|
self.assertEqual('v2.1', versions[1].get('id'))
|
|
self.assertEqual('v2.2', versions[2].get('id'))
|
|
self.assertEqual('v2.3', versions[3].get('id'))
|
|
self.assertEqual('v2.4', versions[4].get('id'))
|
|
self.assertEqual('v2.5', versions[5].get('id'))
|
|
self.assertEqual('v2.6', versions[6].get('id'))
|
|
self.assertEqual('v2.7', versions[7].get('id'))
|
|
self.assertEqual('v2.8', versions[8].get('id'))
|
|
|
|
def test_api_v2_disabled(self):
|
|
versions = self._get_versions_with_config(
|
|
api_v1_enabled=True, api_v2_enabled=False)
|
|
self.assertEqual(1, len(versions))
|
|
self.assertEqual('v1', versions[0].get('id'))
|
|
|
|
def test_api_both_disabled(self):
|
|
versions = self._get_versions_with_config(
|
|
api_v1_enabled=False, api_v2_enabled=False)
|
|
self.assertEqual(0, len(versions))
|