当单元测试子类时，如何在抽象类中注入变量？

Question

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

当单元测试具体类时，抽象类中的方法从具体类中被调用。在我的Unit测试中，我使用Whitebox.setInternalState（smsTemplateObj， gsonObj）；将Gson对象注入到SmsTem板和BaseTem板的私有成员中，但是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

您需要第二个抽象层：

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);
    }
}

Answer 2

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

对于这种情况，有两种解决方案：

• 更改变量名称。如果变量有不同的名称，您可以简单地调用Whitebox.setInternalState（）两次，每个变量一个。
• 使用反射手动设置字段。您也可以在没有Mockito帮助的情况下使用以下片段设置字段。

片段：

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