如何在对子类进行单元测试时在抽象类中注入变量？

Question

我有一个抽象类BaseTemplate 和多个扩展它的类。在其中一个具体类（SmsTemplate extends BaseTemplate）中，我们有一个私有变量Gson。我们在抽象类中也有相同的私有变量（Gson）。

在对具体类进行单元测试时，抽象类中的方法会从具体类中调用。在我的单元测试中，我使用 Whitebox.setInternalState(smsTemplateObj, gsonObj); 将 Gson 对象注入到 SmsTemplate 和 BaseTemplate 的私有成员中，但 Gson 仅在子类中注入。在抽象类中，它的NULL，意思是没有注入。下面是实现。

有人能告诉我如何在抽象类中注入 Gson 对象吗？

abstract class BaseTemplate{

    private Gson gson;//Here its not getting injected

    protected String getContent(Content content){
        return gson.toJson(content); // ERROR - gson here throws NPE as its not injected
    }
}

class SmsTemplate extends BaseTemplate{

    private Gson gson;//Here its getting injected

    public String processTemplate(Content content){
        String strContent = getContent(content);
        ...
        ...
        gson.fromJson(strContent, Template.class);
    }
}

Answer 1

Whitebox.setInternalState() 方法只会设置它遇到的第一个字段的值，它会向上穿过您传递的对象的层次结构。因此，一旦它在您的子类中找到 gson 字段，它就不会进一步查找，也不会更改超类字段。

这种情况有两种解决方案：

• 更改变量名称。如果变量名称不同，您只需调用两次Whitebox.setInternalState()，每个变量调用一次。
• 使用反射手动设置字段。您也可以在没有 Mockito 帮助的情况下使用以下 sn-p 设置字段。

片段：

Field field = smsTemplateObj.getClass().getSuperclass().getDeclaredField("gson");
field.setAccesible(true);
field.set(smsTemplateObj, gsonObj);

Answer 2

您需要第二个抽象层：

abstract class BaseTemplate{

    // private Gson gson;// no Gson here

    protected String getContent(Content content){
        // do what you want without Gson
    }
}

abstract class BaseTemplateWithGson extends BaseTemplate{

    protected Gson gson;

    @Override
    protected String getContent(Content content){
        return gson.toJson(content);
    }
}

class SmsTemplate extends BaseTemplateWithGson {

    public String processTemplate(Content content){
        String strContent = getContent(content);
        ...
        ...
        gson.fromJson(strContent, Template.class);
    }
}