新实例上的Mock方法

Question

我试图使用Mockito框架和Junit5创建JUnit测试用例。我正在处理下面的代码：

    Class ClasstToBeTested {
      FirstClass a = new FirstClass();

      public String methodToBeTested() {
         String str = a.firstMethod();
         return str;
      }
    }

   Class FirstClass {
      SecondClass b = new SecondClass();

      public String firstMethod() {
          String str = b.secondMethod();
          return str;
      }
   }

我有一个像上面这样的类结构，我需要模仿第二种方法。

我试过@spy on FirstClass并嘲笑SecondClass和secondMethod，但嘲笑没有发生。在这种情况下我怎么能嘲笑呢？

注：我不能改变班级结构。

Answer 1

你有一些选择：

1. （首选）使用IoC依赖注入来提供第二类实例，而不是在第一类中构建它：

  class FirstClass {
      private final SecondClass b;

      // Injecting the SecondClass instance
      FistClass(SecondClass b) {
          this.b = b;
      }

      public String firstMethod() {
          String str = b.secondMethod();
          return str;
      }
   }

然后你可以在测试中注入一个模拟。

   class FirstClass {
      SecondClass b = new SecondClass();

      // Annotate with a visibility for test annotation if available.
      // Here one can inject a mock too, but can cause problems if used inadvertently.
      void setSecondClassForTests(SecondClass b) {
         this.b = b;
      }

      public String firstMethod() {
          String str = b.secondMethod();
          return str;
      }
   }

然后在测试中调用setter并通过模拟。

final Field declaredField = instanceOfFirstClass.getClass().getDeclaredFields("b");
declaredField.setAccessible(true);
declaredField.set(instanceOfFirstClass, yourMockedInstance);