FakeAppUpdateManager永远不会调用OnSuccessListener

如何解决FakeAppUpdateManager永远不会调用OnSuccessListener

我已经在我的应用程序中实现了应用程序内更新（即时流），我想对其进行单元测试。在实际应用中它可以正常工作，但是在使用FakeAppUpdateManager时，OnSuccessListener不会被调用。

代码：

    fun checkForUpdates(activity: Activity) {
        val appUpdateInfoTask = appUpdateManager.appUpdateInfo

        appUpdateInfoTask.addOnSuccessListener { appUpdateInfo ->
            if (appUpdateInfo.updateAvailability() != UpdateAvailability.UPDATE_AVAILABLE) {
                return@addOnSuccessListener
            }

            if (appUpdateInfo.isUpdateTypeAllowed(AppUpdateType.IMMEDIATE) &&
                MIN_VERSION_CODE_FOR_IMMEDIATE_UPDATE > BuildConfig.VERSION_CODE) {
                startUpdate(activity,appUpdateInfo)
            }
        }
    }

测试：

@RunWith(RobolectricTestRunner::class)
@Config(sdk = [Build.VERSION_CODES.O_MR1])
class UpdateControllerTest : KoinTest {

    private val updateController: UpdateController by inject()
    private val mainActivity: MainActivity  by inject()
    private val fakeAppUpdateManager: AppUpdateManager by inject()

    @get:Rule
    val koinTestRule = KoinTestRule.create {
        modules(
            module {
                single<AppUpdateManager> { FakeAppUpdateManager(get()).also { it.setUpdateAvailable(999999) } }
                single { UpdateController(get(),get()) }
                single<Context> { ApplicationProvider.getApplicationContext() }
            }
        )
    }

    @get:Rule
    val mockProvider = MockProviderRule.create { clazz ->
        Mockito.mock(clazz.java)
    }

    @Before
    fun setUp() {
        declareMock<MainActivity> {}
    }

    @Test
    fun test() {
        (fakeAppUpdateManager as FakeAppUpdateManager).setUpdateAvailable(999999)

        updateController.checkForUpdates(mainActivity)

        assert((fakeAppUpdateManager as FakeAppUpdateManager).isImmediateFlowVisible).equals(true)
    }
}