commit 49728ffdea72370729d5a615e889bf4e0d700516
author Takeshi Hagikura <thagikura@google.com> Thu Mar 19 18:34:54 2015 +0900
committer Takeshi Hagikura <thagikura@google.com> Thu Mar 19 18:34:54 2015 +0900
tree 4c27a54a29a95a43c976bd922d4f8d2e7f3fd1aa
parent d520e5798696a951c164b8f4a2506773dbbf9044

Use UiThreadTest annotation

Change-Id: If6f2dcfde30855a04acd6effebefa336825f7247

system/AppUsageStatistics/Application/tests/src/com/example/android/appusagestatistics/AppUsageStatisticsFragmentTests.java

diff --git a/system/AppUsageStatistics/Application/tests/src/com/example/android/appusagestatistics/AppUsageStatisticsFragmentTests.java b/system/AppUsageStatistics/Application/tests/src/com/example/android/appusagestatistics/AppUsageStatisticsFragmentTests.java
index bcffdea..68f9711 100644
--- a/system/AppUsageStatistics/Application/tests/src/com/example/android/appusagestatistics/AppUsageStatisticsFragmentTests.java
+++ b/system/AppUsageStatistics/Application/tests/src/com/example/android/appusagestatistics/AppUsageStatisticsFragmentTests.java
@@ -21,6 +21,7 @@
 import android.app.usage.UsageStats;
 import android.app.usage.UsageStatsManager;
 import android.test.ActivityInstrumentationTestCase2;
+import android.test.UiThreadTest;
 
 import java.util.List;
 
@@ -81,16 +82,11 @@
         assertNotNull(usageStatsList);
     }
 
+    @UiThreadTest
     public void testUpdateAppsList() {
-        final List<UsageStats> usageStatsList = mTestFragment
+        List<UsageStats> usageStatsList = mTestFragment
                 .getUsageStatistics(UsageStatsManager.INTERVAL_DAILY);
-        mTestActivity.runOnUiThread(new Runnable() {
-            @Override
-            public void run() {
-                mTestFragment.updateAppsList(usageStatsList);
-            }
-        });
-        getInstrumentation().waitForIdleSync();
+        mTestFragment.updateAppsList(usageStatsList);
 
         // The result depends on if the app is granted the access to App usage statistics.
         if (usageStatsList.size() == 0) {