Synnefo » snf-ganeti » ganeti-local

from ganeti import compat

def GetAllUserFiles(user, mkdir=False, dircheck=True, _homedir_fn=None):

  """Wrapper over L{GetUserFiles} to retrieve files for all SSH key types.

  See L{GetUserFiles} for details.

  @rtype: tuple; (string, dict with string as key, tuple of (string, string) as

    value)

  """

  helper = compat.partial(GetUserFiles, user, mkdir=mkdir, dircheck=dircheck,

                          _homedir_fn=_homedir_fn)

  result = [(kind, helper(kind=kind)) for kind in constants.SSHK_ALL]

  authorized_keys = [i for (_, (_, _, i)) in result]

  assert len(frozenset(authorized_keys)) == 1, \

    "Different paths for authorized_keys were returned"

  return (authorized_keys[0],

          dict((kind, (privkey, pubkey))

               for (kind, (privkey, pubkey, _)) in result))

  def testGetAllUserFiles(self):

    result = ssh.GetAllUserFiles("example7475", mkdir=False, dircheck=False,

                                 _homedir_fn=self._GetTempHomedir)

    self.assertEqual(result,

      (os.path.join(self.tmpdir, ".ssh", "authorized_keys"), {

        constants.SSHK_RSA:

          (os.path.join(self.tmpdir, ".ssh", "id_rsa"),

           os.path.join(self.tmpdir, ".ssh", "id_rsa.pub")),

        constants.SSHK_DSA:

          (os.path.join(self.tmpdir, ".ssh", "id_dsa"),

           os.path.join(self.tmpdir, ".ssh", "id_dsa.pub")),

      }))

    self.assertEqual(os.listdir(self.tmpdir), [])

  def testGetAllUserFilesNoDirectoryNoMkdir(self):

    self.assertRaises(errors.OpExecError, ssh.GetAllUserFiles,

                      "example17270", mkdir=False, dircheck=True,

                      _homedir_fn=self._GetTempHomedir)

    self.assertEqual(os.listdir(self.tmpdir), [])