@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(); }