forked from softlayer/softlayer-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path__init__.py
More file actions
197 lines (147 loc) · 5.66 KB
/
__init__.py
File metadata and controls
197 lines (147 loc) · 5.66 KB
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
"""
SoftLayer.testing
~~~~~~~~~~~~~~~~~
:license: MIT, see LICENSE for more details.
"""
# Disable pylint import error and too many methods error
# pylint: disable=invalid-name
from __future__ import print_function
import logging
import os.path
from click import testing
import mock
import testtools
import SoftLayer
from SoftLayer.CLI import core
from SoftLayer.CLI import environment
from SoftLayer.testing import xmlrpc
FIXTURE_PATH = os.path.abspath(os.path.join(__file__, '..', '..', 'fixtures'))
class MockableTransport(object):
"""Transport which is able to mock out specific API calls."""
def __init__(self, transport):
self.calls = []
self.mocked = {}
self.transport = transport
def __call__(self, call):
self._record_call(call)
key = _mock_key(call.service, call.method)
if key in self.mocked:
return self.mocked[key](call)
# Fall back to another transport (usually with fixtures)
return self.transport(call)
def set_mock(self, service, method):
"""Create a mock and return the mock object for the specific API call.
:param service: API service to mock
:param method: API method to mock
"""
_mock = mock.MagicMock()
self.mocked[_mock_key(service, method)] = _mock
return _mock
def clear(self):
"""Clear out mocks and call history."""
self.calls = []
self.mocked = {}
def _record_call(self, call):
"""Record and log the API call (for later assertions)."""
self.calls.append(call)
details = []
for prop in ['identifier',
'args',
'mask',
'filter',
'limit',
'offset']:
details.append('%s=%r' % (prop, getattr(call, prop)))
logging.info('%s::%s called; %s',
call.service, call.method, '; '.join(details))
def _mock_key(service, method):
"""Key to address a mock object in MockableTransport."""
return '%s::%s' % (service, method)
class TestCase(testtools.TestCase):
"""Testcase class with PEP-8 compatable method names."""
@classmethod
def setUpClass(cls):
"""Stand up fixtured/mockable XML-RPC server."""
cls.mocks = MockableTransport(SoftLayer.FixtureTransport())
cls.server = xmlrpc.create_test_server(cls.mocks)
host, port = cls.server.socket.getsockname()[:2]
cls.endpoint_url = "http://%s:%s" % (host, port)
@classmethod
def tearDownClass(cls):
"""Clean up the http server."""
cls.server.shutdown()
def set_up(self):
"""Aliased from setUp."""
pass
def tear_down(self):
"""Aliased from tearDown."""
pass
def setUp(self): # NOQA
testtools.TestCase.setUp(self)
self.mocks.clear()
transport = SoftLayer.XmlRpcTransport(endpoint_url=self.endpoint_url)
wrapped_transport = SoftLayer.TimingTransport(transport)
self.client = SoftLayer.BaseClient(transport=wrapped_transport)
self.env = environment.Environment()
self.env.client = self.client
self.set_up()
def tearDown(self): # NOQA
testtools.TestCase.tearDown(self)
self.tear_down()
self.mocks.clear()
def calls(self, service=None, method=None, **props):
"""Return all API calls made during the current test."""
conditions = []
if service is not None:
conditions.append(lambda call: call.service == service)
if method is not None:
conditions.append(lambda call: call.method == method)
if props:
conditions.append(lambda call: call_has_props(call, props))
return [call for call in self.mocks.calls
if all(cond(call) for cond in conditions)]
def assert_called_with(self, service, method, **props):
"""Used to assert that API calls were called with given properties.
Props are properties of the given transport.Request object.
"""
if self.calls(service, method, **props):
return
raise AssertionError('%s::%s was not called with given properties: %s'
% (service, method, props))
def assert_no_fail(self, result):
"""Fail when a failing click result has an error"""
if result.exception:
print(result.output)
raise result.exception
self.assertEqual(result.exit_code, 0)
def set_mock(self, service, method):
"""Set and return mock on the current client."""
return self.mocks.set_mock(service, method)
def run_command(self,
args=None,
env=None,
fixtures=True,
fmt='json'):
"""A helper that runs a SoftLayer CLI command.
This returns a click.testing.Result object.
"""
args = args or []
if fixtures is True:
args.insert(0, '--demo')
args.insert(0, '--format=%s' % fmt)
runner = testing.CliRunner()
return runner.invoke(core.cli, args=args, obj=env or self.env)
def call_has_props(call, props):
"""Check if a call has matching properties of a given props dictionary."""
for prop, expected_value in props.items():
actual_value = getattr(call, prop)
if actual_value != expected_value:
logging.info(
'%s::%s property mismatch, %s: expected=%r; actual=%r',
call.service,
call.method,
prop,
expected_value,
actual_value)
return False
return True