@@ -177,13 +177,13 @@ public static void createBucketWithAvroTestFile() throws IOException, URISyntaxE
177177
178178 @ After (order = 1 , value = "@GCS_CSV_TEST or @GCS_TSV_TEST or @GCS_BLOB_TEST " +
179179 "or @GCS_DELIMITED_TEST or @GCS_TEXT_TEST or @GCS_OUTPUT_FIELD_TEST or @GCS_DATATYPE_1_TEST or " +
180- "@GCS_DATATYPE_2_TEST or @GCS_READ_RECURSIVE_TEST or @GCS_DELETE_WILDCARD_TEST or @GCS_CSV_RANGE_TEST or" +
181- " @GCS_PARQUET_TEST or @GCS_AVRO_TEST or @GCS_DATATYPE_TEST or @GCS_AVRO_FILE" )
180+ "@GCS_DATATYPE_2_TEST or @GCS_READ_RECURSIVE_TEST or @GCS_DELETE_WILDCARD_TEST or @GCS_CSV_RANGE_TEST or " +
181+ "@GCS_PARQUET_TEST or @GCS_AVRO_TEST or @GCS_DATATYPE_TEST or @GCS_AVRO_FILE or @GCS_CSV or " +
182+ "@GCS_JSON_TEST" )
182183 public static void deleteSourceBucketWithFile () {
183184 deleteGCSBucket (gcsSourceBucketName );
184185 PluginPropertyUtils .removePluginProp ("gcsSourceBucketName" );
185186 PluginPropertyUtils .removePluginProp ("gcsSourcePath" );
186- gcsSourceBucketName = StringUtils .EMPTY ;
187187 }
188188
189189 @ After (order = 1 , value = "@GCS_DELETE_MULTIPLE_BUCKETS_TEST" )
@@ -211,7 +211,7 @@ public static void createTargetGCSBucketWithCSVFile() throws IOException, URISyn
211211 BeforeActions .scenario .write ("GCS target bucket name - " + gcsTargetBucketName );
212212 }
213213
214- @ After (order = 1 , value = "@GCS_SINK_TEST or @GCS_SINK_EXISTING_BUCKET_TEST" )
214+ @ After (order = 1 , value = "@GCS_SINK_TEST or @GCS_SINK_EXISTING_BUCKET_TEST or @GCS_SINK_MULTI_PART_UPLOAD " )
215215 public static void deleteTargetBucketWithFile () {
216216 deleteGCSBucket (gcsTargetBucketName );
217217 PluginPropertyUtils .removePluginProp ("gcsTargetBucketName" );
@@ -226,7 +226,8 @@ public static void setTempTargetBQTableName() {
226226 BeforeActions .scenario .write ("BQ Target table name - " + bqTargetTable );
227227 }
228228
229- @ After (order = 1 , value = "@BQ_SINK_TEST" )
229+ @ After (order = 1 , value = "@BQ_SINK_TEST or @BQ_UPSERT_SINK_TEST or @BQ_UPDATE_SINK_DEDUPE_TEST or " +
230+ "@BQ_EXISTING_SINK_TEST or @BQ_UPSERT_DEDUPE_SINK_TEST or @BQ_INSERT_SINK_TEST" )
230231 public static void deleteTempTargetBQTable () throws IOException , InterruptedException {
231232 try {
232233 BigQueryClient .dropBqQuery (bqTargetTable );
@@ -270,7 +271,10 @@ public static void createTempSourceBQTable() throws IOException, InterruptedExce
270271
271272 @ After (order = 1 , value = "@BQ_SOURCE_TEST or @BQ_PARTITIONED_SOURCE_TEST or @BQ_SOURCE_DATATYPE_TEST or " +
272273 "@BQ_INSERT_SOURCE_TEST or @BQ_UPDATE_SINK_TEST or @BQ_EXISTING_SOURCE_TEST or @BQ_EXISTING_SINK_TEST or " +
273- "@BQ_EXISTING_SOURCE_DATATYPE_TEST or @BQ_EXISTING_SINK_DATATYPE_TEST" )
274+ "@BQ_EXISTING_SOURCE_DATATYPE_TEST or @BQ_EXISTING_SINK_DATATYPE_TEST or @BQ_UPSERT_SOURCE_TEST or " +
275+ "@BQ_NULL_MODE_SOURCE_TEST or @BQ_UPDATE_SOURCE_DEDUPE_TEST or @BQ_INSERT_INT_SOURCE_TEST or " +
276+ "@BQ_TIME_SOURCE_TEST or @BQ_UPSERT_DEDUPE_SOURCE_TEST or @BQ_PRIMARY_RECORD_SOURCE_TEST or " +
277+ "@BQ_SINGLE_SOURCE_BQMT_TEST" )
274278 public static void deleteTempSourceBQTable () throws IOException , InterruptedException {
275279 BigQueryClient .dropBqQuery (bqSourceTable );
276280 PluginPropertyUtils .removePluginProp ("bqSourceTable" );
@@ -394,7 +398,7 @@ private static void createSourceBQViewWithQueries(String bqCreateViewQueryFile)
394398 }
395399
396400 private static String createGCSBucketWithFile (String filePath ) throws IOException , URISyntaxException {
397- String bucketName = StorageClient .createBucket ("00000000 -e2e-" + UUID .randomUUID ()).getName ();
401+ String bucketName = StorageClient .createBucket ("cdf -e2e-test -" + UUID .randomUUID ()).getName ();
398402 StorageClient .uploadObject (bucketName , filePath , filePath );
399403 PluginPropertyUtils .addPluginProp ("gcsSourceBucketName" , bucketName );
400404 PluginPropertyUtils .addPluginProp ("gcsSourcePath" , "gs://" + bucketName + "/" + filePath );
@@ -936,12 +940,18 @@ public static void createBucketWithMultipleTestFiles() throws IOException, URISy
936940 gcsSourceBucketName = createGCSBucketWithMultipleFiles (PluginPropertyUtils .pluginProp ("gcsMultipleFilesPath" ));
937941 }
938942
943+ @ After (order = 1 , value = "@GCS_MULTIPLE_FILES_TEST or @GCS_MULTIPLE_FILES_REGEX_TEST" )
944+ public static void deleteSourceBucketWithMultipleFile () {
945+ deleteGCSBucket (gcsSourceBucketName );
946+ PluginPropertyUtils .removePluginProp ("gcsSourceBucketName" );
947+ PluginPropertyUtils .removePluginProp ("gcsSourcePath" );
948+ }
949+
939950 @ Before (order = 1 , value = "@GCS_MULTIPLE_FILES_REGEX_TEST" )
940951 public static void createBucketWithMultipleTestFilesWithRegex () throws IOException , URISyntaxException {
941952 gcsSourceBucketName = createGCSBucketWithMultipleFiles (PluginPropertyUtils .pluginProp (
942953 "gcsMultipleFilesFilterRegexPath" ));
943- PluginPropertyUtils .addPluginProp (" bqTargetTable" , bqTargetTable );
944- BeforeActions .scenario .write ("BQ Target Table " + bqTargetTable + " updated successfully" );
954+ PluginPropertyUtils .addPluginProp (" gcsSourceBucketName" , gcsSourceBucketName );
945955 }
946956
947957 @ Before (order = 1 , value = "@BQ_EXISTING_SOURCE_TEST" )
0 commit comments