def poll_for_table_creation(ledger_name):
driver = QldbDriver(ledger_name)
max_poll_time = time() + 15
while True:
tables = driver.list_tables()
count = len(list(tables))
if count == 4 or time() > max_poll_time:
break
sleep(3)
def test_list_tables(self, mock_execute_lambda, mock_client):
mock_client.return_value = mock_client
mock_execute_lambda.return_value = MOCK_LIST_TABLES_RESULT
driver = QldbDriver(MOCK_LEDGER_NAME)
table_names = driver.list_tables()
count = 0
for result in table_names:
self.assertEqual(result, MOCK_LIST_TABLES_RESULT[count])
count += 1
