如何使用JMock测试模拟方法中的模拟方法

时间:2011-04-13 20:35:13

标签: java junit mocking jmock

我无法确定如何模拟特定代码。

这是我的方法:

public void sendNotifications(NotificationType type, Person person)
    {
        List<Notification> notifications = findNotifications(type);
        for(Notification notification : notifications)
        {
            notification.send(person);
        }
    }

我希望能够使用JMock来测试findNotifications是否被调用,并且返回预期的值并调用send()。

findNotifications称我的Dao被嘲笑。通知是一个抽象类。

我有这样的单元测试,但它显然不起作用。它满足了前两个期望但不再满足。

    @Test
public void testSendNotifications2()
{
    final Person person = new Person();
    notificationEntries.add(createEntry1);

    Mockery context = new JUnit4Mockery() {{ setImposteriser(ClassImposteriser.INSTANCE); }};
    final NotificationDao dao = context.mock(NotificationDao.class, "notificationDao");
    final Notification mockedNotification = context.mock(V2ADTNotification.class, "mockNotification");
    notifications.add(mockedNotification);

    final NotifierServiceImpl mockedService = context.mock(NotifierServiceImpl.class, "mockedService");

    //NotifierService service = new NotifierServiceImpl(dao);

    context.checking(new Expectations() {
        {
            one(mockedService).setDao(dao);
            one(mockedService).sendNotifications(NotificationType.CREATE, person);
            one(mockedService).findNotifications(NotificationType.CREATE);
            one(dao).getByNotificationType(NotificationType.CREATE);
            will(returnValue(notificationEntries)); 
            will(returnValue(notifications));

            one(mockedNotification).send(person);
        }
    });

    mockedService.setDao(dao);
    mockedService.sendNotifications(NotificationType.CREATE, person);
    context.assertIsSatisfied();
}

    mockedService.sendNotifications(NotificationType.CREATE, person);
    context.assertIsSatisfied();
}

我怎么能按照我的意愿让它工作?

我试过的另一种方式。它满足前两个期望,但不满足发送期望。

@Test
public void testSendNotifications()
{
    final Person person = new Person();
    notificationEntries.add(createEntry1);

    Mockery context = new JUnit4Mockery() {{ setImposteriser(ClassImposteriser.INSTANCE); }};
    final NotificationDao dao = context.mock(NotificationDao.class, "notificationDao");
    final Notification mockedNotification = context.mock(V2ADTNotification.class, "mockNotification");

    NotifierService service = new NotifierServiceImpl(dao);

    context.checking(new Expectations() {
        {
            one(dao).getByNotificationType(NotificationType.CREATE);
            will(returnValue(notificationEntries)); 

            one(mockedNotification).send(person);
        }
    });

    service.sendNotifications(NotificationType.CREATE, person);
    context.assertIsSatisfied();
}

我很擅长使用Jmock所以我很抱歉,如果看起来我不知道我在做什么(我没有)。

1 个答案:

答案 0 :(得分:4)

  

在模拟方法中测试模拟方法。

你真的不能那样做。实际方法中的调用没有进行,因为你正在调用mock而不是真正的方法。

你的第一次尝试没有达到最后两个期望,因为那些使它们通过的调用仅在sendNotifications的实际实现中进行,而不是在你用它做的模拟中。

第二个更接近可行,但您需要将mockedNotification添加到您已设置为notificationEntries模拟返回的getByNotificationType列表中。