Browse Source

HADOOP-18372. ILoadTestS3ABulkDeleteThrottling failing. (#4642)

Contributed by Ahmar Suhail
ahmarsuhail 2 years ago
parent
commit
c92ff0b4f1

+ 11 - 11
hadoop-tools/hadoop-aws/src/test/java/org/apache/hadoop/fs/s3a/scale/ILoadTestS3ABulkDeleteThrottling.java

@@ -45,6 +45,7 @@ import org.slf4j.LoggerFactory;
 import org.apache.hadoop.conf.Configuration;
 import org.apache.hadoop.conf.Configuration;
 import org.apache.hadoop.fs.Path;
 import org.apache.hadoop.fs.Path;
 import org.apache.hadoop.fs.contract.ContractTestUtils;
 import org.apache.hadoop.fs.contract.ContractTestUtils;
+import org.apache.hadoop.fs.store.audit.AuditSpan;
 import org.apache.hadoop.fs.s3a.S3AFileSystem;
 import org.apache.hadoop.fs.s3a.S3AFileSystem;
 import org.apache.hadoop.fs.s3a.S3ATestUtils;
 import org.apache.hadoop.fs.s3a.S3ATestUtils;
 import org.apache.hadoop.fs.s3a.auth.delegation.Csvout;
 import org.apache.hadoop.fs.s3a.auth.delegation.Csvout;
@@ -144,26 +145,25 @@ public class ILoadTestS3ABulkDeleteThrottling extends S3AScaleTestBase {
   @Override
   @Override
   protected Configuration createScaleConfiguration() {
   protected Configuration createScaleConfiguration() {
     Configuration conf = super.createScaleConfiguration();
     Configuration conf = super.createScaleConfiguration();
-    S3ATestUtils.disableFilesystemCaching(conf);
-    return conf;
-  }
 
 
-  @Override
-  public void setup() throws Exception {
-    final Configuration conf = getConf();
     S3ATestUtils.removeBaseAndBucketOverrides(conf,
     S3ATestUtils.removeBaseAndBucketOverrides(conf,
         EXPERIMENTAL_AWS_INTERNAL_THROTTLING,
         EXPERIMENTAL_AWS_INTERNAL_THROTTLING,
         BULK_DELETE_PAGE_SIZE,
         BULK_DELETE_PAGE_SIZE,
-        USER_AGENT_PREFIX);
+        USER_AGENT_PREFIX,
+        ENABLE_MULTI_DELETE);
     conf.setBoolean(EXPERIMENTAL_AWS_INTERNAL_THROTTLING, throttle);
     conf.setBoolean(EXPERIMENTAL_AWS_INTERNAL_THROTTLING, throttle);
-    Assertions.assertThat(pageSize)
-        .describedAs("page size")
-        .isGreaterThan(0);
     conf.setInt(BULK_DELETE_PAGE_SIZE, pageSize);
     conf.setInt(BULK_DELETE_PAGE_SIZE, pageSize);
     conf.set(USER_AGENT_PREFIX,
     conf.set(USER_AGENT_PREFIX,
         String.format("ILoadTestS3ABulkDeleteThrottling-%s-%04d",
         String.format("ILoadTestS3ABulkDeleteThrottling-%s-%04d",
             throttle, pageSize));
             throttle, pageSize));
 
 
+    S3ATestUtils.disableFilesystemCaching(conf);
+    return conf;
+  }
+
+  @Override
+  public void setup() throws Exception {
+    final Configuration conf = getConf();
     super.setup();
     super.setup();
     Assume.assumeTrue("multipart delete disabled",
     Assume.assumeTrue("multipart delete disabled",
         conf.getBoolean(ENABLE_MULTI_DELETE, true));
         conf.getBoolean(ENABLE_MULTI_DELETE, true));
@@ -246,7 +246,7 @@ public class ILoadTestS3ABulkDeleteThrottling extends S3AScaleTestBase {
         final ContractTestUtils.NanoTimer timer =
         final ContractTestUtils.NanoTimer timer =
             new ContractTestUtils.NanoTimer();
             new ContractTestUtils.NanoTimer();
         Exception ex = null;
         Exception ex = null;
-        try {
+        try (AuditSpan span = span())  {
           fs.removeKeys(fileList, false);
           fs.removeKeys(fileList, false);
         } catch (IOException e) {
         } catch (IOException e) {
           ex = e;
           ex = e;