Commit 3cc64ae2 by samrobinson Committed by kim-vde

Pull files from the device cache after a MH test concludes.

PiperOrigin-RevId: 412251020
parent be0b2b8c
...@@ -83,30 +83,26 @@ import org.checkerframework.checker.nullness.compatqual.NullableType; ...@@ -83,30 +83,26 @@ import org.checkerframework.checker.nullness.compatqual.NullableType;
Uri uri = Uri.parse(uriString); Uri uri = Uri.parse(uriString);
File externalCacheFile = createExternalCacheFile(context, /* filePrefix= */ testId); File externalCacheFile = createExternalCacheFile(context, /* filePrefix= */ testId);
try { InstrumentationRegistry.getInstrumentation()
InstrumentationRegistry.getInstrumentation() .runOnMainSync(
.runOnMainSync( () -> {
() -> { try {
try { testTransformer.startTransformation(
testTransformer.startTransformation( MediaItem.fromUri(uri), externalCacheFile.getAbsolutePath());
MediaItem.fromUri(uri), externalCacheFile.getAbsolutePath()); } catch (IOException e) {
} catch (IOException e) { exceptionReference.set(e);
exceptionReference.set(e); }
} });
});
assertWithMessage("Transformer timed out after " + timeoutSeconds + " seconds.") assertWithMessage("Transformer timed out after " + timeoutSeconds + " seconds.")
.that(countDownLatch.await(timeoutSeconds, SECONDS)) .that(countDownLatch.await(timeoutSeconds, SECONDS))
.isTrue(); .isTrue();
@Nullable Exception exception = exceptionReference.get(); @Nullable Exception exception = exceptionReference.get();
if (exception != null) { if (exception != null) {
throw exception; throw exception;
}
long outputSizeBytes = externalCacheFile.length();
return new TransformationResult(outputSizeBytes);
} finally {
externalCacheFile.delete();
} }
long outputSizeBytes = externalCacheFile.length();
return new TransformationResult(outputSizeBytes);
} }
private static File createExternalCacheFile(Context context, String filePrefix) private static File createExternalCacheFile(Context context, String filePrefix)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment