public void testGetPhotosViaMockServiceHandlesIOException() throws Exception {
Photo photo = (Photo) EasyMock.createStrictMock(Photo.class);
EasyMock.expect(photo.getId()).andReturn("dummyId1");
EasyMock.expect(photo.getId()).andReturn("dummyId2");
PhotosInterface photosInterface =
(PhotosInterface) EasyMock.createStrictMock(PhotosInterface.class);
EasyMock.expect(photosInterface.getInfo("31670708", null)).andReturn(photo);
EasyMock.expect(photosInterface.getSizes("dummyId1")).andThrow(new IOException());
EasyMock.expect(photosInterface.getInfo("31671077", null)).andReturn(photo);
EasyMock.expect(photosInterface.getSizes("dummyId2")).andThrow(new IOException());
Flickr flickr = (Flickr) EasyMock.createStrictMock(Flickr.class);
EasyMock.expect(flickr.getPhotosInterface()).andReturn(photosInterface);
EasyMock.replay(new Object[] {flickr, photosInterface, photo});
FlickrFacade flickrFacade = new FlickrFacade(flickr);
try {
List photos = flickrFacade.getPhotos(PHOTO_IDS_CSV);
assertNotNull(photos);
assertTrue(photos.size() == 0);
} catch (Exception e) {
fail("Thrown exception should've been handled by FlickrFacade: " + e);
}
EasyMock.verify(new Object[] {flickr, photosInterface, photo});
}
public void testGetPhotosetsViaMockService() throws Exception {
Photo photo1 = (Photo) EasyMock.createStrictMock(Photo.class);
EasyMock.expect(photo1.getId()).andReturn("dummyId1");
Photo photo2 = (Photo) EasyMock.createStrictMock(Photo.class);
EasyMock.expect(photo2.getId()).andReturn("dummyId2");
List photosetsPhotos1 = new ArrayList();
photosetsPhotos1.add(photo1);
List photosetsPhotos2 = new ArrayList();
photosetsPhotos2.add(photo2);
PhotosetsInterface photosetsInterface =
(PhotosetsInterface) EasyMock.createStrictMock(PhotosetsInterface.class);
EasyMock.expect(photosetsInterface.getPhotos("711101")).andReturn(photosetsPhotos1);
EasyMock.expect(photosetsInterface.getPhotos("711155")).andReturn(photosetsPhotos2);
PhotosInterface photosInterface =
(PhotosInterface) EasyMock.createStrictMock(PhotosInterface.class);
EasyMock.expect(photosInterface.getSizes("dummyId1")).andReturn(new ArrayList());
EasyMock.expect(photosInterface.getSizes("dummyId2")).andReturn(new ArrayList());
Flickr flickr = (Flickr) EasyMock.createStrictMock(Flickr.class);
EasyMock.expect(flickr.getPhotosetsInterface()).andReturn(photosetsInterface);
EasyMock.expect(flickr.getPhotosInterface()).andReturn(photosInterface);
EasyMock.replay(new Object[] {flickr, photosetsInterface, photosInterface, photo1, photo2});
FlickrFacade flickrFacade = new FlickrFacade(flickr);
List photos = flickrFacade.getPhotosFromPhotosets(PHOTOSET_IDS_CSV);
assertNotNull(photos);
assertTrue(photos.size() == 2);
EasyMock.verify(new Object[] {flickr, photosetsInterface, photosInterface, photo1, photo2});
}
@Test
public void testAllBaseJobsCompleted() throws Exception {
EasyMock.replay(mockFlow1, mockFlow2, props);
final JobManager factory = EasyMock.createStrictMock(JobManager.class);
EasyMock.replay(factory);
final IndividualJobExecutableFlow completedJob1 =
new IndividualJobExecutableFlow("blah", "blah", factory);
final IndividualJobExecutableFlow completedJob2 =
new IndividualJobExecutableFlow("blah", "blah", factory);
flow = new GroupedExecutableFlow("blah", completedJob1, completedJob2);
completedJob1.markCompleted();
completedJob2.markCompleted();
AtomicBoolean callbackWasCalled = new AtomicBoolean(false);
flow.execute(
props,
new OneCallFlowCallback(callbackWasCalled) {
@Override
public void theCallback(Status status) {
Assert.assertEquals(Status.SUCCEEDED, status);
}
});
Assert.assertTrue("Callback wasn't called!?", callbackWasCalled.get());
EasyMock.verify(factory);
}
@Before
public void setUp() throws Exception {
props = EasyMock.createStrictMock(Props.class);
mockFlow1 = EasyMock.createMock(ExecutableFlow.class);
mockFlow2 = EasyMock.createMock(ExecutableFlow.class);
EasyMock.expect(mockFlow1.getName()).andReturn("a").once();
EasyMock.expect(mockFlow2.getName()).andReturn("b").once();
EasyMock.expect(mockFlow1.getStatus()).andReturn(Status.READY).times(3);
EasyMock.expect(mockFlow2.getStatus()).andReturn(Status.READY).times(3);
EasyMock.expect(mockFlow1.getStartTime()).andReturn(null).once();
EasyMock.expect(mockFlow2.getStartTime()).andReturn(null).once();
EasyMock.expect(mockFlow1.getName()).andReturn("1").once();
EasyMock.expect(mockFlow2.getName()).andReturn("2").once();
EasyMock.replay(mockFlow1, mockFlow2, props);
flow = new GroupedExecutableFlow("blah", mockFlow1, mockFlow2);
Assert.assertEquals("1 + 2", flow.getName());
EasyMock.verify(mockFlow1, mockFlow2, props);
EasyMock.reset(mockFlow1, mockFlow2, props);
}
public void testGetPhotosViaMockService() throws Exception {
Photo photo = (Photo) EasyMock.createStrictMock(Photo.class);
EasyMock.expect(photo.getId()).andReturn("dummyId1");
EasyMock.expect(photo.getId()).andReturn("dummyId2");
PhotosInterface photosInterface =
(PhotosInterface) EasyMock.createStrictMock(PhotosInterface.class);
EasyMock.expect(photosInterface.getInfo("31670708", null)).andReturn(photo);
EasyMock.expect(photosInterface.getSizes("dummyId1")).andReturn(new ArrayList());
EasyMock.expect(photosInterface.getInfo("31671077", null)).andReturn(photo);
EasyMock.expect(photosInterface.getSizes("dummyId2")).andReturn(new ArrayList());
Flickr flickr = (Flickr) EasyMock.createStrictMock(Flickr.class);
EasyMock.expect(flickr.getPhotosInterface()).andReturn(photosInterface);
EasyMock.replay(new Object[] {flickr, photosInterface, photo});
FlickrFacade flickrFacade = new FlickrFacade(flickr);
List photos = flickrFacade.getPhotos(PHOTO_IDS_CSV);
assertNotNull(photos);
assertTrue(photos.size() == PHOTO_IDS_CSV.split(",").length);
EasyMock.verify(new Object[] {flickr, photosInterface, photo});
}
