1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
# Copyright 2024 Cloudbase Solutions Srl
# All Rights Reserved.
from cliff import lister
from unittest import mock
from coriolisclient.cli import endpoint_source_options
from coriolisclient.cli import utils as cli_utils
from coriolisclient.tests import test_base
class EndpointSourceOptionFormatterTestCase(test_base.CoriolisBaseTestCase):
"""Test suite for the Coriolis Client Endpoint Source Option Formatter."""
def setUp(self):
super(EndpointSourceOptionFormatterTestCase, self).setUp()
self.endpoint = endpoint_source_options.EndpointSourceOptionFormatter()
def test_get_sorted_list(self):
obj1 = mock.Mock()
obj2 = mock.Mock()
obj3 = mock.Mock()
obj1.name = "app1"
obj2.name = "app2"
obj3.name = "app3"
obj_list = [obj2, obj1, obj3]
result = self.endpoint._get_sorted_list(obj_list)
self.assertEqual(
[obj1, obj2, obj3],
result
)
@mock.patch.object(cli_utils, 'format_json_for_object_property')
def test_get_formatted_data(self, mock_format_json_for_object_property):
obj = mock.Mock()
obj.name = mock.sentinel.name
obj.to_dict = mock.Mock(
return_value={"config_default": mock.sentinel.config_default}
)
result = self.endpoint._get_formatted_data(obj)
self.assertEqual(
(
mock.sentinel.name,
mock_format_json_for_object_property.return_value,
mock.sentinel.config_default
),
result
)
class ListEndpointSourceOptionsTestCase(test_base.CoriolisBaseTestCase):
"""Test suite for the Coriolis Client List Endpoint Source Options."""
def setUp(self):
self.mock_app = mock.Mock()
super(ListEndpointSourceOptionsTestCase, self).setUp()
self.endpoint = endpoint_source_options.ListEndpointSourceOptions(
self.mock_app, mock.sentinel.app_args)
@mock.patch.object(cli_utils, 'add_args_for_json_option_to_parser')
@mock.patch.object(lister.Lister, 'get_parser')
def test_get_parser(
self,
mock_get_parser,
mock_add_args_for_json_option_to_parser
):
result = self.endpoint.get_parser(mock.sentinel.prog_name)
self.assertEqual(
mock_get_parser.return_value,
result
)
mock_get_parser.assert_called_once_with(mock.sentinel.prog_name)
mock_add_args_for_json_option_to_parser.assert_called_once_with(
mock_get_parser.return_value, 'environment')
@mock.patch.object(endpoint_source_options.EndpointSourceOptionFormatter,
'list_objects')
@mock.patch.object(cli_utils, 'get_option_value_from_args')
def test_take_action(
self,
mock_get_option_value_from_args,
mock_list_objects
):
args = mock.Mock()
args.options = mock.sentinel.options
mock_endpoints = mock.Mock()
mock_edo = mock.Mock()
self.mock_app.client_manager.coriolis.endpoints = mock_endpoints
mock_edo = (self.mock_app.
client_manager.coriolis.endpoint_source_options)
result = self.endpoint.take_action(args)
self.assertEqual(
mock_list_objects.return_value,
result
)
mock_get_option_value_from_args.assert_called_once_with(
args, 'environment', error_on_no_value=False)
mock_edo.list.assert_called_once_with(
mock_endpoints.get_endpoint_id_for_name(args.endpoint),
environment=mock_get_option_value_from_args.return_value,
option_names=mock.sentinel.options
)
mock_list_objects.assert_called_once_with(mock_edo.list.return_value)
|
