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'))
