我们如何使用通用返回类型和隐式参数模拟Scala方法

Question

我有一个看起来像这样的配置fetcher。

def getForCountry[A](path: String, fallbackToDefault: Boolean)
                  (implicit loader: ConfigLoader[A], ac: AppContext): A = {
configuration.getOptional[A](s"${ac.country}.$path") match {
  case Some(value)                =>
    value
  case None if fallbackToDefault  =>
    configuration.get[A](path)
  case None if !fallbackToDefault =>
    throw new RuntimeException(s"${ac.country}.$path key not found in configuration")
}

相同方法的调用如下 -

val countrySpecificConfig =
  configurationHelper.getForCountry[Map[String, String]]("googleCloudPlatform.jobConfig.demandBasedPricing", fallbackToDefault = false)

现在我想在我的单元测试中模拟getforcountry方法 -

when(configurationHelper
    .getForCountry[Map[String, String]]("googleCloudPlatform.jobConfig.demandBasedPricing", fallbackToDefault = false))
    .thenReturn(countryPricingWeekConfiguation)

令人惊讶的是，这种期望似乎无法正确设置。执行测试时，模拟返回null。

有关如何进行此操作的任何线索？请随时让我知道您是否需要任何其他细节。

Answer 1

我强烈怀疑隐式 ConfigLoader和 AppContext的不同实例正在通过您的实际方法调用并嘲笑一个实例。如果您使用的是Intellij，请通过启用它们来验证哪些隐式通过。要启用它们，请按ctr+alt+shift++

这是完整的测试模拟您的情况，效果很好：

test("mock example") {
    trait ConfigLoader[T] {}
    trait AppContext { def country: String }
    trait ConfigurationHelper {
      def getForCountry[A](x: String, fallbackToDefault: Boolean = true)(implicit loader: ConfigLoader[A], ac: AppContext): A
    }
    implicit val loader: ConfigLoader[Map[String, String]] = mock[ConfigLoader[Map[String, String]]]
    implicit val ctx: AppContext                           = mock[AppContext]
    val configurationHelper                                = mock[ConfigurationHelper]
    val mockedResult = Map("x" → "1")
    when(
      configurationHelper
        .getForCountry[Map[String, String]]("googleCloudPlatform.jobConfig.demandBasedPricing", fallbackToDefault = false)
    ).thenReturn(mockedResult)
    val countrySpecificConfig =
      configurationHelper
        .getForCountry[Map[String, String]]("googleCloudPlatform.jobConfig.demandBasedPricing", fallbackToDefault = false)
    countrySpecificConfig.foreach(println)
  }
// =========================== Output ====================
// (x,1)

Answer 2

您是否尝试过Mockito-Scala？如果您使用新的语法，则隐式将自动处理（假设您使用惯用语法，并且在测试和您的产品代码中解决了相同的实例）

）

即使您使用传统语法，您的存根也会减少到

when(configurationHelper
    .getForCountry[Map[String, String]]
    (eqTo("googleCloudPlatform.jobConfig.demandBasedPricing"), eqTo(false))(*, *)
    .thenReturn(countryPricingWeekConfiguation)

或使用惯用语法

configurationHelper.getForCountry[Map[String, String]]
    ("googleCloudPlatform.jobConfig.demandBasedPricing",false)
    shouldReturn countryPricingWeekConfiguation

或在测试和产品中的隐含不相同（请注意，我也可以混合诸如 *和诸如'false'之类的原始参数之类的ARG匹配器）

）

configurationHelper.getForCountry[Map[String, String]]
    ("googleCloudPlatform.jobConfig.demandBasedPricing",false)(*,*)
    shouldReturn countryPricingWeekConfiguation

Answer 3

感谢一个吨pritam。以下代码似乎有效。

when(configurationHelper
    .getForCountry[Map[String, String]]
    (ArgumentMatchers.eq("googleCloudPlatform.jobConfig.demandBasedPricing"), ArgumentMatchers.eq(false))
    (ArgumentMatchers.any[ConfigLoader[Map[String, String]]](), ArgumentMatchers.any[AppContext]()))
    .thenReturn(countryPricingWeekConfiguation)