当要模拟的方法有多个参数时,如何使用Jmock编写自定义约束

时间:2011-01-02 17:20:10

标签: java junit jmock jmockit

我在为一些大型代码库编写单元测试用例时遇到困难,我必须模拟很多类,以便我可以轻松地进行测试。我在Jmock的API文档中发现我可以使用的customerconstraint包含一个方法

eval(Object argo)

如果论证满足期望,这将返回true。

但我的方法是用多个参数调用的。如何评估参数并确保调用方法的参数是正确的。提前谢谢。

1 个答案:

答案 0 :(得分:3)

通常,创建与预期参数值相等的对象就足够了:

context.checking(new Expectations() {{
  allowing(calculator).add(1, 2);
  will(returnValue(3));

  DateTime loadTime = new DateTime(12);
  DateTime fetchTime = new DateTime(14);
  allowing(reloadPolicy).shouldReload(loadTime, fetchTime);
  will(returnValue(false));
}});

JMock还提供了一些预定义的约束:

context.checking(new Expectations() {{
  allowing(calculator).sqrt(with(lessThan(0));
  will(throwException(new IllegalArgumentException());
}});

您还可以使用with

使用自定义匹配器
context.checking(new Expectations() {{
  DateTime loadTime = new DateTime(12);
  allowing(reloadPolicy).shouldReload(with(equal(loadTime)), with(timeGreaterThan(loadTime));
  will(returnValue(false));
}});

此处timeGreaterThan可以定义为:

public class TimeGreaterThanMatcher extends TypeSafeMatcher<DateTime> {
    private DateTime minTime;

    public TimeGreaterThanMatcher(DateTime minTime) {
        this.minTime = minTime;
    }

    public boolean matchesSafely(DateTime d) {
        return d != null && minTime.isBefore(d);
    }

    public StringBuffer describeTo(Description description) {
        return description.appendText("a DateTime greater than ").appendValue(minTime);
    }

    public static Matcher<DateTime> timeGreaterThan(DateTime minTime) {
      return new TimeGreaterThanMatcher(minTime);
    }
}

有关详细信息,请参阅JMock Cookbook