fix bug capturing screenshot and bugreport
Change-Id: I5fd623ccdcb6d64b993126e74232088ecacb3d0a
diff --git a/src/com/android/tradefed/testtype/UiAutomatorTest.java b/src/com/android/tradefed/testtype/UiAutomatorTest.java
index 03fb8a7..13d6b9f 100644
--- a/src/com/android/tradefed/testtype/UiAutomatorTest.java
+++ b/src/com/android/tradefed/testtype/UiAutomatorTest.java
@@ -262,7 +262,7 @@
mFailureAction == TestFailureAction.BUGREPORT_AND_SCREENSHOT) {
InputStreamSource screenshot = null;
try {
- screenshot = getDevice().getScreenshot();
+ screenshot = device.getScreenshot();
listener.testLog(prefix + "_screenshot", LogDataType.PNG, screenshot);
} catch (DeviceNotAvailableException e) {
CLog.e(e);
@@ -276,7 +276,7 @@
if (mFailureAction == TestFailureAction.BUGREPORT ||
mFailureAction == TestFailureAction.BUGREPORT_AND_SCREENSHOT) {
InputStreamSource data = null;
- data = getDevice().getBugreport();
+ data = device.getBugreport();
listener.testLog(prefix + "_bugreport", LogDataType.BUGREPORT, data);
if (data != null) {
data.cancel();