def test_get_auth(self):
auth_lines = [
"""id = a1; type = InfrastructureManager; username = someuser; password = somepass """,
"""id = ''; type = VMRC; username = someuser; password = somepass; """
]
auth = Authentication(Authentication.read_auth_data(auth_lines))
auth_data = auth.getAuthInfoByID("a1")
self.assertEqual(auth_data, [{
'id': 'a1',
'password': "******",
'type': 'InfrastructureManager',
'username': '******'
}])
auth_data = auth.getAuthInfo("VMRC")
self.assertEqual(auth_data, [{
'id': '',
'password': "******",
'type': 'VMRC',
'username': '******'
}])
auth_lines = [
"""id = 1a; type = InfrastructureManager; username = someuser; password = somepass """
]
with self.assertRaises(Exception) as ex:
auth = Authentication(Authentication.read_auth_data(auth_lines))
self.assertEqual("Incorrect value in auth item id: 1a",
str(ex.exception))
def test_get_auth(self):
auth_lines = [
"""id = 1; type = InfrastructureManager; username = someuser; password = somepass """,
"""id = 2; type = VMRC; username = someuser; password = somepass; """
]
auth = Authentication(Authentication.read_auth_data(auth_lines))
auth_data = auth.getAuthInfoByID("1")
self.assertEqual(auth_data, [{
'id': '1',
'password': "******",
'type': 'InfrastructureManager',
'username': '******'
}])
auth_data = auth.getAuthInfo("VMRC")
self.assertEqual(auth_data, [{
'id': '2',
'password': "******",
'type': 'VMRC',
'username': '******'
}])
def test_auth_read(self):
auth_lines = [
"""id = a1; type = InfrastructureManager; username = someuser; password = somepass """,
"""id = a2; type = VMRC; username = someuser; password = somepass; """,
"""id = a3; type = OpenNebula; username = someuser; password = "******" """,
"""id = a4; type = EC2; username = someuser; password = '******' """
]
auth = Authentication.read_auth_data(auth_lines)
self.assertEqual(auth, [{
'id': 'a1',
'password': "******",
'type': 'InfrastructureManager',
'username': '******'
}, {
'id': 'a2',
'password': "******",
'type': 'VMRC',
'username': '******'
}, {
'id': 'a3',
'password': "******",
'type': 'OpenNebula',
'username': '******'
}, {
'id': 'a4',
'password': '******',
'type': 'EC2',
'username': '******'
}])
tests_path = os.path.dirname(os.path.abspath(__file__))
shutil.copyfile(os.path.join(tests_path, "../files/privatekey.pem"),
"/tmp/privatekey.pem")
auth = Authentication(
Authentication.read_auth_data(
os.path.join(tests_path, "../files/auth.dat")))
auth_data = auth.getAuthInfoByID("occi")
self.assertEqual(auth_data[0]['proxy'][:37],
"-----BEGIN RSA PRIVATE KEY-----\nMIIEo")
os.unlink("/tmp/privatekey.pem")
