使用 Java 反射访问测试用例中的受保护方法

Question

我正在尝试使用 Java 反射获取和调用驻留在不同类和不同包中的受保护方法。

包含受保护方法的类：

package com.myapp;

public class MyServiceImpl {

   protected List<String> retrieveItems(String status) {
         // Implementation
   }
}

调用类：

package xxx.myapp.tests;

import com.myapp.MyServiceImpl;

public class MyTestCase {

    List<String> items;

    public void setUp() throws Exception {

         MyServiceImpl service = new MyServiceImpl();
         Class clazz = service.getClass();

         // Fails at the next line:
         Method retrieveItems = clazz.getDeclaredMethod("retrieveItems");

         // How to invoke the method and return List<String> items?
         // tried this but it fails?
         retrieveItems.invoke(clazz, "S");
    }
}

编译器抛出这个异常：

java.lang.NoSuchMethodException: com.myapp.MyServiceImpl.retrieveItems()

Answer 1

如果您将测试用例放在同一个包中（com.myapp 而不是com.myapp.tests），他们将可以访问 protected（和 default 包级别）成员。

那你可以直接拨打service.retrieveMembers(status)。

如果您试图将源代码与测试分开，通常最好使用不同的源目录（例如src 目录和test 目录）。

Answer 2

您应该在调用方法中使用创建对象的链接而不是类的链接，并使用 Method.setAccessible(true) 调用来解锁访问：

public void setUp() throws Exception {
    MyServiceImpl service = new MyServiceImpl();
    Class<?> clazz = service.getClass();
    Method retrieveItems = clazz.getDeclaredMethod("retrieveItems", String.class);
    retrieveItems.setAccessible(true);
    items = (List<String>)retrieveItems.invoke(service, "S");
}

Answer 3

不需要反射或继承：

将您的MyTestCase 放在package com.myapp 下，因为“protected”范围也是“package”。 那么MyTestCase就可以访问MyServiceImpl的protected方法了。

Answer 4

为什么不简单地创建一个可以访问受保护方法的派生类，而不是使用那些棘手的反射东西？

更多想法请见Is it bad practice to use Reflection in Unit testing?。

Answer 5

您的代码的问题是getDeclaredMethod 函数通过名称和参数类型查找函数。随叫随到

Method retrieveItems = clazz.getDeclaredMethod("retrieveItems");

代码将查找不带参数的方法retrieveItems()。你正在寻找的方法确实需要一个参数，一个字符串，所以你应该调用

Method retrieveItems = clazz.getDeclaredMethod("retrieveItems", String.class);

这将告诉 Java 搜索 retrieveItems(String)，这就是您要查找的内容。