def test_create_tables_failed(client, current_call, refresh_call, base_scan_schema, base_scan_partition, tmpdir):
client.return_value.__enter__.return_value.get_table.side_effect = NoSuchObjectException()
current_call.return_value = None
conf = {"hive.metastore.uris": 'thrift://hms:port',
"hive.metastore.warehouse.dir": tmpdir}
tables = HiveTables(conf)
with pytest.raises(AlreadyExistsException):
tables.create(base_scan_schema, "test.test_123", base_scan_partition)
assert len(os.listdir(os.path.join(tmpdir, "test.db", "test_123", "metadata"))) == 0
def test_create_tables(client, current_call, base_scan_schema, base_scan_partition, tmpdir):
client.return_value.__enter__.return_value.get_table.side_effect = NoSuchObjectException()
current_call.return_value = None
client.return_value.__enter__.return_value.lock.return_value = LockResponse("x", LockState.WAITING)
client.return_value.__enter__.return_value.check_lock.return_value = LockResponse("x", LockState.ACQUIRED)
tbl = client.return_value.__enter__.return_value.create_table.call_args_list
conf = {"hive.metastore.uris": 'thrift://hms:port',
"hive.metastore.warehouse.dir": tmpdir}
tables = HiveTables(conf)
tables.create(base_scan_schema, "test.test_123", base_scan_partition)
client.return_value.__enter__.return_value.get_table.return_value = MockHMSTable(tbl[0].args[0].parameters)
client.return_value.__enter__.return_value.get_table.side_effect = None
current_call.return_value = tbl[0].args[0].parameters['metadata_location']
tables.load("test.test_123")
