java - 未调用模拟方法

Question

我正在将 testng 与 Mockito 结合使用，由于限制，我无法使用 PowerMock。当我尝试运行测试时，由于某种原因，调用了真正的方法而不是模拟方法!我在具有静态方法的类周围添加了一个包装类。我从文档中知道 Mockito 不支持静态方法。因此，作为解决方法，我编写了一个简单的包装器，它只调用该类中的静态方法。

public class BatTrafficManagerStatsHelperTest {

    @Mock
    private MtsConfFactoryWrapper mtsConfFactoryWrapper = new MtsConfFactoryWrapper();

    private static final String EXPECTED_JSON_STR = "{"
            + "         \"duration\": 100,"
            + "         \"timestamp\": \"1970-01-01T00:00:00\","
            + "         \"cellMetrics\": {"
            + "           \"enb1_cell1\": {"
            + "             \"handoverAttempts\": 2,"
            + "             \"rrcConnectionAttempts\": 1"
            + "           }"
            + "         },"
            + "         \"trafficProfileMetrics\": {"
            + "           \"voipSpeech_2PDN\": {"
            + "             \"srMtSuccess\": 4,"
            + "             \"numberOfUEs\": 100,"
            + "             \"srMoSuccess\": 3,"
            + "             \"attachSuccess\": 5,"
            + "             \"rrcConnectionAttempts\": 1"
            + "           }"
            + "         },"
            + "         \"ueGroupMetrics\": {"
            + "           \"ueGroup1\": {"
            + "             \"handoverAttempts\": 2,"
            + "             \"numberOfUEs\": 100,"
            + "             \"rrcConnectionAttempts\": 1"
            + "           }"
            + "         }"
            + "       }";

    @BeforeMethod
    public void setUp() {
        MockitoAnnotations.initMocks(this);
    }

    /**
     * getBstmString()
     * Setup mock, tests getBstmString() and verify result
     *
     * @throws Exception
     */
    @Test
    public void testGetBtmString() throws Exception {
        final Date date = Mockito.mock(Date.class);
        Mockito.when(date.getTime()).thenReturn(0L);

        final Map<OptimizedRoute, List<StatsQueue>> ueStatsTrackerMock = getUeStatsTrackerMock();
        final Map<Enb, List<OptimizedCellStatsQueue>> cellStatsTrackerMock = getCellStatsTrackerMock();
        final MtsConfBuilder confBuilderMock = Mockito.mock(MtsConfBuilder.class);
        final Tracker trackerMock = Mockito.mock(Tracker.class);

        Mockito.when(mtsConfFactoryWrapper.getConfBuilderr()).thenReturn(confBuilderMock);
        Mockito.when(confBuilderMock.getTracker()).thenReturn(trackerMock);
        Mockito.when(trackerMock.getUeStatsTracker()).thenReturn(
                ueStatsTrackerMock);
        Mockito.when(trackerMock.getCellStatsTracker()).thenReturn(
                cellStatsTrackerMock);
        Mockito.when(trackerMock.getConfiguredTrafficProfiles())
                .thenReturn(
                        getFakeTrafficProfilesList());

        BatTrafficManagerStatsHelper btmStatsHelper = new BatTrafficManagerStatsHelper(date);
        String result = btmStatsHelper.getBtmString(TIME).replace("JsonDataForBatTrafficManagerRegex:", "");

        JsonParser parser = new JsonParser();
        assertEquals(parser.parse(result), parser.parse(EXPECTED_JSON_STR));
    }

}

被测试的原始类如下所示

public class BatTrafficManagerStatsHelper {

    private final Date date;

    /*
     * Constants for BAT Traffic Manager log parsing
     */
    private static final ImmutableMap<String, String> AHL_STATS_CONSTANTS_TO_BTM_CONSTANTS = ImmutableMap.of(
            BatAeroflexHLStatsConstants.RRC_CONNECTION_REQUEST_ATTEMPTS, "rrcConnectionAttempts",
            BatAeroflexHLStatsConstants.HANDOVER_ATTEMPTS, "handoverAttempts",
            BatAeroflexHLStatsConstants.SERVICE_REQ_MO_SUCCESS, "srMoSuccess",
            BatAeroflexHLStatsConstants.SERVICE_REQ_MT_SUCCESS, "srMtSuccess",
            BatAeroflexHLStatsConstants.ATTACH_SUCCESS, "attachSuccess");
    private static final String NUMBER_OF_UES = "numberOfUEs";

    /**
     * Constructor
     *
     * @param date
     */
    public BatTrafficManagerStatsHelper(Date date) {
        this.date = new Date(date.getTime());
    }

    /**
     * Extracts log data and returns it in a format suited
     * for parsing in {@literal BAT Traffic Manager}.
     *
     * @param elapsedTime
     * @return log data for BAT Traffic Manager
     */
    public String getBtmString(long elapsedTime) {
        MtsConfFactoryWrapper mtsConfFactoryWrapper = new MtsConfFactoryWrapper();
        Map<OptimizedRoute, List<StatsQueue>> ueGroupTracker = mtsConfFactoryWrapper.getConfBuilderr().getTracker()
                .getUeStatsTracker();
        Map<Enb, List<OptimizedCellStatsQueue>> cellTracker = mtsConfFactoryWrapper.getConfBuilderr().getTracker()
                .getCellStatsTracker();

        BatTrafficManagerStatsHelperJson json = new BatTrafficManagerStatsHelperJsonBuilder()
                .setDuration(elapsedTime)
                .setTimestamp(getTimeStamp())
                .setCellMetrics(getCellMetrics(cellTracker))
                .setTrafficProfileMetrics(getTrafficProfileMetrics(ueGroupTracker))
                .setUeGroupMetrics(getUeGroupMetrics(ueGroupTracker))
                .build();

        return "JsonDataForBatTrafficManagerRegex:" + getJsonString(json);
    }

}

public class MtsConfFactoryWrapper  {
    public MtsConfFactoryWrapper() {}

    public MtsConfBuilder getConfBuilderr() {
        return MtsConfFactory.getConfBuilder();
    }
}

Answer 1

在您的测试类中使用它，

@Mock
private MtsConfFactoryWrapper mtsConfFactoryWrapper;

而不是

@Mock
private MtsConfFactoryWrapper mtsConfFactoryWrapper = new MtsConfFactoryWrapper();

您不应将新对象分配给 mtsConfFactoryWrapper 变量。当调用 MockitoAnnotations.initMocks(this); 时，模拟对象将被分配给 mtsConfFactoryWrapper 变量。

在您的方法中，您总是创建一个新对象，如下所示，

MtsConfFactoryWrapper mtsConfFactoryWrapper = new MtsConfFactoryWrapper();

因此，即使您在测试中模拟了该对象，当调用该方法时，它也会再次创建一个新对象。尝试将其注入(inject)到类中，而不是在方法中创建新实例。

你的类应该是这样的，

public class BatTrafficManagerStatsHelper {

    private MtsConfFactoryWrapper mtsConfFactoryWrapper;

    public BatTrafficManagerStatsHelper(MtsConfFactoryWrapper mtsConfFactoryWrapper) {
        this.mtsConfFactoryWrapper = mtsConfFactoryWrapper;
    }

    // other stuffs
}

在测试类中创建 BatTrafficManagerStatsHelper 对象时，将模拟对象传递给构造函数，如下所示，

public class BatTrafficManagerStatsHelperTest {
    @Test
    public void testGetBtmString() throws Exception {
        //your code

        BatTrafficManagerStatsHelper btmStatsHelper = new BatTrafficManagerStatsHelper(passYourMockedObjectHere);
    }
}