@Before
public void setupAlbums() {
- when(core.getAlbum(anyString())).thenAnswer(new Answer<Album>() {
+ when(core.getAlbum(anyString())).thenAnswer(new Answer<Optional<Album>>() {
@Override
- public Album answer(InvocationOnMock invocation)
+ public Optional<Album> answer(InvocationOnMock invocation)
throws Throwable {
- return albums.get(invocation.getArguments()[0]);
+ return Optional.fromNullable(albums.get(invocation.getArguments()[0]));
}
});
}