Skip to content

Commit 4b5336e

Browse files
committed
added hook for bq tables
1 parent dfeddf7 commit 4b5336e

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

src/e2e-test/java/io/cdap/plugin/tests/hooks/TestSetupHooks.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -256,7 +256,7 @@ public static void setNewConnectionName() {
256256
BeforeActions.scenario.write("New Connection name: " + connectionName);
257257
}
258258

259-
@After(order = 1, value = "@BQ_SINK_CLEANUP")
259+
@After(order = 1, value = "@BQ_SINK_CLEANUP or @BQ_SINK")
260260
public static void deleteTempTargetBQTable() throws IOException, InterruptedException {
261261
try {
262262
BigQueryClient.dropBqQuery(bqTargetTable);
@@ -270,4 +270,14 @@ public static void deleteTempTargetBQTable() throws IOException, InterruptedExce
270270
}
271271
}
272272
}
273+
274+
@After(order = 1, value = "@BQ_SOURCE_TEST_RECEIVING_SLIP_LINE or @BQ_SOURCE_UPDATE_SERVICE_OFFERING or " +
275+
"@BQ_SOURCE_UPDATE_VENDOR_CATALOG_ITEM or @BQ_SOURCE_UPDATE_AGENT_ASSIST_RECOMMENDATION or " +
276+
"@BQ_SOURCE_UPDATE_RECEIVING_SLIP_LINE or @BQ_SOURCE_SERVICE_OFFERING or @BQ_SOURCE_VENDOR_CATALOG_ITEM or " +
277+
"@BQ_SOURCE_AGENT_ASSIST_RECOMMENDATION")
278+
public static void deleteTempSourceBQTable() throws IOException, InterruptedException {
279+
BigQueryClient.dropBqQuery(bqSourceTable);
280+
PluginPropertyUtils.removePluginProp("bqSourceTable");
281+
BeforeActions.scenario.write("BQ source Table " + bqSourceTable + " deleted successfully");
282+
}
273283
}

0 commit comments

Comments
 (0)