如何使用 EasyMock 模拟另一个方法调用的方法？

Question

我需要在 void 方法中模拟一个方法。

这是我的示例代码：

class MyClass {

    public MyClass(Session s, Boolean b1, Boolean b2)

    void myMethod(some paramaters...) {

        // some code
        int count= setSize();
    }

    int setSize() {

        // some calculation....
        return size;
    }

现在在我的测试类中，我想模拟 setSize() 以返回我自己的值，比如 300。

我确实喜欢：

MyClass mockclass = createNiceMock(MyClass.class);
EasyMock.expect(mockimplyZero.setBatchSize()).andReturn(Integer.valueOf(300));

mockclass.myMethod(parameters....)

当调用myMethod 时，它没有正确进入方法。 我认为这可能是 EasyMock 正在为 MyClass 构造函数设置默认值。如何正确模拟？

MyClass 中除了构造函数 myMethod 和 setSize 之外没有其他方法

Answer 1

您可以使用部分模拟来做到这一点。这是一个接近您的代码的示例。

首先是测试类。您将需要创建它的部分模拟。 getSize 应该被模拟，但 myMethod 应该被模拟，因为它是经过测试的方法。

此外，您经常需要调用构造函数来正确初始化类（经典模拟不会调用任何构造函数）。

class MyClass {

  private boolean b1;
  private boolean b2;

  public MyClass(boolean b1, boolean b2) {
    this.b1 = b1;
    this.b2 = b2;
  }

  int myMethod() {
    return getSize();
  }

  int getSize() {
    return 42;
  }

  public boolean getB1() {
    return b1;
  }

  public boolean getB2() {
    return b2;
  }
}

测试将如下所示

import org.junit.Test;

import static org.easymock.EasyMock.*;
import static org.junit.Assert.*;

public class MyClassTest {

  @Test
  public void test() {
    // Create a partial mock by calling its constructor
    // and only mocking getSize
    MyClass mock = createMockBuilder(MyClass.class)
        .withConstructor(true, true)
        .addMockedMethod("getSize")
        .createMock();

    // Record that getSize should return 8 (instead of 42)
    expect(mock.getSize()).andReturn(8);

    // All recording done. So put the mock in replay mode
    replay(mock);

    // Then, these assertions are to prove that the partial mock is 
    // actually doing what we expect. This is just to prove my point. Your
    // actual code will verify that myMethod is doing was is expected

    // 1. Verify that the constructor was correctly called
    assertEquals(true, mock.getB1());
    assertEquals(true, mock.getB2());
    // 2. Verify that getSize was indeed mocked 
    assertEquals(8, mock.myMethod());

    // Check everything expected was indeed called
    verify(mock);
  }
}

工作完成。请注意，这不一定是设计不佳的标志。我在测试Template method pattern时经常使用它。

Answer 2

您不应该在 同一个 类上测试另一个方法时模拟 一个 方法。理论上你可以做到这一点（使用 Mokito spy 例如）。

从这个意义上说，你在错误的层面上接近这个问题：你实际上不应该关心你的测试方法在你的测试类中调用了哪些其他方法。但是，如果您必须对测试进行调整，那么（例如）一种方法可以让您的测试代码在调用 mymethod() 之前设置该 size 字段。

或者：您将关注点分开，并将“大小”部分移动到它自己的类 X 中。然后您的测试类可以保存 X 的实例；然后可以模拟该实例。

长话短说：您想退后一步阅读一些有关如何使用 EasyMock 的教程。这不是你可以通过反复试验来学习的。