481 lines
13 KiB
Python
481 lines
13 KiB
Python
# Copyright (c) 2015 Huawei Technologies Co., Ltd.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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 mock import patch
|
|
import pecan
|
|
from pecan.configuration import set_config
|
|
from pecan.testing import load_test_app
|
|
|
|
from oslo_config import cfg
|
|
from oslo_config import fixture as fixture_config
|
|
import oslo_db.exception as db_exc
|
|
|
|
from tricircle.api import app
|
|
from tricircle.common import context
|
|
from tricircle.common import policy
|
|
from tricircle.db import core
|
|
from tricircle.tests import base
|
|
|
|
|
|
OPT_GROUP_NAME = 'keystone_authtoken'
|
|
cfg.CONF.import_group(OPT_GROUP_NAME, "keystonemiddleware.auth_token")
|
|
|
|
|
|
def fake_admin_context():
|
|
context_paras = {'is_admin': True}
|
|
return context.Context(**context_paras)
|
|
|
|
|
|
def fake_non_admin_context():
|
|
context_paras = {}
|
|
return context.Context(**context_paras)
|
|
|
|
|
|
class API_FunctionalTest(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(API_FunctionalTest, self).setUp()
|
|
|
|
self.addCleanup(set_config, {}, overwrite=True)
|
|
|
|
cfg.CONF.clear()
|
|
cfg.CONF.register_opts(app.common_opts)
|
|
|
|
self.CONF = self.useFixture(fixture_config.Config()).conf
|
|
|
|
self.CONF.set_override('auth_strategy', 'noauth')
|
|
self.CONF.set_override('tricircle_db_connection', 'sqlite://')
|
|
|
|
core.initialize()
|
|
core.ModelBase.metadata.create_all(core.get_engine())
|
|
|
|
self.context = context.get_admin_context()
|
|
|
|
policy.populate_default_rules()
|
|
|
|
self.app = self._make_app()
|
|
|
|
def _make_app(self, enable_acl=False):
|
|
self.config = {
|
|
'app': {
|
|
'root': 'tricircle.api.controllers.root.RootController',
|
|
'modules': ['tricircle.api'],
|
|
'enable_acl': enable_acl,
|
|
'errors': {
|
|
400: '/error',
|
|
'__force_dict__': True
|
|
}
|
|
},
|
|
}
|
|
|
|
return load_test_app(self.config)
|
|
|
|
def tearDown(self):
|
|
super(API_FunctionalTest, self).tearDown()
|
|
cfg.CONF.unregister_opts(app.common_opts)
|
|
pecan.set_config({}, overwrite=True)
|
|
core.ModelBase.metadata.drop_all(core.get_engine())
|
|
policy.reset()
|
|
|
|
|
|
class TestPodController(API_FunctionalTest):
|
|
"""Test version listing on root URI."""
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_admin_context)
|
|
def test_post_no_input(self):
|
|
pods = [
|
|
# missing pod
|
|
{
|
|
"pod_xxx":
|
|
{
|
|
"dc_name": "dc1",
|
|
"pod_az_name": "az1"
|
|
},
|
|
"expected_error": 400
|
|
}]
|
|
|
|
for test_pod in pods:
|
|
response = self.app.post_json(
|
|
'/v1.0/pods',
|
|
dict(pod_xxx=test_pod['pod_xxx']),
|
|
expect_errors=True)
|
|
|
|
self.assertEqual(response.status_int,
|
|
test_pod['expected_error'])
|
|
|
|
def fake_create_ag_az(context, ag_name, az_name):
|
|
raise db_exc.DBDuplicateEntry
|
|
|
|
def fake_create_ag_az_exp(context, ag_name, az_name):
|
|
raise Exception
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_admin_context)
|
|
@patch.object(core, 'create_resource',
|
|
new=fake_create_ag_az_exp)
|
|
def test_post_exception(self):
|
|
pods = [
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod1",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 500
|
|
},
|
|
]
|
|
|
|
self._test_and_check(pods)
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_admin_context)
|
|
def test_post_invalid_input(self):
|
|
|
|
pods = [
|
|
|
|
# missing az and pod
|
|
{
|
|
"pod":
|
|
{
|
|
"dc_name": "dc1",
|
|
"pod_az_name": "az1"
|
|
},
|
|
"expected_error": 422
|
|
},
|
|
|
|
# missing pod
|
|
{
|
|
"pod":
|
|
{
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1",
|
|
"az_name": "az1"
|
|
},
|
|
"expected_error": 422
|
|
},
|
|
|
|
# missing pod
|
|
{
|
|
"pod":
|
|
{
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1",
|
|
"az_name": "",
|
|
},
|
|
"expected_error": 422
|
|
},
|
|
|
|
# missing az
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1"
|
|
},
|
|
"expected_error": 422
|
|
},
|
|
|
|
# az & pod == ""
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1",
|
|
"az_name": ""
|
|
},
|
|
"expected_error": 422
|
|
},
|
|
|
|
# invalid pod
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1",
|
|
"az_name": "az1"
|
|
|
|
},
|
|
"expected_error": 422
|
|
}
|
|
|
|
]
|
|
|
|
self._test_and_check(pods)
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_admin_context)
|
|
def test_post_duplicate_top_region(self):
|
|
|
|
pods = [
|
|
|
|
# the first time to create TopRegion
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "TopRegion",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1"
|
|
},
|
|
"expected_error": 200
|
|
},
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "TopRegion2",
|
|
"pod_az_name": "",
|
|
"dc_name": "dc1"
|
|
},
|
|
"expected_error": 409
|
|
},
|
|
|
|
]
|
|
|
|
self._test_and_check(pods)
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_admin_context)
|
|
def test_post_duplicate_pod(self):
|
|
|
|
pods = [
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod1",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 200
|
|
},
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod1",
|
|
"pod_az_name": "az2",
|
|
"dc_name": "dc2",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 409
|
|
},
|
|
|
|
]
|
|
|
|
self._test_and_check(pods)
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_admin_context)
|
|
def test_post_pod_duplicate_top_region(self):
|
|
|
|
pods = [
|
|
|
|
# the first time to create TopRegion
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "TopRegion",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc1"
|
|
},
|
|
"expected_error": 200
|
|
},
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "TopRegion",
|
|
"pod_az_name": "az2",
|
|
"dc_name": "dc2",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 409
|
|
},
|
|
|
|
]
|
|
|
|
self._test_and_check(pods)
|
|
|
|
def _test_and_check(self, pods):
|
|
|
|
for test_pod in pods:
|
|
response = self.app.post_json(
|
|
'/v1.0/pods',
|
|
dict(pod=test_pod['pod']),
|
|
expect_errors=True)
|
|
|
|
self.assertEqual(response.status_int,
|
|
test_pod['expected_error'])
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_admin_context)
|
|
def test_get_all(self):
|
|
|
|
pods = [
|
|
|
|
# the first time to create TopRegion
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "TopRegion",
|
|
"pod_az_name": "",
|
|
"dc_name": "dc1",
|
|
"az_name": ""
|
|
},
|
|
"expected_error": 200
|
|
},
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod1",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc2",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 200
|
|
},
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod2",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc2",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 200
|
|
},
|
|
|
|
]
|
|
|
|
self._test_and_check(pods)
|
|
|
|
response = self.app.get('/v1.0/pods')
|
|
|
|
self.assertEqual(response.status_int, 200)
|
|
self.assertIn('TopRegion', response)
|
|
self.assertIn('Pod1', response)
|
|
self.assertIn('Pod2', response)
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_admin_context)
|
|
def test_get_delete_one(self):
|
|
|
|
pods = [
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod1",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc2",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 200,
|
|
},
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod2",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc2",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 200,
|
|
},
|
|
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod3",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc2",
|
|
"az_name": "AZ2"
|
|
},
|
|
"expected_error": 200,
|
|
},
|
|
|
|
]
|
|
|
|
self._test_and_check(pods)
|
|
|
|
response = self.app.get('/v1.0/pods')
|
|
self.assertEqual(response.status_int, 200)
|
|
|
|
return_pods = response.json
|
|
|
|
for ret_pod in return_pods['pods']:
|
|
|
|
_id = ret_pod['pod_id']
|
|
single_ret = self.app.get('/v1.0/pods/' + str(_id))
|
|
|
|
self.assertEqual(single_ret.status_int, 200)
|
|
|
|
one_pod_ret = single_ret.json
|
|
get_one_pod = one_pod_ret['pod']
|
|
|
|
self.assertEqual(get_one_pod['pod_id'],
|
|
ret_pod['pod_id'])
|
|
|
|
self.assertEqual(get_one_pod['region_name'],
|
|
ret_pod['region_name'])
|
|
|
|
self.assertEqual(get_one_pod['pod_az_name'],
|
|
ret_pod['pod_az_name'])
|
|
|
|
self.assertEqual(get_one_pod['dc_name'],
|
|
ret_pod['dc_name'])
|
|
|
|
self.assertEqual(get_one_pod['az_name'],
|
|
ret_pod['az_name'])
|
|
|
|
@patch.object(context, 'extract_context_from_environ',
|
|
new=fake_non_admin_context)
|
|
def test_non_admin_action(self):
|
|
|
|
pods = [
|
|
{
|
|
"pod":
|
|
{
|
|
"region_name": "Pod1",
|
|
"pod_az_name": "az1",
|
|
"dc_name": "dc2",
|
|
"az_name": "AZ1"
|
|
},
|
|
"expected_error": 401,
|
|
},
|
|
]
|
|
self._test_and_check(pods)
|
|
|
|
response = self.app.get('/v1.0/pods/1234567890',
|
|
expect_errors=True)
|
|
self.assertEqual(response.status_int, 401)
|
|
|
|
response = self.app.get('/v1.0/pods',
|
|
expect_errors=True)
|
|
self.assertEqual(response.status_int, 401)
|
|
|
|
response = self.app.delete('/v1.0/pods/1234567890',
|
|
expect_errors=True)
|
|
self.assertEqual(response.status_int, 401)
|