|
45 | 45 | import org.apache.hadoop.conf.Configuration; |
46 | 46 | import org.apache.hadoop.fs.Path; |
47 | 47 | import org.apache.hadoop.fs.contract.ContractTestUtils; |
| 48 | +import org.apache.hadoop.fs.store.audit.AuditSpan; |
48 | 49 | import org.apache.hadoop.fs.s3a.S3AFileSystem; |
49 | 50 | import org.apache.hadoop.fs.s3a.S3ATestUtils; |
50 | 51 | import org.apache.hadoop.fs.s3a.auth.delegation.Csvout; |
@@ -144,26 +145,25 @@ public ILoadTestS3ABulkDeleteThrottling( |
144 | 145 | @Override |
145 | 146 | protected Configuration createScaleConfiguration() { |
146 | 147 | Configuration conf = super.createScaleConfiguration(); |
147 | | - S3ATestUtils.disableFilesystemCaching(conf); |
148 | | - return conf; |
149 | | - } |
150 | 148 |
|
151 | | - @Override |
152 | | - public void setup() throws Exception { |
153 | | - final Configuration conf = getConf(); |
154 | 149 | S3ATestUtils.removeBaseAndBucketOverrides(conf, |
155 | 150 | EXPERIMENTAL_AWS_INTERNAL_THROTTLING, |
156 | 151 | BULK_DELETE_PAGE_SIZE, |
157 | | - USER_AGENT_PREFIX); |
| 152 | + USER_AGENT_PREFIX, |
| 153 | + ENABLE_MULTI_DELETE); |
158 | 154 | conf.setBoolean(EXPERIMENTAL_AWS_INTERNAL_THROTTLING, throttle); |
159 | | - Assertions.assertThat(pageSize) |
160 | | - .describedAs("page size") |
161 | | - .isGreaterThan(0); |
162 | 155 | conf.setInt(BULK_DELETE_PAGE_SIZE, pageSize); |
163 | 156 | conf.set(USER_AGENT_PREFIX, |
164 | 157 | String.format("ILoadTestS3ABulkDeleteThrottling-%s-%04d", |
165 | 158 | throttle, pageSize)); |
166 | 159 |
|
| 160 | + S3ATestUtils.disableFilesystemCaching(conf); |
| 161 | + return conf; |
| 162 | + } |
| 163 | + |
| 164 | + @Override |
| 165 | + public void setup() throws Exception { |
| 166 | + final Configuration conf = getConf(); |
167 | 167 | super.setup(); |
168 | 168 | Assume.assumeTrue("multipart delete disabled", |
169 | 169 | conf.getBoolean(ENABLE_MULTI_DELETE, true)); |
@@ -246,7 +246,7 @@ private File deleteFiles(final int requestCount, |
246 | 246 | final ContractTestUtils.NanoTimer timer = |
247 | 247 | new ContractTestUtils.NanoTimer(); |
248 | 248 | Exception ex = null; |
249 | | - try { |
| 249 | + try (AuditSpan span = span()) { |
250 | 250 | fs.removeKeys(fileList, false); |
251 | 251 | } catch (IOException e) { |
252 | 252 | ex = e; |
|
0 commit comments