0

目前我编写了一个小型演示应用程序，它使用 Ktor 作为其应用程序环境，并使用 Kodein 作为依赖注入框架。

在应用程序的初始化过程中，我确实导入了一些模块，其中一个是我想在集成测试的初始化过程中替换的：

fun Application.module(testing: Boolean = false) {
logger.debug { "Starting main" }

restModule()

di {
  bind<Json>() with singleton {
    Json {
      ...
    }
  }

  import(persistenceModule)
}

在测试中，我想使用不同的persistenceModule，例如。一个内存模块。我的测试初始化​​如下：

fun start() {
  val configPath = ClassLoader.getSystemResource("application-acceptanceTest.conf").file
  engine = embeddedServer(CIO, commandLineEnvironment(arrayOf("-config=$configPath")))
  engine.start()

  val disposable = engine.environment.monitor.subscribe(ApplicationStarted) { application: Application ->
    started = true
  }

  while (!started) {
    Thread.sleep(10)
  }
  disposable.dispose()
}

我已经试过打电话

engine.application.di

但这让我（很明显）只能访问已经初始化的 Ktor 功能。这样的事情有可能吗？

4

1 回答 1

1

Kodein-DI 允许您覆盖依赖项。关于以下界面：

interface Repository {
    fun save()
    fun find()
}

你可以有一个生产实现，包含在它自己的 DI 模块中：

class PersistenceRepository : Repository {
    /* implementation */
}

val persistenceModule = DI.Module("persistenceModule") {
    bind<Repository>() with singleton { PersistenceRepository() }
}

以及同一接口的测试实现：

class MemoryRepository : Repository {
    /* implementation */
}

val memoryModule = DI.Module("memoryModule") {
    bind<Repository>(overrides = true) with singleton { MemoryRepository() }
}

注意overrides需要显式的参数。

您可以将 DI 容器传递给您的 Ktor 函数：

val mainDI = DI {
    import(persistenceModule)
}

fun Application.main(di: DI) {
    di { extend(di) }
}

并在您的测试中扩展mainDI，以覆盖正确的绑定memoryModule：

class ApplicationTest {
    val testDI = DI {
        extend(mainDI)
        import(memoryModule, allowOverride = true)
    }
    
    @Test
    fun myTest() {
        withTestApplication({ main(testDI) })
        // ...
    }
}
于 2021-03-26T05:45:56.250 回答