Skip to content

Commit c589fe6

Browse files
committed
fixed catalog loading
1 parent d2878e6 commit c589fe6

File tree

3 files changed

+17
-7
lines changed

3 files changed

+17
-7
lines changed

client-spark/src/test/java/io/whitefox/api/client/ITDeltaSharingClient.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ void showS3IcebergTable1withQueryTableApi() {
4444
storageManagerInitializer.createIcebergTableWithGlueMetastore();
4545
var ds = spark.read().format("deltaSharing").load(icebergTablePath);
4646
var expectedSchema = new StructType(new StructField[] {
47-
new StructField("id", DataType.fromDDL("long"), true, new Metadata(emptyScalaMap()))
47+
new StructField("id", DataType.fromDDL("long"), false, new Metadata(emptyScalaMap()))
4848
});
4949
var expectedData = spark
5050
.createDataFrame(
@@ -70,6 +70,8 @@ void showS3Table1withQueryTableApi() {
7070
var expectedSchema = new StructType(new StructField[] {
7171
new StructField("id", DataType.fromDDL("long"), true, new Metadata(emptyScalaMap()))
7272
});
73+
74+
ds.show();
7375
var expectedData = spark
7476
.createDataFrame(
7577
List.of(

client-spark/src/test/java/io/whitefox/api/utils/S3TestConfig.java

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ public class S3TestConfig {
44
private final String region;
55
private final String accessKey;
66
private final String secretKey;
7+
private final String glueCatalogId;
78

89
public String getRegion() {
910
return region;
@@ -17,16 +18,22 @@ public String getSecretKey() {
1718
return secretKey;
1819
}
1920

20-
public S3TestConfig(String region, String accessKey, String secretKey) {
21+
public String getGlueCatalogId() {
22+
return glueCatalogId;
23+
}
24+
25+
public S3TestConfig(String region, String accessKey, String secretKey, String glueCatalogId) {
2126
this.region = region;
2227
this.accessKey = accessKey;
2328
this.secretKey = secretKey;
29+
this.glueCatalogId = glueCatalogId;
2430
}
2531

2632
public static S3TestConfig loadFromEnv() {
2733
return new S3TestConfig(
2834
System.getenv().get("WHITEFOX_TEST_AWS_REGION"),
2935
System.getenv().get("WHITEFOX_TEST_AWS_ACCESS_KEY_ID"),
30-
System.getenv().get("WHITEFOX_TEST_AWS_SECRET_ACCESS_KEY"));
36+
System.getenv().get("WHITEFOX_TEST_AWS_SECRET_ACCESS_KEY"),
37+
System.getenv().get("WHITEFOX_TEST_GLUE_CATALOG_ID"));
3138
}
3239
}

client-spark/src/test/java/io/whitefox/api/utils/StorageManagerInitializer.java

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,13 +62,14 @@ public TableInfo createIcebergTableWithGlueMetastore() {
6262
var shareRequest = createShareRequest();
6363
ignoreConflict(() -> schemaV1Api.createSchema(shareRequest.getName(), schemaRequest));
6464
var createTableRequest = createIcebergTableRequest();
65+
ApiUtils.recoverConflictLazy(
66+
() -> tableV1Api.createTableInProvider(provider.getName(), createTableRequest),
67+
() -> tableV1Api.describeTableInProvider(provider.getName(), createTableRequest.getName()));
6568
ignoreConflict(() -> schemaV1Api.addTableToSchema(
6669
shareRequest.getName(),
6770
schemaRequest,
6871
addTableToSchemaRequest(providerRequest.getName(), createTableRequest.getName())));
69-
return ApiUtils.recoverConflictLazy(
70-
() -> tableV1Api.createTableInProvider(provider.getName(), createTableRequest),
71-
() -> tableV1Api.describeTableInProvider(provider.getName(), createTableRequest.getName()));
72+
return tableV1Api.describeTableInProvider(provider.getName(), createTableRequest.getName());
7273
}
7374

7475
private String createSchemaRequest(TableFormat tableFormat) {
@@ -129,7 +130,7 @@ private CreateMetastore createMetastoreRequest(
129130
.type(type)
130131
.skipValidation(true)
131132
.properties(new MetastoreProperties(new GlueProperties()
132-
.catalogId("catalogId") // TODO
133+
.catalogId(s3TestConfig.getGlueCatalogId())
133134
.credentials(new SimpleAwsCredentials()
134135
.region(s3TestConfig.getRegion())
135136
.awsAccessKeyId(s3TestConfig.getAccessKey())

0 commit comments

Comments
 (0)