如何限制 Java ServiceLoader 仅加载测试包中定义的服务提供者

Question

我有一个使用 ServiceLoader 的应用程序。
有一个服务接口 MyFancyService 及其实现：AppClassA、AppClassB 和测试实现：TestClassC、TestClassD

应用包中有META-INF.services，测试包中有另一个META-INF.services。

应用程序中的 META-INF.services 指向提供程序类 AppClassA 和 AppClassB
测试点中的 META-INF.services 到提供程序类 TestClassC 和 TestClassD

在运行应用程序时，ServiceLoader.load(MyFancyService.class)) 仅加载 AppClassA、AppClassB。显然，类路径中没有测试类，这就是原因。我明白这一点，这是可取的

在运行测试时，ServiceLoader.load(MyFancyService.java)) 会加载所有类 AppClassA、AppClassB、TestClassC、TestClassD。

有没有办法限制服务加载器在运行测试时只加载TestClassC、TestClassD？

Answer 1

当测试运行时，它在类路径中同时具有“测试”和“生产”类/资源。 这就是加载所有四个服务的原因，这是一种理想的行为，您显然需要 test 和 prod 类路径。

所以，归根结底，你有两个选择：

选项 1

使用类似 PowerMock/PowerMockito 的工具来模拟静态方法 ServiceLoader.load。我自己没有尝试过，但总的来说，除非它是某种遗留代码，否则我不推荐使用它。

选项 2

重构代码并模拟对服务加载器的调用，如下所示：

inteface MyServiceLoader {
  void loadMyStuff(Class<?> type);
}

class MyServiceLoaderImpl implements MyServiceLoader {
   public void loadMyStuff(Class<?> type() {
       ServiceLoader.load(type);
   }
}


// and now in ClassUnderTest:

public class ClassUnderTest {

    private MyServiceLoader myServiceLoader;

    public void thisIsAMethodToTest() {
      ...
      myServiceLoader.loadMyStuff(MyFancyInterface.class); 
    }
}

在测试中，您可以使用 mockito 模拟接口 MyServiceLoader.loadMyStuff 并仅加载测试实现。