def test_generate_specfied_ssh_key_files(self):
temp_dir_name = tempfile.mkdtemp(prefix="ssh_dir_")
# cleanup temporary directory and its contents
self.addCleanup(shutil.rmtree, path=temp_dir_name)
# first create file paths for the keys to be generated
fd, private_key_file = tempfile.mkstemp(dir=temp_dir_name)
os.close(fd)
public_key_file = private_key_file + '.pub'
os.remove(private_key_file)
args = mock.MagicMock()
args.ssh_key_value = public_key_file
args.generate_ssh_keys = True
# 1 verify we generate key files if not existing
validate_ssh_key(args)
generated_public_key_string = args.ssh_key_value
self.assertTrue(bool(args.ssh_key_value))
self.assertTrue(is_valid_ssh_rsa_public_key(generated_public_key_string))
self.assertTrue(os.path.isfile(private_key_file))
# 2 verify we load existing key files
# for convinience we will reuse the generated file in the previous step
args2 = mock.MagicMock()
args2.ssh_key_value = generated_public_key_string
args2.generate_ssh_keys = False
validate_ssh_key(args2)
# we didn't regenerate
self.assertEqual(generated_public_key_string, args.ssh_key_value)
# 3 verify we do not generate unless told so
fd, private_key_file2 = tempfile.mkstemp(dir=temp_dir_name)
os.close(fd)
public_key_file2 = private_key_file2 + '.pub'
args3 = mock.MagicMock()
args3.ssh_key_value = public_key_file2
args3.generate_ssh_keys = False
with self.assertRaises(CLIError):
validate_ssh_key(args3)
# 4 verify file naming if the pub file doesn't end with .pub
fd, public_key_file4 = tempfile.mkstemp(dir=temp_dir_name)
os.close(fd)
public_key_file4 += '1' # make it nonexisting
args4 = mock.MagicMock()
args4.ssh_key_value = public_key_file4
args4.generate_ssh_keys = True
validate_ssh_key(args4)
self.assertTrue(os.path.isfile(public_key_file4 + '.private'))
self.assertTrue(os.path.isfile(public_key_file4))
def test_generate_specfied_ssh_key_files(self):
temp_dir_name = tempfile.mkdtemp(prefix="ssh_dir_")
# cleanup temporary directory and its contents
self.addCleanup(shutil.rmtree, path=temp_dir_name)
# first create file paths for the keys to be generated
_, private_key_file = tempfile.mkstemp(dir=temp_dir_name)
public_key_file = private_key_file + '.pub'
os.remove(private_key_file)
args = mock.MagicMock()
args.ssh_key_value = public_key_file
args.generate_ssh_keys = True
# 1 verify we generate key files if not existing
validate_ssh_key(args)
generated_public_key_string = args.ssh_key_value
self.assertTrue(bool(args.ssh_key_value))
self.assertTrue(
is_valid_ssh_rsa_public_key(generated_public_key_string))
self.assertTrue(os.path.isfile(private_key_file))
# 2 verify we load existing key files
# for convinience we will reuse the generated file in the previous step
args2 = mock.MagicMock()
args2.ssh_key_value = generated_public_key_string
args2.generate_ssh_keys = False
validate_ssh_key(args2)
# we didn't regenerate
self.assertEqual(generated_public_key_string, args.ssh_key_value)
# 3 verify we do not generate unless told so
_, private_key_file2 = tempfile.mkstemp(dir=temp_dir_name)
public_key_file2 = private_key_file2 + '.pub'
args3 = mock.MagicMock()
args3.ssh_key_value = public_key_file2
args3.generate_ssh_keys = False
with self.assertRaises(CLIError):
validate_ssh_key(args3)
# 4 verify file naming if the pub file doesn't end with .pub
_, public_key_file4 = tempfile.mkstemp(dir=temp_dir_name)
public_key_file4 += '1' # make it nonexisting
args4 = mock.MagicMock()
args4.ssh_key_value = public_key_file4
args4.generate_ssh_keys = True
validate_ssh_key(args4)
self.assertTrue(os.path.isfile(public_key_file4 + '.private'))
self.assertTrue(os.path.isfile(public_key_file4))
def test_generate_specfied_ssh_key_files(self):
_, private_key_file = tempfile.mkstemp()
public_key_file = private_key_file + '.pub'
args = mock.MagicMock()
args.ssh_key_value = public_key_file
args.generate_ssh_keys = True
# 1 verify we generate key files if not existing
validate_ssh_key(args)
generated_public_key_string = args.ssh_key_value
self.assertTrue(bool(args.ssh_key_value))
self.assertTrue(
_is_valid_ssh_rsa_public_key(generated_public_key_string))
self.assertTrue(os.path.isfile(private_key_file))
# 2 verify we load existing key files
# for convinience we will reuse the generated file in the previous step
args2 = mock.MagicMock()
args2.ssh_key_value = generated_public_key_string
args2.generate_ssh_keys = False
validate_ssh_key(args2)
# we didn't regenerate
self.assertEqual(generated_public_key_string, args.ssh_key_value)
# 3 verify we do not generate unless told so
_, private_key_file2 = tempfile.mkstemp()
public_key_file2 = private_key_file2 + '.pub'
args3 = mock.MagicMock()
args3.ssh_key_value = public_key_file2
args3.generate_ssh_keys = False
with self.assertRaises(CLIError):
validate_ssh_key(args3)
# 4 verify file naming if the pub file doesn't end with .pub
_, public_key_file4 = tempfile.mkstemp()
public_key_file4 += '1' # make it nonexisting
args4 = mock.MagicMock()
args4.ssh_key_value = public_key_file4
args4.generate_ssh_keys = True
validate_ssh_key(args4)
self.assertTrue(os.path.isfile(public_key_file4 + '.private'))
self.assertTrue(os.path.isfile(public_key_file4))
def test_generate_specfied_ssh_key_files(self):
_, private_key_file = tempfile.mkstemp()
public_key_file = private_key_file + '.pub'
args = mock.MagicMock()
args.ssh_key_value = public_key_file
args.generate_ssh_keys = True
# 1 verify we generate key files if not existing
validate_ssh_key(args)
generated_public_key_string = args.ssh_key_value
self.assertTrue(bool(args.ssh_key_value))
self.assertTrue(is_valid_ssh_rsa_public_key(generated_public_key_string))
self.assertTrue(os.path.isfile(private_key_file))
# 2 verify we load existing key files
# for convinience we will reuse the generated file in the previous step
args2 = mock.MagicMock()
args2.ssh_key_value = generated_public_key_string
args2.generate_ssh_keys = False
validate_ssh_key(args2)
# we didn't regenerate
self.assertEqual(generated_public_key_string, args.ssh_key_value)
# 3 verify we do not generate unless told so
_, private_key_file2 = tempfile.mkstemp()
public_key_file2 = private_key_file2 + '.pub'
args3 = mock.MagicMock()
args3.ssh_key_value = public_key_file2
args3.generate_ssh_keys = False
with self.assertRaises(CLIError):
validate_ssh_key(args3)
# 4 verify file naming if the pub file doesn't end with .pub
_, public_key_file4 = tempfile.mkstemp()
public_key_file4 += '1' # make it nonexisting
args4 = mock.MagicMock()
args4.ssh_key_value = public_key_file4
args4.generate_ssh_keys = True
validate_ssh_key(args4)
self.assertTrue(os.path.isfile(public_key_file4 + '.private'))
self.assertTrue(os.path.isfile(public_key_file4))