tests: update onboarding permissions behavior tests

Author: itsaky

.idea/androidTestResultsUserPreferences.xml

@@ -98,9 +98,6 @@ class OnboardingActivityPermissionsBehaviorTest {
       device.getObjectWithText("DENY").clickAndWaitForWindowUpdate(device)
     }
 
-    // verify AndroidIDE is visible
-    assertThat(device.currentPackageName).isEqualTo(BuildInfo.PACKAGE_NAME)
-
     grantButton = device.getObjectWithText(permissionDesc)
       .parent!!
       .parent!!

app/src/androidTest/java/com/itsaky/androidide/activities/OnboardingActivityPermissionsBehaviorTest.kt

@@ -17,9 +17,17 @@
 
 package com.itsaky.androidide.activities
 
+import android.Manifest
+import android.content.Context
+import android.content.Intent
+import android.net.Uri
+import android.provider.Settings
 import androidx.test.ext.junit.runners.AndroidJUnit4
 import androidx.test.platform.app.InstrumentationRegistry
+import androidx.test.rule.GrantPermissionRule
+import androidx.test.runner.permission.PermissionRequester
 import androidx.test.uiautomator.UiDevice
+import androidx.test.uiautomator.Until
 import com.adevinta.android.barista.assertion.BaristaCheckedAssertions.assertChecked
 import com.adevinta.android.barista.assertion.BaristaCheckedAssertions.assertUnchecked
 import com.adevinta.android.barista.assertion.BaristaEnabledAssertions.assertDisabled
@@ -32,7 +40,10 @@ import com.adevinta.android.barista.rule.BaristaRule
 import com.google.common.truth.Truth.assertThat
 import com.itsaky.androidide.R
 import com.itsaky.androidide.preferences.internal.statOptIn
+import com.itsaky.androidide.testing.android.LAUNCH_TIMEOUT
+import com.itsaky.androidide.testing.android.getApplicationContext
 import com.itsaky.androidide.testing.android.stringRes
+import com.itsaky.androidide.utils.isAtLeastT
 import org.junit.FixMethodOrder
 import org.junit.Rule
 import org.junit.Test
@@ -52,6 +63,10 @@ class OnboardingActivityStatesTest {
   @Rule
   val baristaActivityRule = BaristaRule.create(OnboardingActivity::class.java)
 
+  @JvmField
+  @Rule
+  val grantPermissionsRule = GrantPermissionRule.grant(Manifest.permission.WRITE_EXTERNAL_STORAGE)
+
   @Test
   fun A_testOnboarding_welcomeScreen() {
     baristaActivityRule.launchActivity()
@@ -64,6 +79,8 @@ class OnboardingActivityStatesTest {
 
   @Test
   fun B_testOnboarding_statConsentScreen() {
+    tryGrantPermissions()
+
     baristaActivityRule.launchActivity()
 
     clickOn(R.id.next)
@@ -97,10 +114,12 @@ class OnboardingActivityStatesTest {
   @Test
   fun F_testOnboarding_toolsSetup() {
 
+    tryGrantPermissions()
+
     baristaActivityRule.launchActivity()
 
-    val device = UiDevice.getInstance(InstrumentationRegistry.getInstrumentation())
-    device.grantAndroidIDEPermissions()
+    clickOn(R.id.next)
+    clickOn(R.id.next)
 
     assertDisplayed(stringRes(R.string.title_install_tools))
     assertDisplayed(stringRes(R.string.subtitle_install_tools))
@@ -139,4 +158,25 @@ class OnboardingActivityStatesTest {
     assertEnabled(R.id.sdk_version_layout)
     assertEnabled(R.id.jdk_version_layout)
   }
+
+  private fun tryGrantPermissions() {
+    if (isAtLeastT()) {
+      PermissionRequester().apply {
+        addPermissions(Manifest.permission.POST_NOTIFICATIONS)
+        requestPermissions()
+      }
+    }
+
+    val device = UiDevice.getInstance(InstrumentationRegistry.getInstrumentation())
+    val context = getApplicationContext<Context>()
+
+    device.performActionAndWait({
+      val intent = Intent(Settings.ACTION_MANAGE_UNKNOWN_APP_SOURCES,
+        Uri.parse("package:${context.packageName}"))
+      intent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK)
+      context.startActivity(intent)
+    }, Until.newWindow(), LAUNCH_TIMEOUT)
+
+    device.handlePermissionInSettings(true)
+  }
 }

app/src/androidTest/java/com/itsaky/androidide/activities/OnboardingActivityStatesTest.kt

@@ -72,7 +72,7 @@ fun UiDevice.nextButton(): UiObject2 {
 fun UiDevice.handlePermissionInSettings(
   allowPermission: Boolean = true,
 ) {
-  assertThat(currentPackageName).isAnyOf("com.android.settings", "com.android.permissioncontroller")
+  assertThat(currentPackageName).startsWith("com.android.")
   assertThat(hasObjectWithText(R.string.app_name)).isTrue()
 
   val switch = findObject(By.clazz(Switch::class.java))
@@ -81,14 +81,21 @@ fun UiDevice.handlePermissionInSettings(
   // verify initial state
   assertThat(switch.isCheckable).isTrue()
   assertThat(switch.isEnabled).isTrue()
-  assertThat(switch.isChecked).isFalse()
 
   if (allowPermission) {
-    // enable permission
-    switch.click()
+    if (!switch.isChecked) {
+      // enable permission
+      switch.click()
+    }
 
     // verify changed state
     assertThat(switch.isChecked).isTrue()
+  } else if (switch.isChecked) {
+    // disable permission
+    switch.click()
+
+    // verify changed state
+    assertThat(switch.isChecked).isFalse()
   }
 
   // navigate back without enabling the permission