def test_auth_add_again(self, uuid4):
"""Testing the auth add command execution"""
sys.argv = ['/bin/rho', "auth", "add", "--name", "auth_2",
"--username", "user", "--sshkeyfile",
TMP_KEY, "--vault",
TMP_VAULT_PASS]
creds = [{u'id': u'1', u'name': u'auth_1', u'username': u'user',
u'password': u'', u'sudo_password': None,
u'ssh_key_file': u'/tmp/privatekey'}]
with redirect_credentials(creds):
AuthAddCommand().main()
self.assertEqual(creds,
[{u'id': u'1',
u'name': u'auth_1',
u'username': u'user',
u'password': u'',
u'sudo_password': None,
u'ssh_key_file': u'/tmp/privatekey'},
{u'id': u'2',
u'name': u'auth_2',
u'username': u'user',
u'password': None,
u'sudo_password': None,
u'ssh_key_file': u'/tmp/privatekey'}])
def test_a_auth_add_again(self):
"""Testing the auth add command execution"""
sys.argv = [
'/bin/rho', "auth", "add", "--name", "auth2", "--username", "user",
"--sshkeyfile", "./privatekey", "--vault", TMP_VAULT_PASS
]
AuthAddCommand().main()
def test_auth_add_bad_key(self, uuid4):
"""Testing the auth add command execution"""
sys.argv = ['/bin/rho', "auth", "add", "--name", "auth_1",
"--username", "user", "--sshkeyfile",
"/not/a/valid/path", "--vault",
TMP_VAULT_PASS]
auth_add_out = six.StringIO()
with self.assertRaises(SystemExit):
with redirect_stdout(auth_add_out):
AuthAddCommand().main()
self.assertIn("/not/a/valid/path", auth_add_out)