@Test
public void testGetDataExceptions() throws Exception {
EasyMock.expect(uploadResource.getSize()).andStubReturn((long) Long.MAX_VALUE);
EasyMock.expect(uploadResource.getInputStream()).andStubAnswer(INPUT_STREAM_SUPPLIER);
environment.getControl().replay();
try {
uploadedFile.getData();
fail();
} catch (FileUploadException e) {
// expected - ignore
}
environment.getControl().reset();
EasyMock.expect(uploadResource.getSize()).andStubReturn(1L);
EasyMock.expect(uploadResource.getInputStream())
.andStubThrow(new IOException("No stream available"));
environment.getControl().replay();
try {
uploadedFile.getData();
fail();
} catch (FileUploadException e) {
// expected - ignore
assertTrue(e.getCause() instanceof IOException);
}
}
@Test
public void testResource() throws Exception {
EasyMock.expect(uploadResource.getSize()).andStubReturn((long) TEST_DATA.length());
EasyMock.expect(uploadResource.getInputStream()).andStubAnswer(INPUT_STREAM_SUPPLIER);
uploadResource.write(EasyMock.eq("output.png"));
EasyMock.expectLastCall();
uploadResource.delete();
EasyMock.expectLastCall();
environment.getControl().replay();
assertEquals(TEST_DATA.length(), uploadedFile.getSize());
assertNotNull(uploadedFile.getInputStream());
assertEquals(
TEST_DATA, new String(StreamUtils.toByteArray(uploadedFile.getInputStream()), "US-ASCII"));
assertEquals(TEST_DATA, new String(uploadedFile.getData(), "US-ASCII"));
uploadedFile.write("output.png");
uploadedFile.delete();
}
