def test_remove_tenants(self):
vms = [(str(uuid.uuid4()), 'vm1')]
privileges = self.get_privileges()
default_datastore, default_privileges = self.get_default_datastore_and_privileges()
error_info, tenant1 = self.auth_mgr.create_tenant('tenant1', 'Some tenant', default_datastore,
default_privileges, vms, privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
vms = [(str(uuid.uuid4()), 'vm2'), (str(uuid.uuid4()), 'vm3')]
privileges = []
error_info, tenant2 = self.auth_mgr.create_tenant('tenant2', 'Some tenant', default_datastore,
default_privileges, vms, privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant2.id))
tenant2.set_datastore_access_privileges(self.auth_mgr.conn, privileges)
error_info = self.auth_mgr.remove_tenant(tenant2.id, False)
self.assertEqual(error_info, None)
# Check tenants table
error_info, tenants_row = auth.get_row_from_tenants_table(self.auth_mgr.conn, tenant2.id)
self.assertEqual(error_info, None)
self.assertEqual(tenants_row, None)
# check vms table
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn, tenant2.id)
self.assertEqual(error_info, None)
self.assertEqual(vms_row, [])
# check privileges table
error_info, privileges_row = auth.get_row_from_privileges_table(self.auth_mgr.conn, tenant2.id)
self.assertEqual(error_info, None)
self.assertEqual(privileges_row, [])
def test_add_vms(self):
""" Test add_vms() API """
vms = []
privileges = []
default_datastore, default_privileges = self.get_default_datastore_and_privileges()
error_info, tenant1 = self.auth_mgr.create_tenant('tenant1', 'Some tenant', default_datastore,
default_privileges, vms, privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
vm1_uuid = str(uuid.uuid4())
vm2_uuid = str(uuid.uuid4())
vms = [(vm1_uuid, 'vm1'), (vm2_uuid, 'vm2')]
error_info = tenant1.add_vms(self.auth_mgr.conn, vms)
self.assertEqual(error_info, None)
# check vms table
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn,tenant1.id)
self.assertEqual(error_info, None)
expected_output = [(vm1_uuid, 'vm1', tenant1.id),
(vm2_uuid, 'vm2', tenant1.id)
]
self.assertEqual(len(vms_row), 2)
actual_output = [(vms_row[0][auth_data_const.COL_VM_ID],
vms_row[0][auth_data_const.COL_VM_NAME],
vms_row[0][auth_data_const.COL_TENANT_ID]),
(vms_row[1][auth_data_const.COL_VM_ID],
vms_row[1][auth_data_const.COL_VM_NAME],
vms_row[1][auth_data_const.COL_TENANT_ID]),
]
self.assertEqual(actual_output, expected_output)
def test_add_vms(self):
""" Test add_vms() API """
vms = []
privileges = []
default_datastore = self.get_default_datastore()
default_datastore_url = self.get_datastore_url(default_datastore)
error_info, tenant1 = self.auth_mgr.create_tenant(name=self.tenant_name,
description='Some tenant',
vms=vms,
privileges=privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
vms = [(self.vm1_uuid, self.vm1_name), (self.vm2_uuid, self.vm2_name)]
error_info = tenant1.add_vms(self.auth_mgr.conn, vms)
self.assertEqual(error_info, None)
# check vms table
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn,tenant1.id)
self.assertEqual(error_info, None)
expected_output = [(self.vm1_uuid, tenant1.id),
(self.vm2_uuid, tenant1.id)
]
self.assertEqual(len(vms_row), 2)
actual_output = [(vms_row[0][auth_data_const.COL_VM_ID],
vms_row[0][auth_data_const.COL_TENANT_ID]),
(vms_row[1][auth_data_const.COL_VM_ID],
vms_row[1][auth_data_const.COL_TENANT_ID]),
]
self.assertEqual(actual_output, expected_output)
def test_remove_vms(self):
"""
"""
privileges = self.get_privileges()
default_datastore, default_privileges = self.get_default_datastore_and_privileges()
vm1_uuid = str(uuid.uuid4())
vm2_uuid = str(uuid.uuid4())
vms = [(vm1_uuid, 'vm1'), (vm2_uuid, 'vm2')]
error_info, tenant1 = self.auth_mgr.create_tenant('tenant1', 'Some tenant', default_datastore,
default_privileges, vms, privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
error_info = tenant1.remove_vms(self.auth_mgr.conn, vms)
self.assertEqual(error_info, None)
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
self.assertEqual(vms_row, [])
def test_remove_tenants(self):
vms = [(self.vm1_uuid, self.vm1_name)]
privileges = self.get_privileges()
default_datastore = self.get_default_datastore()
default_datastore_url = self.get_datastore_url(default_datastore)
error_info, tenant1 = self.auth_mgr.create_tenant(name=self.tenant_name,
description='Some tenant',
vms=vms,
privileges=privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
vms = [(self.vm2_uuid, self.vm2_name), (self.vm3_uuid, self.vm3_name)]
privileges = []
error_info, tenant2 = self.auth_mgr.create_tenant(name=self.tenant_2_name,
description='Some tenant',
vms=vms,
privileges=privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant2.id))
tenant2.set_datastore_access_privileges(self.auth_mgr.conn, privileges)
error_info = self.auth_mgr.remove_tenant(tenant2.id, False)
self.assertEqual(error_info, None)
# Check tenants table
error_info, tenants_row = auth.get_row_from_tenants_table(self.auth_mgr.conn, tenant2.id)
self.assertEqual(error_info, None)
self.assertEqual(tenants_row, None)
# check vms table
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn, tenant2.id)
self.assertEqual(error_info, None)
self.assertEqual(vms_row, [])
# check privileges table
error_info, privileges_row = auth.get_row_from_privileges_table(self.auth_mgr.conn, tenant2.id)
self.assertEqual(error_info, None)
self.assertEqual(privileges_row, [])
def test_remove_vms(self):
"""
"""
privileges = self.get_privileges()
default_datastore = self.get_default_datastore()
default_datastore_url = self.get_datastore_url(default_datastore)
vms = [(self.vm1_uuid, self.vm1_name), (self.vm2_uuid, self.vm2_name)]
error_info, tenant1 = self.auth_mgr.create_tenant(name=self.tenant_name,
description='Some tenant',
vms=vms,
privileges=privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
error_info = tenant1.remove_vms(self.auth_mgr.conn, vms)
self.assertEqual(error_info, None)
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
self.assertEqual(vms_row, [])
def test_create_tenant(self):
""" Test create_tenant() API """
vm1_uuid = str(uuid.uuid4())
vms = [(vm1_uuid, 'vm1')]
#privileges = []
privileges = self.get_privileges()
default_datastore, default_privileges = self.get_default_datastore_and_privileges()
error_info, tenant1 = self.auth_mgr.create_tenant('tenant1', 'Some tenant', default_datastore,
default_privileges, vms, privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
# Check tenants table
error_info, tenants_row = auth.get_row_from_tenants_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
expected_output = [tenant1.id,
'tenant1',
'Some tenant',
'default_ds']
actual_output = [tenants_row[auth_data_const.COL_ID],
tenants_row[auth_data_const.COL_NAME],
tenants_row[auth_data_const.COL_DESCRIPTION],
tenants_row[auth_data_const.COL_DEFAULT_DATASTORE]
]
self.assertEqual(actual_output, expected_output)
# check vms table
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
expected_output = [vm1_uuid,
'vm1',
tenant1.id]
self.assertEqual(len(vms_row), 1)
actual_output = [vms_row[0][auth_data_const.COL_VM_ID],
vms_row[0][auth_data_const.COL_VM_NAME],
vms_row[0][auth_data_const.COL_TENANT_ID]
]
self.assertEqual(actual_output, expected_output)
# check privileges table
error_info, privileges_row = auth.get_row_from_privileges_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
self.assertEqual(len(privileges_row), 2)
expected_privileges = [tenant1.id,
privileges[0][auth_data_const.COL_DATASTORE],
privileges[0][auth_data_const.COL_GLOBAL_VISIBILITY],
privileges[0][auth_data_const.COL_CREATE_VOLUME],
privileges[0][auth_data_const.COL_DELETE_VOLUME],
privileges[0][auth_data_const.COL_MOUNT_VOLUME],
privileges[0][auth_data_const.COL_MAX_VOLUME_SIZE],
privileges[0][auth_data_const.COL_USAGE_QUOTA]
]
expected_default_privileges = [tenant1.id,
default_privileges[auth_data_const.COL_DATASTORE],
default_privileges[auth_data_const.COL_GLOBAL_VISIBILITY],
default_privileges[auth_data_const.COL_CREATE_VOLUME],
default_privileges[auth_data_const.COL_DELETE_VOLUME],
default_privileges[auth_data_const.COL_MOUNT_VOLUME],
default_privileges[auth_data_const.COL_MAX_VOLUME_SIZE],
default_privileges[auth_data_const.COL_USAGE_QUOTA]
]
expected_output = [expected_privileges,
expected_default_privileges
]
actual_privileges = [privileges_row[0][auth_data_const.COL_TENANT_ID],
privileges_row[0][auth_data_const.COL_DATASTORE],
privileges_row[0][auth_data_const.COL_GLOBAL_VISIBILITY],
privileges_row[0][auth_data_const.COL_CREATE_VOLUME],
privileges_row[0][auth_data_const.COL_DELETE_VOLUME],
privileges_row[0][auth_data_const.COL_MOUNT_VOLUME],
privileges_row[0][auth_data_const.COL_MAX_VOLUME_SIZE],
privileges_row[0][auth_data_const.COL_USAGE_QUOTA]
]
actual_default_privileges = [privileges_row[1][auth_data_const.COL_TENANT_ID],
privileges_row[1][auth_data_const.COL_DATASTORE],
privileges_row[1][auth_data_const.COL_GLOBAL_VISIBILITY],
privileges_row[1][auth_data_const.COL_CREATE_VOLUME],
privileges_row[1][auth_data_const.COL_DELETE_VOLUME],
privileges_row[1][auth_data_const.COL_MOUNT_VOLUME],
privileges_row[1][auth_data_const.COL_MAX_VOLUME_SIZE],
privileges_row[1][auth_data_const.COL_USAGE_QUOTA]
]
actual_output = [actual_privileges, actual_default_privileges]
self.assertEqual(actual_output, expected_output)
def test_create_tenant(self):
""" Test create_tenant() API """
vm1_uuid = str(uuid.uuid4())
vms = [(vm1_uuid)]
privileges = self.get_privileges()
default_datastore = self.get_default_datastore()
default_datastore_url = self.get_datastore_url(default_datastore)
error_info, tenant1 = self.auth_mgr.create_tenant(name='tenant1',
description='Some tenant',
vms=vms,
privileges=privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
# Check tenants table
error_info, tenants_row = auth.get_row_from_tenants_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
expected_output = [tenant1.id,
'tenant1',
'Some tenant']
actual_output = [tenants_row[auth_data_const.COL_ID],
tenants_row[auth_data_const.COL_NAME],
tenants_row[auth_data_const.COL_DESCRIPTION]
]
self.assertEqual(actual_output, expected_output)
# check vms table
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
expected_output = [vm1_uuid,
tenant1.id]
self.assertEqual(len(vms_row), 1)
actual_output = [vms_row[0][auth_data_const.COL_VM_ID],
vms_row[0][auth_data_const.COL_TENANT_ID]
]
self.assertEqual(actual_output, expected_output)
# check privileges table
error_info, privileges_row = auth.get_row_from_privileges_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
self.assertEqual(len(privileges_row), 1)
expected_privileges = [tenant1.id,
privileges[0][auth_data_const.COL_DATASTORE_URL],
privileges[0][auth_data_const.COL_ALLOW_CREATE],
privileges[0][auth_data_const.COL_MAX_VOLUME_SIZE],
privileges[0][auth_data_const.COL_USAGE_QUOTA]
]
expected_output = [expected_privileges
]
actual_privileges = [privileges_row[0][auth_data_const.COL_TENANT_ID],
privileges_row[0][auth_data_const.COL_DATASTORE_URL],
privileges_row[0][auth_data_const.COL_ALLOW_CREATE],
privileges_row[0][auth_data_const.COL_MAX_VOLUME_SIZE],
privileges_row[0][auth_data_const.COL_USAGE_QUOTA]
]
actual_output = [actual_privileges]
self.assertEqual(actual_output, expected_output)
def test_create_tenant(self):
""" Test create_tenant() API """
vms = [(self.vm1_uuid, self.vm1_name)]
privileges = self.get_privileges()
default_datastore = self.get_default_datastore()
default_datastore_url = self.get_datastore_url(default_datastore)
error_info, tenant1 = self.auth_mgr.create_tenant(name=self.tenant_name,
description='Some tenant',
vms=vms,
privileges=privileges)
self.assertEqual(error_info, None)
self.assertTrue(uuid.UUID(tenant1.id))
# Check tenants table
error_info, tenants_row = auth.get_row_from_tenants_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
expected_output = [tenant1.id,
self.tenant_name,
'Some tenant']
actual_output = [tenants_row[auth_data_const.COL_ID],
tenants_row[auth_data_const.COL_NAME],
tenants_row[auth_data_const.COL_DESCRIPTION]
]
self.assertEqual(actual_output, expected_output)
# check vms table
error_info, vms_row = auth.get_row_from_vms_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
expected_output = [self.vm1_uuid,
tenant1.id]
self.assertEqual(len(vms_row), 1)
actual_output = [vms_row[0][auth_data_const.COL_VM_ID],
vms_row[0][auth_data_const.COL_TENANT_ID]
]
self.assertEqual(actual_output, expected_output)
# check privileges table
error_info, privileges_row = auth.get_row_from_privileges_table(self.auth_mgr.conn, tenant1.id)
self.assertEqual(error_info, None)
self.assertEqual(len(privileges_row), 1)
expected_privileges = [tenant1.id,
privileges[0][auth_data_const.COL_DATASTORE_URL],
privileges[0][auth_data_const.COL_ALLOW_CREATE],
privileges[0][auth_data_const.COL_MAX_VOLUME_SIZE],
privileges[0][auth_data_const.COL_USAGE_QUOTA]
]
expected_output = [expected_privileges
]
actual_privileges = [privileges_row[0][auth_data_const.COL_TENANT_ID],
privileges_row[0][auth_data_const.COL_DATASTORE_URL],
privileges_row[0][auth_data_const.COL_ALLOW_CREATE],
privileges_row[0][auth_data_const.COL_MAX_VOLUME_SIZE],
privileges_row[0][auth_data_const.COL_USAGE_QUOTA]
]
actual_output = [actual_privileges]
self.assertEqual(actual_output, expected_output)