4669bcc628
Now that we no longer support py27, we can use the standard library unittest.mock module instead of the third party mock lib. Change-Id: I12e1a2a4a22116cabd09a3b808f871d98e4bd1f2 Signed-off-by: Sean McGinnis <sean.mcginnis@gmail.com>
337 lines
12 KiB
Python
337 lines
12 KiB
Python
# Copyright 2017 Intel Corporation.
|
|
#
|
|
# 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 unittest import mock
|
|
|
|
from osc_lib import exceptions
|
|
from osc_lib.tests import utils as tests_utils
|
|
|
|
from neutronclient.osc.v2.sfc import sfc_service_graph
|
|
from neutronclient.tests.unit.osc.v2.sfc import fakes
|
|
|
|
|
|
def _get_id(client, id_or_name, resource):
|
|
return id_or_name
|
|
|
|
|
|
class TestListSfcServiceGraph(fakes.TestNeutronClientOSCV2):
|
|
_service_graphs = fakes.FakeSfcServiceGraph.create_sfc_service_graphs(
|
|
count=1)
|
|
columns = ('ID', 'Name', 'Branching Points')
|
|
columns_long = ('ID', 'Name', 'Branching Points', 'Description', 'Project')
|
|
_service_graph = _service_graphs[0]
|
|
data = [
|
|
_service_graph['id'],
|
|
_service_graph['name'],
|
|
_service_graph['port_chains']
|
|
]
|
|
data_long = [
|
|
_service_graph['id'],
|
|
_service_graph['name'],
|
|
_service_graph['port_chains'],
|
|
_service_graph['description'],
|
|
_service_graph['project_id']
|
|
]
|
|
_service_graph1 = {'service_graphs': _service_graph}
|
|
_service_graph_id = _service_graph['id']
|
|
|
|
def setUp(self):
|
|
super(TestListSfcServiceGraph, self).setUp()
|
|
mock.patch(
|
|
'neutronclient.osc.v2.sfc.sfc_service_graph._get_id',
|
|
new=_get_id).start()
|
|
self.neutronclient.list_sfc_service_graphs = mock.Mock(
|
|
return_value={'service_graphs': self._service_graphs}
|
|
)
|
|
# Get the command object to test
|
|
self.cmd = sfc_service_graph.ListSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
def test_list_sfc_service_graphs(self):
|
|
arglist = []
|
|
verifylist = []
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns = self.cmd.take_action(parsed_args)[0]
|
|
sgs = self.neutronclient.list_sfc_service_graphs()['service_graphs']
|
|
sg = sgs[0]
|
|
data = [
|
|
sg['id'],
|
|
sg['name'],
|
|
sg['port_chains']
|
|
]
|
|
self.assertEqual(list(self.columns), columns)
|
|
self.assertEqual(self.data, data)
|
|
|
|
def test_list_sfc_service_graphs_with_long_option(self):
|
|
arglist = ['--long']
|
|
verifylist = [('long', True)]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns = self.cmd.take_action(parsed_args)[0]
|
|
sgs = self.neutronclient.list_sfc_service_graphs()['service_graphs']
|
|
sg = sgs[0]
|
|
data = [
|
|
sg['id'],
|
|
sg['name'],
|
|
sg['port_chains'],
|
|
sg['description'],
|
|
sg['project_id']
|
|
]
|
|
self.assertEqual(list(self.columns_long), columns)
|
|
self.assertEqual(self.data_long, data)
|
|
|
|
|
|
class TestCreateSfcServiceGraph(fakes.TestNeutronClientOSCV2):
|
|
_service_graph = fakes.FakeSfcServiceGraph.create_sfc_service_graph()
|
|
|
|
columns = ('ID', 'Name', 'Branching Points')
|
|
columns_long = ('Branching Points', 'Description', 'ID', 'Name', 'Project')
|
|
|
|
def get_data(self):
|
|
return (
|
|
self._service_graph['port_chains'],
|
|
self._service_graph['description'],
|
|
self._service_graph['id'],
|
|
self._service_graph['name'],
|
|
self._service_graph['project_id'],
|
|
)
|
|
|
|
def setUp(self):
|
|
super(TestCreateSfcServiceGraph, self).setUp()
|
|
mock.patch('neutronclient.osc.v2.sfc.sfc_service_graph._get_id',
|
|
new=_get_id).start()
|
|
self.neutronclient.create_sfc_service_graph = mock.Mock(
|
|
return_value={'service_graph': self._service_graph})
|
|
self.data = self.get_data()
|
|
self.cmd = sfc_service_graph.CreateSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
def test_create_sfc_service_graph(self):
|
|
arglist = []
|
|
verifylist = []
|
|
|
|
self.assertRaises(tests_utils.ParserException, self.check_parser,
|
|
self.cmd, arglist, verifylist)
|
|
|
|
def test_create_sfc_service_graph_without_loop(self):
|
|
bp1_str = 'pc1:pc2,pc3'
|
|
bp2_str = 'pc2:pc4'
|
|
self.cmd = sfc_service_graph.CreateSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
arglist = [
|
|
"--description", self._service_graph['description'],
|
|
"--branching-point", bp1_str,
|
|
"--branching-point", bp2_str,
|
|
self._service_graph['name']]
|
|
|
|
pcs = {'pc1': ['pc2', 'pc3'], 'pc2': ['pc4']}
|
|
|
|
verifylist = [
|
|
("description", self._service_graph['description']),
|
|
("branching_points", [bp1_str, bp2_str]),
|
|
("name", self._service_graph['name'])
|
|
]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = (self.cmd.take_action(parsed_args))
|
|
|
|
self.neutronclient.create_sfc_service_graph.assert_called_once_with({
|
|
'service_graph': {
|
|
'description': self._service_graph['description'],
|
|
'name': self._service_graph['name'],
|
|
'port_chains': pcs
|
|
}
|
|
})
|
|
self.assertEqual(self.columns_long, columns)
|
|
self.assertEqual(self.data, data)
|
|
|
|
def test_create_sfc_service_graph_with_loop(self):
|
|
bp1_str = 'pc1:pc2,pc3;'
|
|
bp2_str = 'pc2:pc1'
|
|
self.cmd = sfc_service_graph.CreateSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
arglist = [
|
|
"--description", self._service_graph['description'],
|
|
"--branching-point", bp1_str,
|
|
"--branching-point", bp2_str,
|
|
self._service_graph['name']]
|
|
|
|
verifylist = [
|
|
("description", self._service_graph['description']),
|
|
("branching_points", [bp1_str, bp2_str]),
|
|
("name", self._service_graph['name'])
|
|
]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
self.assertRaises(
|
|
exceptions.CommandError, self.cmd.take_action, parsed_args)
|
|
|
|
def test_create_sfc_service_graph_invalid_port_chains(self):
|
|
bp1_str = 'pc1:pc2,pc3:'
|
|
self.cmd = sfc_service_graph.CreateSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
arglist = [
|
|
"--description", self._service_graph['description'],
|
|
"--branching-point", bp1_str,
|
|
self._service_graph['name']]
|
|
|
|
verifylist = [
|
|
("description", self._service_graph['description']),
|
|
("branching_points", [bp1_str]),
|
|
("name", self._service_graph['name'])
|
|
]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
self.assertRaises(
|
|
exceptions.CommandError, self.cmd.take_action, parsed_args)
|
|
|
|
def test_create_sfc_service_graph_duplicate_src_chains(self):
|
|
bp1_str = 'pc1:pc2,pc3;'
|
|
bp2_str = 'pc1:pc4'
|
|
self.cmd = sfc_service_graph.CreateSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
arglist = [
|
|
"--description", self._service_graph['description'],
|
|
"--branching-point", bp1_str,
|
|
"--branching-point", bp2_str,
|
|
self._service_graph['name']]
|
|
|
|
verifylist = [
|
|
("description", self._service_graph['description']),
|
|
("branching_points", [bp1_str, bp2_str]),
|
|
("name", self._service_graph['name'])
|
|
]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
self.assertRaises(
|
|
exceptions.CommandError, self.cmd.take_action, parsed_args)
|
|
|
|
|
|
class TestDeleteSfcServiceGraph(fakes.TestNeutronClientOSCV2):
|
|
|
|
_service_graph = fakes.FakeSfcServiceGraph.create_sfc_service_graphs(
|
|
count=1)
|
|
|
|
def setUp(self):
|
|
super(TestDeleteSfcServiceGraph, self).setUp()
|
|
self.neutronclient.delete_sfc_service_graph = mock.Mock(
|
|
return_value=None)
|
|
self.cmd = sfc_service_graph.DeleteSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
def test_delete_sfc_service_graph(self):
|
|
client = self.app.client_manager.neutronclient
|
|
mock_service_graph_delete = client.delete_sfc_service_graph
|
|
arglist = [
|
|
self._service_graph[0]['id'],
|
|
]
|
|
verifylist = [
|
|
('service_graph', self._service_graph[0]['id']),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
result = self.cmd.take_action(parsed_args)
|
|
mock_service_graph_delete.assert_called_once_with(
|
|
self._service_graph[0]['id'])
|
|
self.assertIsNone(result)
|
|
|
|
|
|
class TestShowSfcServiceGraph(fakes.TestNeutronClientOSCV2):
|
|
|
|
_sg = fakes.FakeSfcServiceGraph.create_sfc_service_graph()
|
|
columns = ('ID', 'Name', 'Branching Points')
|
|
columns_long = ('Branching Points', 'Description', 'ID', 'Name', 'Project')
|
|
data = (
|
|
_sg['id'],
|
|
_sg['name'],
|
|
_sg['port_chains']
|
|
)
|
|
data_long = (
|
|
_sg['port_chains'],
|
|
_sg['description'],
|
|
_sg['id'],
|
|
_sg['name'],
|
|
_sg['project_id']
|
|
)
|
|
|
|
_service_graph = {'service_graph': _sg}
|
|
_service_graph_id = _sg['id']
|
|
|
|
def setUp(self):
|
|
super(TestShowSfcServiceGraph, self).setUp()
|
|
mock.patch(
|
|
'neutronclient.osc.v2.sfc.sfc_service_graph._get_id',
|
|
new=_get_id).start()
|
|
self.neutronclient.show_sfc_service_graph = mock.Mock(
|
|
return_value=self._service_graph
|
|
)
|
|
# Get the command object to test
|
|
self.cmd = sfc_service_graph.ShowSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
def test_service_graph_show(self):
|
|
client = self.app.client_manager.neutronclient
|
|
mock_service_graph_show = client.show_sfc_service_graph
|
|
arglist = [
|
|
self._service_graph_id,
|
|
]
|
|
verifylist = [
|
|
('service_graph', self._service_graph_id),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
mock_service_graph_show.assert_called_once_with(self._service_graph_id)
|
|
self.assertEqual(self.columns_long, columns)
|
|
self.assertEqual(self.data_long, data)
|
|
|
|
|
|
class TestSetSfcServiceGraph(fakes.TestNeutronClientOSCV2):
|
|
_service_graph = fakes.FakeSfcServiceGraph.create_sfc_service_graph()
|
|
_service_graph_name = _service_graph['name']
|
|
_service_graph_id = _service_graph['id']
|
|
|
|
def setUp(self):
|
|
super(TestSetSfcServiceGraph, self).setUp()
|
|
mock.patch('neutronclient.osc.v2.sfc.sfc_service_graph._get_id',
|
|
new=_get_id).start()
|
|
self.neutronclient.update_sfc_service_graph = mock.Mock(
|
|
return_value=None)
|
|
self.cmd = sfc_service_graph.SetSfcServiceGraph(
|
|
self.app, self.namespace)
|
|
|
|
def test_set_service_graph(self):
|
|
client = self.app.client_manager.neutronclient
|
|
mock_service_graph_update = client.update_sfc_service_graph
|
|
arglist = [
|
|
self._service_graph_name,
|
|
'--name', 'name_updated',
|
|
'--description', 'desc_updated'
|
|
]
|
|
verifylist = [
|
|
('service_graph', self._service_graph_name),
|
|
('name', 'name_updated'),
|
|
('description', 'desc_updated'),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
result = self.cmd.take_action(parsed_args)
|
|
attrs = {'service_graph': {
|
|
'name': 'name_updated',
|
|
'description': 'desc_updated'}
|
|
}
|
|
mock_service_graph_update.assert_called_once_with(
|
|
self._service_graph_name, attrs)
|
|
self.assertIsNone(result)
|