@BeforeMethod(groups = "unit")
public void init() {
support = new EventSupportImpl();
eventDispatcher = EasyMock.createMock(EventDispatcher.class);
messageSource = EasyMock.createMock(MessageSource.class);
collection.add(eventDispatcher);
collection.add(messageSource);
support.setEventDispatcher(eventDispatcher);
support.setMessageSource(messageSource);
event = null;
eventDispatcher.fireEvent(
EasyMockUtils.record(
new EasyMockUtils.Recorder<EventBase>() {
public void record(final EventBase object) {
event = object;
}
}));
EasyMock.expect(messageSource.getMessage("code", null)).andStubReturn("message");
collection.replay();
final User user = new User();
user.setCagridId(userId);
template = new EventBaseImpl();
template.setJobId(jobId);
template.setUserId(userId);
template.setJobName(jobName);
template.setWorkflowId(workflowId);
}
@Test(groups = "unit")
public void downloadSkipCheck() throws RemoteException {
final HasStreamInputContext file = EasyMock.createMock(HasStreamInputContext.class);
EasyMock.expect(accessProvider.getFile(fileId)).andReturn(file);
mockObjects.replay();
assert file.equals(service.download(fileId, gridId, false));
mockObjects.verifyAndReset();
}
@Test(groups = "unit")
public void delete() throws RemoteException {
EasyMock.expect(authorizationProvider.canDelete(fileId, gridId)).andReturn(true);
EasyMock.expect(accessProvider.deleteFile(fileId)).andReturn(true);
mockObjects.replay();
assert service.delete(fileId, gridId);
mockObjects.verifyAndReset();
}
@Test(groups = "unit")
public void canUpload() throws RemoteException {
EasyMock.expect(authorizationProvider.canUpload(fileId, gridId)).andReturn(false);
mockObjects.replay();
assert !service.canUpload(fileId, gridId);
mockObjects.verifyAndReset();
allowUpload();
mockObjects.replay();
assert service.canUpload(fileId, gridId);
mockObjects.verifyAndReset();
}
@Test(groups = "unit")
public void exists() throws RemoteException {
EasyMock.expect(accessProvider.fileExists(fileId)).andReturn(true);
mockObjects.replay();
assert service.exists(fileId);
mockObjects.verifyAndReset();
EasyMock.expect(accessProvider.fileExists(fileId)).andReturn(false);
mockObjects.replay();
assert !service.exists(fileId);
mockObjects.verifyAndReset();
}
@Test(groups = "unit")
public void canDownload() throws RemoteException {
expectCannotDownload();
mockObjects.replay();
assert !service.canDownload(fileId, gridId);
mockObjects.verifyAndReset();
expectCanDownload();
mockObjects.replay();
assert service.canDownload(fileId, gridId);
mockObjects.verifyAndReset();
}
@Test(groups = "unit", dataProvider = "bool1", dataProviderClass = TestNGDataProviders.class)
public void upload(final boolean shouldCommit) throws RemoteException {
handleShouldCommit(shouldCommit);
allowUpload();
mockObjects.replay();
final InputStream inputStream = new ByteArrayInputStream("moo".getBytes());
final UploadCallback callback = service.upload(fileId, gridId);
mockObjects.verifyAndReset();
final long length = 3L;
EasyMock.expect(accessProvider.putFile(EasyMock.eq(fileId), EasyMock.same(inputStream)))
.andReturn(length);
expectFinalizeAndReplay(shouldCommit, length);
callback.onUpload(inputStream);
mockObjects.verifyAndReset();
}
@Test(groups = "unit")
public void length() throws RemoteException {
expectCanDownload();
EasyMock.expect(accessProvider.getFileMetadata(fileId)).andReturn(new FileMetadata(12L, 13L));
mockObjects.replay();
assert service.getLength(fileId, gridId) == 13L;
}
@BeforeMethod(groups = "unit")
public void init() {
factory = new RawExtractJobFactoryImpl();
stagingDirectoryFactory = EasyMock.createMock(CredentialedStagingDirectoryFactory.class);
stagingDirectory = EasyMock.createMock(StagingDirectory.class);
EasyMock.expect(stagingDirectory.getAbsolutePath()).andStubReturn(PATH);
EasyMock.expect(stagingDirectory.getSep()).andStubReturn("/");
final Supplier<DisposableResourceTracker> trackerSupplier = EasyMockUtils.createMockSupplier();
tracker = createMock(DisposableResourceTracker.class);
expect(trackerSupplier.get()).andReturn(tracker);
replay(trackerSupplier);
rawPopulator = EasyMock.createMock(InputContext.class);
converter = EasyMock.createMock(DTAToMzXMLConverter.class);
options = new DTAToMzXMLOptions();
factory.setDtaToMzXMLConverter(converter);
factory.setDtaToMxXMLOptions(options);
factory.setDisposableResourceTrackerSupplier(trackerSupplier);
factory.setCredentialedStagingDirectoryFactory(stagingDirectoryFactory);
mockObjects =
MockObjectCollection.fromObjects(
stagingDirectoryFactory, stagingDirectory, tracker, rawPopulator, converter);
}
protected void postProcessAndVerify() {
try {
job.postprocess(completedNormally);
} finally {
mockObjects.verifyAndReset();
}
}
private void expectFinalizeAndReplay(final boolean shouldCommit, final long length) {
fileService.recordLength(fileId, length);
if (shouldCommit) {
fileService.commit(fileId);
}
mockObjects.replay();
}
@Test(groups = "unit", dataProvider = "bool1", dataProviderClass = TestNGDataProviders.class)
public void uploadStream(final boolean shouldCommit) throws IOException {
handleShouldCommit(shouldCommit);
allowUpload();
final ByteArrayOutputStream underlyingOutputStream = new ByteArrayOutputStream();
EasyMock.expect(accessProvider.getPutFileOutputStream(fileId))
.andReturn(underlyingOutputStream);
mockObjects.replay();
final OutputStream outputStream = service.prepareUploadStream(fileId, gridId);
mockObjects.verifyAndReset();
expectFinalizeAndReplay(shouldCommit, 3L);
InputContexts.forString("moo").get(outputStream);
outputStream.close();
mockObjects.verifyAndReset();
assert new String(underlyingOutputStream.toByteArray()).equals("moo");
}
@Test(groups = "unit", expectedExceptions = RuntimeException.class)
public void getMetadatasException() throws RemoteException {
EasyMock.expect(
authorizationProvider.canDownloadAll(
EasyMock.aryEq(new String[] {fileId, fileId2}), EasyMock.eq(gridId)))
.andReturn(false);
mockObjects.replay();
service.getFileMetadata(Lists.newArrayList(fileId, fileId2), gridId);
}
@Test(groups = "unit")
public void getMetadatas() {
EasyMock.expect(
authorizationProvider.canDownloadAll(
EasyMock.aryEq(new String[] {fileId, fileId2}), EasyMock.eq(gridId)))
.andReturn(true);
EasyMock.expect(accessProvider.getFileMetadata(fileId)).andReturn(new FileMetadata(13L, 12L));
EasyMock.expect(accessProvider.getFileMetadata(fileId2)).andReturn(new FileMetadata(14L, 11L));
mockObjects.replay();
final List<FileMetadata> metadatas =
service.getFileMetadata(Lists.newArrayList(fileId, fileId2), gridId);
assert metadatas.get(0).getLength() == 12L;
assert metadatas.get(1).getLength() == 11L;
}
@BeforeMethod(groups = "unit")
public void init() {
service = new StorageManagerImpl();
gridId = UUID.randomUUID().toString();
fileId = UUID.randomUUID().toString();
fileId2 = UUID.randomUUID().toString();
accessProvider = EasyMock.createMock(AccessProvider.class);
authorizationProvider = EasyMock.createMock(AuthorizationProvider.class);
fileService = EasyMock.createMock(FileService.class);
service.setAccessProvider(accessProvider);
service.setAuthorizationProvider(authorizationProvider);
service.setFileService(fileService);
mockObjects =
MockObjectCollection.fromObjects(accessProvider, authorizationProvider, fileService);
}
@AfterMethod(groups = "unit")
public void verifyDispatched() {
collection.verifyAndReset();
verifyEvent();
}
protected void expectPreprocessingAndReplayMocks() {
expectPreprocessingIfNeeded();
expectCleanUp();
mockObjects.replay();
}
@Test(groups = "unit", expectedExceptions = RuntimeException.class)
public void uploadAccessException() throws RemoteException {
EasyMock.expect(authorizationProvider.canUpload(fileId, gridId)).andReturn(false);
mockObjects.replay();
service.upload(fileId, gridId);
}
@Test(groups = "unit", expectedExceptions = RuntimeException.class)
public void dateModifiedAccessException() throws RemoteException {
expectCannotDownload();
mockObjects.replay();
service.getDateModified(fileId, gridId);
}
