190 lines
7.1 KiB
Python
190 lines
7.1 KiB
Python
# 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 copy
|
|
|
|
from karborclient.osc.v1 import triggers as osc_triggers
|
|
from karborclient.tests.unit.osc.v1 import fakes
|
|
from karborclient.v1 import triggers
|
|
|
|
|
|
TRIGGER_INFO = {
|
|
"id": "2a9ce1f3-cc1a-4516-9435-0ebb13caa398",
|
|
"name": "My backup trigger",
|
|
"type": "time",
|
|
"properties": {
|
|
"format": "calendar",
|
|
"pattern": "BEGIN:VEVENT\\nRRULE:FREQ=HOURLY;INTERVAL=1;\\nEND:VEVENT",
|
|
"start_time": "2015-12-17T08:30:00",
|
|
"end_time": "2016-03-17T08:30:00",
|
|
"window": "3600"
|
|
}
|
|
}
|
|
|
|
|
|
class TestTriggers(fakes.TestDataProtection):
|
|
def setUp(self):
|
|
super(TestTriggers, self).setUp()
|
|
self.triggers_mock = self.app.client_manager.data_protection.triggers
|
|
self.triggers_mock.reset_mock()
|
|
|
|
|
|
class TestListTriggers(TestTriggers):
|
|
def setUp(self):
|
|
super(TestListTriggers, self).setUp()
|
|
self.triggers_mock.list.return_value = [triggers.Trigger(
|
|
None, copy.deepcopy(TRIGGER_INFO))]
|
|
|
|
# Command to test
|
|
self.cmd = osc_triggers.ListTriggers(self.app, None)
|
|
|
|
def test_triggers_list(self):
|
|
arglist = ['--name', 'My backup trigger']
|
|
verifylist = [('name', 'My backup trigger')]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
|
|
# Check that columns are correct
|
|
expected_columns = (
|
|
['Id', 'Name', 'Type', 'Properties'])
|
|
self.assertEqual(expected_columns, columns)
|
|
|
|
# Check that data is correct
|
|
expected_data = [("2a9ce1f3-cc1a-4516-9435-0ebb13caa398",
|
|
"My backup trigger",
|
|
"time",
|
|
{"format": "calendar",
|
|
"pattern": "BEGIN:VEVENT\\nRRULE:FREQ=HOURLY;INTERVAL=1;\\nEND:VEVENT", # noqa
|
|
"start_time": "2015-12-17T08:30:00",
|
|
"end_time": "2016-03-17T08:30:00",
|
|
"window": "3600"})]
|
|
self.assertEqual(expected_data, list(data))
|
|
|
|
|
|
class TestCreateTrigger(TestTriggers):
|
|
def setUp(self):
|
|
super(TestCreateTrigger, self).setUp()
|
|
self.triggers_mock.create.return_value = triggers.Trigger(
|
|
None, copy.deepcopy(TRIGGER_INFO))
|
|
# Command to test
|
|
self.cmd = osc_triggers.CreateTrigger(self.app, None)
|
|
|
|
def test_trigger_create(self):
|
|
arglist = ['My backup trigger',
|
|
'time',
|
|
"'format'='calendar',"
|
|
"'pattern'='BEGIN:VEVENT\\nRRULE:FREQ=HOURLY;INTERVAL=1;\\nEND:VEVENT'," # noqa
|
|
"'start_time'='2015-12-17T08:30:00',"
|
|
"'end_time'='2016-03-17T08:30:00',"
|
|
"'window'='3600'"]
|
|
verifylist = [('name', 'My backup trigger'),
|
|
('type', 'time'),
|
|
('properties', "'format'='calendar',"
|
|
"'pattern'='BEGIN:VEVENT\\nRRULE:FREQ=HOURLY;INTERVAL=1;\\nEND:VEVENT'," # noqa
|
|
"'start_time'='2015-12-17T08:30:00',"
|
|
"'end_time'='2016-03-17T08:30:00',"
|
|
"'window'='3600'")]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.cmd.take_action(parsed_args)
|
|
|
|
# Check that correct arguments were passed
|
|
self.triggers_mock.create.assert_called_once_with(
|
|
'My backup trigger',
|
|
'time',
|
|
"'format'='calendar',"
|
|
"'pattern'='BEGIN:VEVENT\\nRRULE:FREQ=HOURLY;INTERVAL=1;\\nEND:VEVENT'," # noqa
|
|
"'start_time'='2015-12-17T08:30:00',"
|
|
"'end_time'='2016-03-17T08:30:00',"
|
|
"'window'='3600'")
|
|
|
|
|
|
class TestUpdateTrigger(TestTriggers):
|
|
def setUp(self):
|
|
super(TestUpdateTrigger, self).setUp()
|
|
self.triggers_mock.get.return_value = triggers.Trigger(
|
|
None, copy.deepcopy(TRIGGER_INFO))
|
|
self.triggers_mock.update.return_value = triggers.Trigger(
|
|
None, copy.deepcopy(TRIGGER_INFO))
|
|
# Command to test
|
|
self.cmd = osc_triggers.UpdateTrigger(self.app, None)
|
|
|
|
def test_trigger_update(self):
|
|
arglist = ['2a9ce1f3-cc1a-4516-9435-0ebb13caa398',
|
|
'--name', 'My backup trigger']
|
|
verifylist = [('trigger_id', '2a9ce1f3-cc1a-4516-9435-0ebb13caa398'),
|
|
('name', 'My backup trigger')]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.cmd.take_action(parsed_args)
|
|
|
|
# Check that correct arguments were passed
|
|
self.triggers_mock.update.assert_called_once_with(
|
|
'2a9ce1f3-cc1a-4516-9435-0ebb13caa398',
|
|
{'name': 'My backup trigger'})
|
|
|
|
|
|
class TestDeleteTrigger(TestTriggers):
|
|
def setUp(self):
|
|
super(TestDeleteTrigger, self).setUp()
|
|
self.triggers_mock.get.return_value = triggers.Trigger(
|
|
None, copy.deepcopy(TRIGGER_INFO))
|
|
# Command to test
|
|
self.cmd = osc_triggers.DeleteTrigger(self.app, None)
|
|
|
|
def test_trigger_delete(self):
|
|
arglist = ['2a9ce1f3-cc1a-4516-9435-0ebb13caa398']
|
|
verifylist = [('trigger', ['2a9ce1f3-cc1a-4516-9435-0ebb13caa398'])]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.cmd.take_action(parsed_args)
|
|
|
|
# Check that correct arguments were passed
|
|
self.triggers_mock.delete.assert_called_once_with(
|
|
'2a9ce1f3-cc1a-4516-9435-0ebb13caa398')
|
|
|
|
|
|
class TestShowTrigger(TestTriggers):
|
|
def setUp(self):
|
|
super(TestShowTrigger, self).setUp()
|
|
self._trigger_info = copy.deepcopy(TRIGGER_INFO)
|
|
self.triggers_mock.get.return_value = triggers.Trigger(
|
|
None, self._trigger_info)
|
|
|
|
# Command to test
|
|
self.cmd = osc_triggers.ShowTrigger(self.app, None)
|
|
|
|
def test_trigger_show(self):
|
|
arglist = ['2a9ce1f3-cc1a-4516-9435-0ebb13caa398']
|
|
verifylist = [('trigger', '2a9ce1f3-cc1a-4516-9435-0ebb13caa398')]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
|
|
# Check that columns are correct
|
|
expected_columns = (
|
|
'id', 'name', 'properties', 'type')
|
|
self.assertEqual(expected_columns, columns)
|
|
|
|
# Check that data is correct
|
|
self.assertEqual(self._trigger_info['id'], data[0])
|
|
self.assertEqual(self._trigger_info['name'], data[1])
|
|
self.assertEqual(self._trigger_info['properties'], data[2])
|
|
self.assertEqual(self._trigger_info['type'], data[3])
|