java - 如何测试一个方法内部调用了不同的方法？

Question

我在服务类中有一个名为 addSong(song,userId) 的单元测试方法。我从 Dao 类 调用其中的三个方法。我正在使用 Easy mock 来模拟 dao 类。在设置中，我首先模拟在 addSong(song,userId) 中调用的所有方法，然后调用 service.addsong(song,userId) 方法进行测试。

但我收到以下错误:

Java.lang.IllegalStateException: missing behavior definition for the preceding method call:
MusicPlayerDao.addSong(song)
Usage is: expect(a.foo()).andXXX()
    at org.easymock.internal.MockInvocationHandler.invoke(MockInvocationHandler.java:42)
    at org.easymock.internal.ObjectMethodsFilter.invoke(ObjectMethodsFilter.java:94)
    at org.easymock.internal.ClassProxyFactory$MockMethodInterceptor.intercept(ClassProxyFactory.java:97)
    at service.MusicPlayerDao$$EnhancerByCGLIB$$45bc3ca1.addSong(<generated>)
    at service.MusicPlayerServiceImpl.addSong(MusicPlayerServiceImpl.java:43)
    at AddSongTest.addSongs(AddSongTest.java:90)

这是我的代码:

private void addSongSetup() throws SQLException{
    this.album = new Album();
    album.setAlbumName("album");
    this.genre = new Genre();
    genre.setGenreName("genre");
    this.song = new Song("song",this.album,3,"artist","composer",this.genre);
    EasyMock.expect(this.dao.addSong(song)).andReturn(1).anyTimes();
    EasyMock.expect(this.dao.addGenre(genre, 1)).andReturn(1).anyTimes();
    EasyMock.expect(this.dao.addAlbum(album, 1)).andReturn(1).anyTimes();
    EasyMock.expect(this.dao.userIdSongsMapping(1,1)).andReturn(1).anyTimes();
}

@Test
public void addSongs(){

    this.album = new Album();
    album.setAlbumName("album");
    this.genre = new Genre();
    genre.setGenreName("genre");
    this.song = new Song("song",this.album,3,"artist","composer",this.genre);
    try {
        System.out.println(this.dao.addSong(song));
        boolean status = this.service.addSong(song, 1);
        assertEquals(true,status);
    } catch (Exception e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    }
}

我的服务类中的addSong方法:

public boolean addSong(Song song, int userId) throws Exception {

    MusicPlayerDaoInterface musicPlayerDao = MusicPlayerDao.getInstance();
    boolean status = false;
    int songId = 0;

    TransactionManager transactionManager = TransactionManagerImpl
            .getInstance();
    try {
        if (song != null) {
            if (song.getTitle() != null) {
                transactionManager.begin();
                songId = musicPlayerDao.addSong(song);
                song.setSongId(songId);
                if (song.getGenre() != null
                        && song.getGenre().getGenreName() != null) {
                    musicPlayerDao.addGenre(song.getGenre(),
                            song.getSongId());
                }
                if (song.getAlbum() != null
                        && song.getAlbum().getAlbumName() != null) {
                    musicPlayerDao.addAlbum(song.getAlbum(),
                            song.getSongId());
                }
                if (userId != 0 && songId != 0) {
                    musicPlayerDao.userIdSongsMapping(userId,
                            song.getSongId());
                }
                transactionManager.commit();
                status = true;
            }
        }
    } catch (SQLException | RollbackException | HeuristicMixedException
            | HeuristicRollbackException e) {
        transactionManager.rollback();
        status = false;
        throw e;

    }

    return status;
}

我不知道我是否走错了。请帮忙。

Answer 1

我认为您在记录预期行为后缺少 EasyMock.replay 语句。类似的东西

EasyMock.replay(this.dao);

来自EasyMock guide :

To get a Mock Object, we need to

1. create a Mock Object for the interface we would like to simulate
2. record the expected behavior
3. switch the Mock Object to replay state