148 lines
6.1 KiB
Python
148 lines
6.1 KiB
Python
# Copyright (c) 2015 Huawei Tech. 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.
|
|
|
|
import mock
|
|
from mock import patch
|
|
import pecan
|
|
import unittest
|
|
|
|
from oslo_utils import uuidutils
|
|
|
|
from tricircle.common import client
|
|
from tricircle.common import constants
|
|
from tricircle.common import context
|
|
from tricircle.db import api
|
|
from tricircle.db import core
|
|
from tricircle.db import models
|
|
from tricircle.nova_apigw.controllers import volume
|
|
|
|
|
|
class FakeVolume(object):
|
|
def to_dict(self):
|
|
pass
|
|
|
|
|
|
class VolumeTest(unittest.TestCase):
|
|
def setUp(self):
|
|
core.initialize()
|
|
core.ModelBase.metadata.create_all(core.get_engine())
|
|
self.context = context.get_admin_context()
|
|
self.project_id = 'test_project'
|
|
self.controller = volume.VolumeController(self.project_id, '')
|
|
|
|
def _prepare_pod(self, bottom_pod_num=1):
|
|
t_pod = {'pod_id': 't_pod_uuid', 'pod_name': 't_region',
|
|
'az_name': ''}
|
|
api.create_pod(self.context, t_pod)
|
|
if bottom_pod_num == 1:
|
|
b_pod = {'pod_id': 'b_pod_uuid', 'pod_name': 'b_region',
|
|
'az_name': 'b_az'}
|
|
api.create_pod(self.context, b_pod)
|
|
return t_pod, b_pod
|
|
b_pods = []
|
|
for i in xrange(1, bottom_pod_num + 1):
|
|
b_pod = {'pod_id': 'b_pod_%d_uuid' % i,
|
|
'pod_name': 'b_region_%d' % i,
|
|
'az_name': 'b_az_%d' % i}
|
|
api.create_pod(self.context, b_pod)
|
|
b_pods.append(b_pod)
|
|
return t_pod, b_pods
|
|
|
|
@patch.object(pecan, 'abort')
|
|
@patch.object(client.Client, 'action_resources')
|
|
@patch.object(context, 'extract_context_from_environ')
|
|
def test_attach_volume(self, mock_context, mock_action, mock_abort):
|
|
mock_context.return_value = self.context
|
|
mock_action.return_value = FakeVolume()
|
|
|
|
t_pod, b_pods = self._prepare_pod(bottom_pod_num=2)
|
|
b_pod1 = b_pods[0]
|
|
b_pod2 = b_pods[1]
|
|
t_server_id = uuidutils.generate_uuid()
|
|
b_server_id = t_server_id
|
|
with self.context.session.begin():
|
|
core.create_resource(
|
|
self.context, models.ResourceRouting,
|
|
{'top_id': t_server_id, 'bottom_id': b_server_id,
|
|
'pod_id': b_pod1['pod_id'], 'project_id': self.project_id,
|
|
'resource_type': constants.RT_SERVER})
|
|
|
|
t_volume1_id = uuidutils.generate_uuid()
|
|
b_volume1_id = t_volume1_id
|
|
t_volume2_id = uuidutils.generate_uuid()
|
|
b_volume2_id = t_volume1_id
|
|
with self.context.session.begin():
|
|
core.create_resource(
|
|
self.context, models.ResourceRouting,
|
|
{'top_id': t_volume1_id, 'bottom_id': b_volume1_id,
|
|
'pod_id': b_pod1['pod_id'], 'project_id': self.project_id,
|
|
'resource_type': constants.RT_VOLUME})
|
|
core.create_resource(
|
|
self.context, models.ResourceRouting,
|
|
{'top_id': t_volume2_id, 'bottom_id': b_volume2_id,
|
|
'pod_id': b_pod2['pod_id'], 'project_id': self.project_id,
|
|
'resource_type': constants.RT_VOLUME})
|
|
|
|
# success case
|
|
self.controller.server_id = t_server_id
|
|
body = {'volumeAttachment': {'volumeId': t_volume1_id}}
|
|
self.controller.post(**body)
|
|
body = {'volumeAttachment': {'volumeId': t_volume1_id,
|
|
'device': '/dev/vdb'}}
|
|
self.controller.post(**body)
|
|
calls = [mock.call('server_volume', self.context,
|
|
'create_server_volume',
|
|
b_server_id, b_volume1_id, None),
|
|
mock.call('server_volume', self.context,
|
|
'create_server_volume',
|
|
b_server_id, b_volume1_id, '/dev/vdb')]
|
|
mock_action.assert_has_calls(calls)
|
|
|
|
# failure case, bad request
|
|
body = {'volumeAttachment': {'volumeId': t_volume2_id}}
|
|
self.controller.post(**body)
|
|
body = {'fakePara': ''}
|
|
self.controller.post(**body)
|
|
body = {'volumeAttachment': {}}
|
|
self.controller.post(**body)
|
|
# each part of path should not start with digit
|
|
body = {'volumeAttachment': {'volumeId': t_volume1_id,
|
|
'device': '/dev/001disk'}}
|
|
self.controller.post(**body)
|
|
# the first part should be "dev", and only two parts are allowed
|
|
body = {'volumeAttachment': {'volumeId': t_volume1_id,
|
|
'device': '/dev/vdb/disk'}}
|
|
self.controller.post(**body)
|
|
body = {'volumeAttachment': {'volumeId': t_volume1_id,
|
|
'device': '/disk/vdb'}}
|
|
self.controller.post(**body)
|
|
calls = [mock.call(400, 'Server and volume not in the same pod'),
|
|
mock.call(400, 'Request body not found'),
|
|
mock.call(400, 'Volume not set'),
|
|
mock.call(400, 'Invalid device path'),
|
|
mock.call(400, 'Invalid device path'),
|
|
mock.call(400, 'Invalid device path')]
|
|
mock_abort.assert_has_calls(calls)
|
|
|
|
# failure case, resource not found
|
|
body = {'volumeAttachment': {'volumeId': 'fake_volume_id'}}
|
|
self.controller.post(**body)
|
|
self.controller.server_id = 'fake_server_id'
|
|
body = {'volumeAttachment': {'volumeId': t_volume1_id}}
|
|
self.controller.post(**body)
|
|
calls = [mock.call(404, 'Volume not found'),
|
|
mock.call(404, 'Server not found')]
|
|
mock_abort.assert_has_calls(calls)
|