Index: build/android/pylib/uiautomator/test_runner.py |
diff --git a/build/android/pylib/uiautomator/test_runner.py b/build/android/pylib/uiautomator/test_runner.py |
index f4e573063d3b53c0348651767557959b5845b41e..c7239b436346fc34feb441426e0e662231d3b920 100644 |
--- a/build/android/pylib/uiautomator/test_runner.py |
+++ b/build/android/pylib/uiautomator/test_runner.py |
@@ -63,7 +63,8 @@ class TestRunner(instr_test_runner.TestRunner): |
def _RunTest(self, test, timeout): |
self.device.ClearApplicationState(self._package) |
if self.flags: |
- if 'Feature:FirstRunExperience' in self.test_pkg.GetTestAnnotations(test): |
+ annotations = self.test_pkg.GetTestAnnotations(test) |
+ if ('FirstRunExperience' == annotations(test).get('Feature', None)): |
self.flags.RemoveFlags(['--disable-fre']) |
else: |
self.flags.AddFlags(['--disable-fre']) |