mirror of https://github.com/nextcloud/android
30 lines
1.1 KiB
Kotlin
30 lines
1.1 KiB
Kotlin
/*
|
|
* Nextcloud - Android Client
|
|
*
|
|
* SPDX-FileCopyrightText: 2023 Álvaro Brey <alvaro@alvarobrey.com>
|
|
* SPDX-FileCopyrightText: 2023 Nextcloud GmbH
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
package com.nextcloud.test
|
|
|
|
import android.app.Instrumentation
|
|
import androidx.test.platform.app.InstrumentationRegistry
|
|
import dagger.android.AndroidInjector
|
|
import org.junit.rules.TestRule
|
|
import org.junit.runner.Description
|
|
import org.junit.runners.model.Statement
|
|
|
|
class InjectionOverrideRule(private val overrideInjectors: Map<Class<*>, AndroidInjector<*>>) : TestRule {
|
|
override fun apply(base: Statement, description: Description): Statement = object : Statement() {
|
|
override fun evaluate() {
|
|
val instrumentation: Instrumentation = InstrumentationRegistry.getInstrumentation()
|
|
val testApp = instrumentation.targetContext.applicationContext as TestMainApp
|
|
overrideInjectors.entries.forEach {
|
|
testApp.addTestInjector(it.key, it.value)
|
|
}
|
|
base.evaluate()
|
|
testApp.clearTestInjectors()
|
|
}
|
|
}
|
|
}
|