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
|
"""
SoftLayer.tests.CLI.modules.config_tests
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
:license: MIT, see LICENSE for more details.
"""
import json
import os
import sys
import tempfile
from unittest import mock as mock
import SoftLayer
from SoftLayer import auth
from SoftLayer.CLI.config import setup as config
from SoftLayer.CLI import exceptions
from SoftLayer import consts
from SoftLayer import testing
from SoftLayer import transports
class TestHelpShow(testing.TestCase):
def set_up(self):
transport = transports.XmlRpcTransport(endpoint_url='http://endpoint-url',)
self.env.client = SoftLayer.BaseClient(
transport=transport,
auth=auth.BasicAuthentication('username', 'api-key'))
def test_show(self):
result = self.run_command(['config', 'show'])
self.assert_no_fail(result)
default_config = {
'Username': 'username',
'API Key': 'api-key',
'Endpoint URL': 'http://endpoint-url',
'Timeout': 'not set',
'Theme': 'dark'
}
result_config = json.loads(result.output)
self.assertEqual(result_config.get('Username'), default_config.get('Username'))
self.assertEqual(result_config.get('API Key'), default_config.get('API Key'))
self.assertEqual(result_config.get('Endpoint URL'), default_config.get('Endpoint URL'))
self.assertEqual(result_config.get('Timeout'), default_config.get('Timeout'))
# Could be either theme since were not setting it in this test
self.assertIn(result_config.get('Theme'), ['dark', 'light'])
class TestHelpSetup(testing.TestCase):
def set_up(self):
super(TestHelpSetup, self).set_up()
# NOTE(kmcdonald): since the endpoint_url is changed with the client
# in these commands, we need to ensure that a fixtured transport is
# used.
transport = testing.MockableTransport(SoftLayer.FixtureTransport())
self.env.client = SoftLayer.BaseClient(transport=transport)
self.config_file = "./test_config_file"
def tearDown(self):
# Windows doesn't let you write and read from temp files
# So use a real file instead.
if os.path.exists(self.config_file):
os.remove(self.config_file)
@mock.patch('SoftLayer.Client')
@mock.patch('SoftLayer.CLI.formatting.confirm')
@mock.patch('SoftLayer.CLI.environment.Environment.getpass')
@mock.patch('SoftLayer.CLI.environment.Environment.input')
def test_setup(self, mocked_input, getpass, confirm_mock, client):
client.return_value = self.env.client
if (sys.platform.startswith("win")):
self.skipTest("Test doesn't work in Windows")
with tempfile.NamedTemporaryFile() as config_file:
confirm_mock.return_value = True
getpass.return_value = 'A' * 64
mocked_input.side_effect = ['public', 'user', 0, 'dark']
result = self.run_command(['--config=%s' % config_file.name, 'config', 'setup'])
self.assert_no_fail(result)
self.assertIn('Configuration Updated Successfully', result.output)
contents = config_file.read().decode("utf-8")
self.assertIn('[softlayer]', contents)
self.assertIn('username = user', contents)
self.assertIn('api_key = AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA', contents)
self.assertIn('endpoint_url = %s' % consts.API_PUBLIC_ENDPOINT, contents)
self.assertIn('theme = dark', contents)
@mock.patch('SoftLayer.Client')
@mock.patch('SoftLayer.CLI.formatting.confirm')
@mock.patch('SoftLayer.CLI.environment.Environment.getpass')
@mock.patch('SoftLayer.CLI.environment.Environment.input')
def test_setup_float_timeout(self, mocked_input, getpass, confirm_mock, client):
client.return_value = self.env.client
confirm_mock.return_value = True
getpass.return_value = 'A' * 64
mocked_input.side_effect = ['public', 'user', 10.0, 'None']
result = self.run_command(['--config=%s' % self.config_file, 'config', 'setup'])
self.assert_no_fail(result)
self.assertIn('Configuration Updated Successfully', result.output)
with open(self.config_file, 'r') as config_file:
contents = config_file.read()
self.assertIn('[softlayer]', contents)
self.assertIn('username = user', contents)
self.assertIn('api_key = AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA', contents)
self.assertIn('endpoint_url = %s' % consts.API_PUBLIC_ENDPOINT, contents)
self.assertNotIn('timeout = 10.0\n', contents)
self.assertIn('timeout = 10\n', contents)
@mock.patch('SoftLayer.Client')
@mock.patch('SoftLayer.CLI.formatting.confirm')
@mock.patch('SoftLayer.CLI.environment.Environment.getpass')
@mock.patch('SoftLayer.CLI.environment.Environment.input')
def test_setup_cancel(self, mocked_input, getpass, confirm_mock, client):
client.return_value = self.env.client
with tempfile.NamedTemporaryFile() as config_file:
confirm_mock.return_value = False
getpass.return_value = 'A' * 64
mocked_input.side_effect = ['public', 'user', 0, 'None']
result = self.run_command(['--config=%s' % config_file.name, 'config', 'setup'])
self.assertEqual(result.exit_code, 2)
self.assertIsInstance(result.exception, exceptions.CLIAbort)
@mock.patch('SoftLayer.CLI.environment.Environment.getpass')
@mock.patch('SoftLayer.CLI.environment.Environment.input')
def test_github_1074(self, mocked_input, getpass):
"""Tests to make sure directly using an endpoint works"""
mocked_input.side_effect = ['test-endpoint']
endpoint_url = config.get_endpoint_url(self.env)
self.assertEqual(endpoint_url, 'test-endpoint')
@mock.patch('SoftLayer.CLI.environment.Environment.getpass')
@mock.patch('SoftLayer.CLI.environment.Environment.input')
def test_get_endpoint(self, mocked_input, getpass):
"""Tests to make sure directly using an endpoint works"""
mocked_input.side_effect = ['private', 'custom', 'test.com', 'public', 'test-endpoint']
# private
endpoint_url = config.get_endpoint_url(self.env)
self.assertEqual(endpoint_url, consts.API_PRIVATE_ENDPOINT)
# custom - test.com
endpoint_url = config.get_endpoint_url(self.env)
self.assertEqual(endpoint_url, 'test.com')
# public
endpoint_url = config.get_endpoint_url(self.env)
self.assertEqual(endpoint_url, consts.API_PUBLIC_ENDPOINT)
# test-endpoint
endpoint_url = config.get_endpoint_url(self.env)
self.assertEqual(endpoint_url, 'test-endpoint')
@mock.patch('SoftLayer.CLI.environment.Environment.input')
@mock.patch('SoftLayer.CLI.config.setup.get_sso_url')
@mock.patch('SoftLayer.CLI.config.setup.get_accounts')
@mock.patch('SoftLayer.API.IAMClient.authenticate_with_passcode')
@mock.patch('SoftLayer.API.IAMClient.refresh_iam_token')
@mock.patch('SoftLayer.API.IAMClient.call')
def test_sso_login(self, api_call, token, passcode, get_accounts, get_sso_url, mocked_input):
"""Tests to make sure directly using an endpoint works"""
mocked_input.side_effect = ['n', '123qweasd']
get_sso_url.return_value = "https://test.com/"
get_accounts.return_value = {"account_id": 12345, "ims_id": 5555}
passcode.return_value = {"access_token": "aassddffggh", "refresh_token": "qqqqqqq"}
token.return_value = {"access_token": "zzzzzz", "refresh_token": "fffffff"}
test_key = "zz112233"
user_object_1 = {
"apiAuthenticationKeys": [{"authenticationKey": test_key}],
"username": "testerson",
"id": 99}
api_call.side_effect = [user_object_1]
user, apikey = config.sso_login(self.env)
self.assertEqual("testerson", user)
self.assertEqual(test_key, apikey)
|
