easymock教程-改变同一个方法调用的行为

阅读更多

    在easymock中，对于mock对象的同一个方法，可以为每一次的调用定制不同的行为。在record阶段easymock会精确的记录我们录入的行为，基于每一次的方法调用。

 

    这里有一个官网文档中的例子：

    expect(mock.voteForRemoval( " Document " ))
        .andReturn(( byte )  42 ).times( 3 )
        .andThrow( new  RuntimeException()).times( 4 )
        .andReturn(( byte )  - 42 );

    对于mock.voteForRemoval("Document")方法的调用，.andReturn((byte) 42).times(3) 表明前3次调用将返回42，.andThrow(new RuntimeException()).times(4)表示随后的4次调用(第4,5,6,7次)都将抛出异常，andReturn((byte) -42)表示第8次调用时将返回-42。

 

    我们来验证一下:

public   class  ChangeBehehaviorTest  extends  Assert  {
    private static interface Service {
        public byte voteForRemoval(String name);
    }

    @Test
    public void testExecute() {
        final String name = "Document";

        Service mock = EasyMock.createMock("service", Service.class);

        EasyMock.expect(mock.voteForRemoval("Document")).andReturn((byte) 42).times(3).andThrow(new RuntimeException())
                .times(4).andReturn((byte) -42);

        EasyMock.replay(mock);

        for (int i = 0; i < 3; i++) {
            assertEquals(42, mock.voteForRemoval(name));
        }
        for (int i = 0; i < 4; i++) {
            try {
                mock.voteForRemoval(name);
                fail("should throw RuntimeException");
            } catch (RuntimeException e) {

            }
        }
        assertEquals(-42, mock.voteForRemoval(name));
        EasyMock.verify(mock);
    }
}

    测试案例顺利通过。