Sample integration test with FakeAppUpdateManager to validate in app update implementation.

MainActivityTest_FlexibleCompletes.kt

@Test
fun testFlexibleUpdate_Completes() {
  // Setup flexible update.
  fakeAppUpdateManager.partiallyAllowedUpdateType = AppUpdateType.FLEXIBLE
  fakeAppUpdateManager.setUpdateAvailable(2)

  ActivityScenario.launch(MainActivity::class.java)

  // Validate that flexible update is prompted to the user.
  assertTrue(fakeAppUpdateManager.isConfirmationDialogVisible)

  // Simulate user's and download behavior.
  fakeAppUpdateManager.userAcceptsUpdate()

  fakeAppUpdateManager.downloadStarts()

  fakeAppUpdateManager.downloadCompletes()

  // Perform a click on the Snackbar to complete the update process.
  onView(
      allOf(
          isDescendantOfA(instanceOf(Snackbar.SnackbarLayout::class.java)),
          instanceOf(AppCompatButton::class.java)
      )
  ).perform(ViewActions.click())

  // Validate that update is completed and app is restarted.
  assertTrue(fakeAppUpdateManager.isInstallSplashScreenVisible)

  fakeAppUpdateManager.installCompletes()
}