Skip to content

Commit db1352a

Browse files
author
Hamza Jugon
committed
Add integration test for scheduler
1 parent 9a93bd7 commit db1352a

File tree

3 files changed

+17
-23
lines changed

3 files changed

+17
-23
lines changed

beekeeper-integration-tests/src/test/java/com/expediagroup/beekeeper/integration/BeekeeperExpiredMetadataSchedulerApiaryIntegrationTest.java

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -214,16 +214,12 @@ public void expiredMetadataMultipleAlterPartitionTableEvents() throws SQLExcepti
214214
}
215215

216216
@Test
217-
public void expiredMetadataAlterIcebergTableEventShouldBeIgnored()
218-
throws SQLException, IOException, URISyntaxException {
219-
insertExpiredMetadata(LOCATION_A + "-old", null);
220-
221-
AlterTableSqsMessage alterTableSqsMessage = new AlterTableSqsMessage(LOCATION_A, true, true);
222-
amazonSQS.sendMessage(sendMessageRequest(alterTableSqsMessage.getFormattedString()));
217+
public void expiredMetadataCreateIcebergTableEvent() throws SQLException, IOException, URISyntaxException {
218+
CreateTableSqsMessage createTableSqsMessage = new CreateTableSqsMessage(LOCATION_A, true, true);
219+
amazonSQS.sendMessage(sendMessageRequest(createTableSqsMessage.getFormattedString()));
223220

224-
await().atMost(TIMEOUT, TimeUnit.SECONDS).until(() -> getUpdatedExpiredMetadataRowCount() == 1);
221+
await().atMost(TIMEOUT, TimeUnit.SECONDS).until(() -> getExpiredMetadataRowCount() == 0);
225222

226-
// iceberg table event should be ignored
227223
List<HousekeepingMetadata> expiredMetadata = getExpiredMetadata();
228224
assertThat(expiredMetadata.size()).isEqualTo(0);
229225
}

beekeeper-integration-tests/src/test/java/com/expediagroup/beekeeper/integration/model/AlterTableSqsMessage.java

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -51,21 +51,6 @@ public AlterTableSqsMessage(
5151
setExpired(isExpired);
5252
}
5353

54-
public AlterTableSqsMessage(
55-
String tableLocation,
56-
boolean isExpired,
57-
boolean isIceberg
58-
) throws IOException, URISyntaxException {
59-
super(ALTER_TABLE);
60-
setTableLocation(tableLocation);
61-
setOldTableLocation(DUMMY_LOCATION);
62-
setOldTableName(TABLE_NAME_VALUE);
63-
setExpired(isExpired);
64-
if (isIceberg) {
65-
setIceberg();
66-
}
67-
}
68-
6954
public void setOldTableLocation(String oldTableLocation) {
7055
apiaryEventMessageJsonObject.add(EVENT_TABLE_OLD_LOCATION_KEY, new JsonPrimitive(oldTableLocation));
7156
}

beekeeper-integration-tests/src/test/java/com/expediagroup/beekeeper/integration/model/CreateTableSqsMessage.java

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,4 +30,17 @@ public CreateTableSqsMessage(
3030
setTableLocation(tableLocation);
3131
setExpired(isExpired);
3232
}
33+
34+
public CreateTableSqsMessage(
35+
String tableLocation,
36+
boolean isIceberg,
37+
boolean isExpired
38+
) throws IOException, URISyntaxException {
39+
super(CREATE_TABLE);
40+
setTableLocation(tableLocation);
41+
setExpired(isExpired);
42+
if (isIceberg) {
43+
setIceberg();
44+
}
45+
}
3346
}

0 commit comments

Comments
 (0)