def test_create_table_async(self, mock_table_info_repo):
context = mock.Mock(tenant='fake_tenant')
table_name = 'fake_table'
table_schema = 'fake_table_schema'
mock_storage_driver = mock.Mock()
mock_storage_driver.create_table.return_value = "fake_internal_name"
table_info_save_args_list = []
def side_effect(*args):
table_info_save_args_list.append(copy.deepcopy(args))
mock_table_info_repo.save.side_effect = side_effect
storage_manager = async_simple_impl.AsyncSimpleStorageManager(
mock_storage_driver, mock_table_info_repo)
storage_manager.create_table(context, table_name, table_schema)
# called once, length of call_args_list indicates number of calls
self.assertEqual(1, len(table_info_save_args_list))
# CallList is tuple of Mock and TableInfo
self.assertEqual(2, len(table_info_save_args_list[0]))
# TableInfo status should be creating initially when
# table_info_repo.save is called
self.assertEqual(models.TableMeta.TABLE_STATUS_CREATING,
table_info_save_args_list[0][1].status)
# wait for async create table call to finish
for i in range(10):
if mock_table_info_repo.update.called:
table_info_update_args_list = (
mock_table_info_repo.update.call_args_list)
# called once
# length of call_args_list indicates number of calls
self.assertEqual(1, len(table_info_update_args_list))
self.assertEqual(2, len(table_info_update_args_list[0]))
# tuple of Mock, TableInfo, and status list
self.assertEqual(3, len(table_info_update_args_list[0][0]))
# TableInfo status should be active by now
self.assertEqual(models.TableMeta.TABLE_STATUS_ACTIVE,
table_info_update_args_list[0][0][1].status)
break
else:
time.sleep(1)
# create_table method has been called
self.assertTrue(mock_storage_driver.create_table.called)
def test_notify_delete_table_async(self, mock_table_info_repo):
self.cleanup_test_notifier()
context = mock.Mock(tenant='fake_tenant')
table_name = 'fake_table'
mock_storage_driver = mock.Mock()
mock_storage_driver.delete_table.return_value = True
class FakeTableInfo:
status = models.TableMeta.TABLE_STATUS_ACTIVE
name = table_name
in_use = False
id = None
schema = None
creation_date_time = None
internal_name = None
mock_table_info_repo = mock.Mock()
mock_table_info_repo.get.return_value = FakeTableInfo()
storage_manager = async_simple_impl.AsyncSimpleStorageManager(
mock_storage_driver, mock_table_info_repo)
storage_manager.delete_table(context, table_name)
# wait for async delete table call to finish
for i in range(10):
if (mock_table_info_repo.delete.called
and len(self.get_notifications()) == 1):
# delete_table method of mock_storage_driver has been called
break
else:
time.sleep(1)
else:
self.fail("Couldn't wait for async request completion")
# delete method of mock_table_info_repo has been called
self.assertTrue(mock_table_info_repo.delete.called)
# check notification queue
self.assertEqual(len(self.get_notifications()), 1)
start_event = self.get_notifications()[0]
self.assertEqual(start_event['priority'], 'INFO')
self.assertEqual(start_event['event_type'],
notifier.EVENT_TYPE_TABLE_DELETE)
self.assertEqual(start_event['payload']['table_name'], table_name)
def test_delete_table_async(self, mock_table_info_repo):
context = mock.Mock(tenant='fake_tenant')
table_name = 'fake_table'
mock_storage_driver = mock.Mock()
mock_storage_driver.delete_table.return_value = True
class FakeTableInfo:
status = models.TableMeta.TABLE_STATUS_ACTIVE
name = table_name
in_use = False
id = None
schema = None
creation_date_time = None
internal_name = 'fake'
mock_table_info_repo = mock.Mock()
mock_table_info_repo.get.return_value = FakeTableInfo()
storage_manager = async_simple_impl.AsyncSimpleStorageManager(
mock_storage_driver, mock_table_info_repo)
storage_manager.delete_table(context, table_name)
table_info_update_args_list = (
mock_table_info_repo.update.call_args_list)
# called once, length of call_args_list indicates number of calls
self.assertEqual(1, len(table_info_update_args_list))
self.assertEqual(2, len(table_info_update_args_list[0]))
# tuple of Mock, TableInfo, and status list
self.assertEqual(3, len(table_info_update_args_list[0][0]))
# TableInfo status should be deleting
self.assertEqual(models.TableMeta.TABLE_STATUS_DELETING,
table_info_update_args_list[0][0][1].status)
for i in range(10):
if mock_table_info_repo.delete.called:
# delete_table method of mock_storage_driver has been called
self.assertTrue(mock_storage_driver.delete_table.called)
break
else:
time.sleep(1)
# delete method of mock_table_info_repo has been called
self.assertTrue(mock_table_info_repo.delete.called)
def test_notify_create_table_async(self, mock_table_info_repo):
self.cleanup_test_notifier()
context = mock.Mock(tenant='fake_tenant')
table_name = 'fake_table'
table_schema = 'fake_table_schema'
mock_storage_driver = mock.Mock()
mock_storage_driver.create_table.return_value = True
storage_manager = async_simple_impl.AsyncSimpleStorageManager(
mock_storage_driver, mock_table_info_repo)
storage_manager.create_table(context, table_name, table_schema)
# wait for async create table call to finish
for i in range(10):
if (mock_table_info_repo.update.called
and len(self.get_notifications()) == 1):
break
else:
time.sleep(1)
else:
self.fail("Couldn't wait for async request completion")
# create_table method has been called
self.assertTrue(mock_storage_driver.create_table.called)
# check notification queue
self.assertEqual(len(self.get_notifications()), 1)
end_event = self.get_notifications()[0]
self.assertEqual(end_event['priority'], 'INFO')
self.assertEqual(end_event['event_type'],
notifier.EVENT_TYPE_TABLE_CREATE)
self.assertEqual(end_event['payload']['schema'], table_schema)
def test_delete_table_create_or_delete_failed(self, mock_table_info_repo):
context = mock.Mock(tenant='fake_tenant')
table_name = 'fake_table'
mock_storage_driver = mock.Mock()
mock_storage_driver.delete_table.return_value = True
class FakeTableInfoCreateFailed:
status = models.TableMeta.TABLE_STATUS_CREATE_FAILED
name = table_name
in_use = False
id = None
schema = None
creation_date_time = None
internal_name = None
mock_table_info_repo = mock.Mock()
mock_table_info_repo.get.return_value = FakeTableInfoCreateFailed()
storage_manager = async_simple_impl.AsyncSimpleStorageManager(
mock_storage_driver, mock_table_info_repo)
storage_manager.delete_table(context, table_name)
table_info_update_args_list = (
mock_table_info_repo.update.call_args_list)
# called once, length of call_args_list indicates number of calls
self.assertEqual(1, len(table_info_update_args_list))
self.assertEqual(2, len(table_info_update_args_list[0]))
# tuple of Mock, TableInfo, and status list
self.assertEqual(3, len(table_info_update_args_list[0][0]))
# TableInfo status should be deleting
self.assertEqual(models.TableMeta.TABLE_STATUS_DELETING,
table_info_update_args_list[0][0][1].status)
for i in range(10):
if mock_table_info_repo.delete.called:
# delete_table method of mock_storage_driver should not
# have been called since table does not actually exist
self.assertFalse(mock_storage_driver.delete_table.called)
break
else:
time.sleep(1)
# delete method of mock_table_info_repo has been called
self.assertTrue(mock_table_info_repo.delete.called)
class FakeTableInfoDeleteFailedTableNotExist:
status = models.TableMeta.TABLE_STATUS_DELETE_FAILED
name = table_name
in_use = False
id = None
schema = None
creation_date_time = None
internal_name = None
mock_table_info_repo = mock.Mock()
mock_table_info_repo.get.return_value = (
FakeTableInfoDeleteFailedTableNotExist())
storage_manager = async_simple_impl.AsyncSimpleStorageManager(
mock_storage_driver, mock_table_info_repo)
storage_manager.delete_table(context, table_name)
table_info_update_args_list = (
mock_table_info_repo.update.call_args_list)
# called once, length of call_args_list indicates number of calls
self.assertEqual(1, len(table_info_update_args_list))
self.assertEqual(2, len(table_info_update_args_list[0]))
# tuple of Mock, TableInfo, and status list
self.assertEqual(3, len(table_info_update_args_list[0][0]))
# TableInfo status should be deleting
self.assertEqual(models.TableMeta.TABLE_STATUS_DELETING,
table_info_update_args_list[0][0][1].status)
for i in range(10):
if mock_table_info_repo.delete.called:
# delete_table method of mock_storage_driver should not
# have been called since table does not actually exist
self.assertFalse(mock_storage_driver.delete_table.called)
break
else:
time.sleep(1)
# delete method of mock_table_info_repo has been called
self.assertTrue(mock_table_info_repo.delete.called)
class FakeTableInfoDeleteFailedButTableExist:
status = models.TableMeta.TABLE_STATUS_DELETE_FAILED
name = table_name
in_use = False
id = None
schema = None
creation_date_time = None
internal_name = 'fake'
mock_table_info_repo = mock.Mock()
mock_table_info_repo.get.return_value = (
FakeTableInfoDeleteFailedButTableExist())
storage_manager = async_simple_impl.AsyncSimpleStorageManager(
mock_storage_driver, mock_table_info_repo)
storage_manager.delete_table(context, table_name)
table_info_update_args_list = (
mock_table_info_repo.update.call_args_list)
# called once, length of call_args_list indicates number of calls
self.assertEqual(1, len(table_info_update_args_list))
self.assertEqual(2, len(table_info_update_args_list[0]))
# tuple of Mock, TableInfo, and status list
self.assertEqual(3, len(table_info_update_args_list[0][0]))
# TableInfo status should be deleting
self.assertEqual(models.TableMeta.TABLE_STATUS_DELETING,
table_info_update_args_list[0][0][1].status)
for i in range(10):
if mock_table_info_repo.delete.called:
# delete_table method of mock_storage_driver should
# have been called because table exists
self.assertTrue(mock_storage_driver.delete_table.called)
break
else:
time.sleep(1)
# delete method of mock_table_info_repo has been called
self.assertTrue(mock_table_info_repo.delete.called)