From 1955144ffc1b90aaa083861b67e5c57936a53df2 Mon Sep 17 00:00:00 2001
From: Tom Hsu <tomhsu@google.com>
Date: Wed, 2 Oct 2024 07:47:10 +0000
Subject: [PATCH] Avoid test flaky

Change-Id: Id1290e6979fdafa93ded6f50e49d6b3464a80b50
flag: EXEMPT bugfix
fix: 365673174
Test: atest
---
 .../src/com/android/settings/vpn2/VpnSettingsTest.java    | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tests/unit/src/com/android/settings/vpn2/VpnSettingsTest.java b/tests/unit/src/com/android/settings/vpn2/VpnSettingsTest.java
index 953a524750d..9e138965289 100644
--- a/tests/unit/src/com/android/settings/vpn2/VpnSettingsTest.java
+++ b/tests/unit/src/com/android/settings/vpn2/VpnSettingsTest.java
@@ -130,6 +130,7 @@ public class VpnSettingsTest {
     }
 
     @Test
+    @UiThreadTest
     public void setShownAdvancedPreferences_hasGeneralVpn_returnsVpnCountAs1() {
         Set<Preference> updates = new ArraySet<>();
         AppPreference pref =
@@ -144,6 +145,7 @@ public class VpnSettingsTest {
     }
 
     @Test
+    @UiThreadTest
     public void setShownAdvancedPreferences_hasAdvancedVpn_returnsAdvancedVpnCountAs1() {
         Set<Preference> updates = new ArraySet<>();
         AppPreference pref =
@@ -158,6 +160,7 @@ public class VpnSettingsTest {
     }
 
     @Test
+    @UiThreadTest
     public void setShownAdvancedPreferences_noVpn_returnsEmpty() {
         Set<Preference> updates = new ArraySet<>();
 
@@ -170,6 +173,7 @@ public class VpnSettingsTest {
     }
 
     @Test
+    @UiThreadTest
     public void getVpnApps_isAdvancedVpn_returnsOne() throws Exception {
         ApplicationInfo info = new ApplicationInfo();
         info.uid = 1111;
@@ -198,6 +202,7 @@ public class VpnSettingsTest {
     }
 
     @Test
+    @UiThreadTest
     public void clickVpn_VpnConnected_doesNotStartVpnLaunchIntent()
             throws PackageManager.NameNotFoundException {
         Set<Preference> updates = new ArraySet<>();
@@ -217,6 +222,7 @@ public class VpnSettingsTest {
     }
 
     @Test
+    @UiThreadTest
     public void clickVpn_VpnDisconnected_startsVpnLaunchIntent()
             throws PackageManager.NameNotFoundException {
         Set<Preference> updates = new ArraySet<>();
@@ -238,6 +244,7 @@ public class VpnSettingsTest {
     }
 
     @Test
+    @UiThreadTest
     public void clickAdvancedVpn_VpnConnectedDisconnectDialogDisabled_startsAppLaunchIntent()
             throws PackageManager.NameNotFoundException {
         Set<Preference> updates = new ArraySet<>();
@@ -262,6 +269,7 @@ public class VpnSettingsTest {
     }
 
     @Test
+    @UiThreadTest
     public void clickAdvancedVpn_VpnConnectedDisconnectDialogEnabled_doesNotStartAppLaunchIntent()
             throws PackageManager.NameNotFoundException {
         Set<Preference> updates = new ArraySet<>();