def test_edit_host_cred(self):
"""Testing the edit credential command successfully."""
cred_out = StringIO()
url_get = BASE_URL + CREDENTIAL_URI
url_patch = BASE_URL + CREDENTIAL_URI + '1/'
data = [{
'id': 1,
'name': 'cred1',
'cred_type': NETWORK_CRED_TYPE,
'username': '******',
'password': '******'
}]
with requests_mock.Mocker() as mocker:
mocker.get(url_get, status_code=200, json=data)
mocker.patch(url_patch, status_code=200)
aec = CredEditCommand(SUBPARSER)
args = Namespace(name='cred1',
username='******',
filename=TMP_KEY,
password=None,
sudo_password=None,
ssh_passphrase=None)
with redirect_stdout(cred_out):
aec.main(args)
self.assertEqual(cred_out.getvalue(),
'Credential "cred1" was updated\n')
def test_edit_cred_ssl_err(self):
"""Testing the edit credential command with a connection error."""
cred_out = StringIO()
url = BASE_URL + CREDENTIAL_URI
with requests_mock.Mocker() as mocker:
mocker.get(url, exc=requests.exceptions.SSLError)
aec = CredEditCommand(SUBPARSER)
args = Namespace(name='credential1',
username='******',
filename=TMP_KEY,
password=None,
sudo_password=None)
with self.assertRaises(SystemExit):
with redirect_stdout(cred_out):
aec.main(args)
self.assertEqual(cred_out.getvalue(), SSL_ERROR_MSG)