How to get Worker to run synchronously in Android Instrumentation test? - android-testing

Question

I am running instrumentation tests on a worker WorkerGetData which fetches data from repository.
My test is failing because the worker is running asynchronously on a default dispatcher. The test fails on assertion with expected: SUCCEEDED but was : RUNNING.
How can I get the worker to run synchronously in the test?
#RunWith(AndroidJUnit4::class)
#HiltAndroidTest
class TestClass {
#get:Rule
val hiltRule = HiltAndroidRule(this)
#Inject
lateinit var workerConfiguration: Configuration
#Before
fun setup() {
hiltRule.inject()
context = InstrumentationRegistry.getInstrumentation().targetContext
val config = workerConfiguration
// Initialize WorkManager for instrumentation tests.
WorkManagerTestInitHelper.initializeTestWorkManager(context, config)
}
#Test
fun testGetDataWorker() {
val request = OneTimeWorkRequestBuilder<WorkerGetData>()
.build()
val workManager = WorkManager.getInstance(context)
workManager.enqueue(request).result.get()
val workInfo = workManager.getWorkInfoById(request.id).get()
assertThat(workInfo.state).isEqualTo(WorkInfo.State.SUCCEEDED)
}
}
#Module
#TestInstallIn(
components = [SingletonComponent::class],
replaces = [WorkManagerModule::class]
)
object FakeWorkManagerModule {
#Singleton
#Provides
fun provideWorkManagerConfiguration(
myDelegatingWorkerFactory: MyDelegatingWorkerFactory,
): Configuration {
return Configuration.Builder()
.setExecutor(SynchronousExecutor())
.setMinimumLoggingLevel(android.util.Log.DEBUG)
.setWorkerFactory(myDelegatingWorkerFactory)
.build()
}
}