def testCredsFileWithBlankFinalBlock(self): creds1 = credential.load_credentials_file( os.path.join(TESTDATA, 'credentials1.yaml')) creds2 = credential.load_credentials_file( os.path.join(TESTDATA, 'blank_final_group_credentials.yaml')) self.assertEqual(len(creds1), len(creds2)) self.assertEqual(creds1.credentials[0], creds2.credentials[0]) self.assertEqual(creds1.credentials[1], creds2.credentials[1]) self.assertEqual(creds1.credentials[0], creds2.get_credential('ar1.foo')) self.assertEqual(creds1.credentials[1], creds2.get_credential('xr1.foo'))
def testGetCredentialInvalidInputs(self): creds = credential.load_credentials_file( os.path.join(TESTDATA, 'credentials1.yaml')) self.assertRaises(errors.NoMatchingCredentialError, creds.get_credential, '') self.assertRaises(errors.NoMatchingCredentialError, creds.get_credential, None) self.assertRaises(TypeError, creds.get_credential, 5)
def testCredsFile(self): creds = credential.load_credentials_file( os.path.join(TESTDATA, 'credentials1.yaml')) self.assertEqual(len(creds), 2) self.assertEqual(len(creds.credentials), 2) cred0 = creds.credentials[0] cred1 = creds.credentials[1] self.assertEqual(cred0.username, 'fred') self.assertEqual(cred0.regexp_string, '^ar.*$') self.assert_(cred0.connect_method is None) self.assertEqual(cred1.password, 'bar') self.assertEqual(cred1.enable_password, 'enable_bar') self.assertEqual(cred1.connect_method, 'sshv2') self.assert_(cred1.ssh_private_key is None)
def testGetCredential(self): creds = credential.load_credentials_file( os.path.join(TESTDATA, 'credentials1.yaml')) self.assertEqual(creds.credentials[0], creds.get_credential('ar1.foo')) self.assertEqual(creds.credentials[1], creds.get_credential('xr1.foo'))