def test_set_user_ssh_key(self):
keydata = "THIS IS TOTALLY A SSH KEY"
users.create_user("test_user", homedir="/home/test_user", root=self.tmpdir)
users.set_user_ssh_key("test_user", keydata, root=self.tmpdir)
keyfile = self.tmpdir + "/home/test_user/.ssh/authorized_keys"
self.assertTrue(os.path.isfile(keyfile))
with open(keyfile) as f:
output_keydata = f.read()
self.assertEqual(keydata, output_keydata.strip())
def test_set_user_ssh_key(self):
keydata = "THIS IS TOTALLY A SSH KEY"
users.create_user("test_user", homedir="/home/test_user", root=self.tmpdir)
with patch("pyanaconda.core.users.util.restorecon") as restorecon_mock:
users.set_user_ssh_key("test_user", keydata, root=self.tmpdir)
restorecon_mock.assert_called_once_with(
["/home/test_user/.ssh"],
root=self.tmpdir
)
keyfile = self.tmpdir + "/home/test_user/.ssh/authorized_keys"
assert os.path.isfile(keyfile)
with open(keyfile) as f:
output_keydata = f.read()
assert keydata == output_keydata.strip()
def _set_ssh_keys(self):
for key_data in self._ssh_key_data_list:
users.set_user_ssh_key(key_data.username, key_data.key)
