如何在 Scala 中模拟 LocalDate.now()

Question

我有 object

object func1{
   def something(arg1: Int): String{
    // var date = something2()
    // Using date
   }

   def something2(): LocalDate{
      LocalDate.now()
    }
}
  

我想写一个测试方法，我怎么能模拟something2

Answer 1

你不能制作 static 方法。 所以，你需要做这样的事情：

def something(val now: () => Date = LocalDate.now) = ...

然后，在您的生产代码中，您像往常一样执行val foo = something() ，但在您的测试中，您可以执行类似

something(_ => new Date(0l)) shouldBe foo

或者

val date = mock[Function0[Date]]
when(date.apply()).thenReturn(new Date(0l))
something(date) shouldBe foo

Answer 2

如果代码的功能取决于时间，那么绝对建议您再次查看正在测试的代码，并明确传递Clock作为参数或类似的东西（例如其他评论中建议的 function）以进行它的行为易于测试。

如果您正在测试它以确保某处的字段确实包含LocalDate.now()返回的时间，请考虑您可能正在测试LocalDate的行为而不是您自己的代码，所以可能 mocking 并没有真正给你任何事物。

如果计时是函数行为的基础并且您无法控制它，我相信您可能会以某种方式使用 Mockito 的能力来模拟 static 个对象。 我想出了以下编译，但我不确定它是否有效（因为我相信它依赖于一些字节码操作黑魔法，这可能会使您的测试套件变得脆弱，所以要小心）：

import scala.util.{Try, Using}
import java.time.{Clock, Instant, LocalDate, ZoneId}
import org.mockito.Mockito._

def withClock[A](clock: Clock)(f: => Unit): Try[Unit] = {
  Using(mockStatic(classOf[LocalDate])) { mocked =>
    mocked.when(() => LocalDate.now()).thenReturn(LocalDate.now(clock))
    f
  }
}

withClock(Clock.fixed(Instant.ofEpochMilli(42), ZoneId.of("UTC"))) {
  assert(LocalDate.now().toEpochDay() == 0)
}