@Test public void testSaveAndRestoreWithObject() { EasyMock.expect(storageRpcMock.continueRewrite(RESPONSE_WITH_OBJECT)) .andReturn(RESPONSE_WITH_OBJECT); EasyMock.expect(storageRpcMock.continueRewrite(RESPONSE_WITH_OBJECT)) .andReturn(RESPONSE_WITH_OBJECT_DONE); EasyMock.replay(storageRpcMock); copyWriter = new CopyWriter(options, RESPONSE_WITH_OBJECT); copyWriter.copyChunk(); assertTrue(!copyWriter.isDone()); assertEquals(21L, copyWriter.getTotalBytesCopied()); assertEquals(42L, copyWriter.getBlobSize()); RestorableState<CopyWriter> rewriterState = copyWriter.capture(); CopyWriter restoredRewriter = rewriterState.restore(); assertEquals(result, restoredRewriter.getResult()); assertTrue(restoredRewriter.isDone()); assertEquals(42L, restoredRewriter.getTotalBytesCopied()); assertEquals(42L, restoredRewriter.getBlobSize()); }
@Test public void testSaveAndRestoreWithoutObject() { EasyMock.expect(storageRpcMock.continueRewrite(RESPONSE_WITHOUT_OBJECT)) .andReturn(RESPONSE_WITHOUT_OBJECT); EasyMock.expect(storageRpcMock.continueRewrite(RESPONSE_WITHOUT_OBJECT)) .andReturn(RESPONSE_WITHOUT_OBJECT_DONE); EasyMock.replay(storageRpcMock); copyWriter = new CopyWriter(options, RESPONSE_WITHOUT_OBJECT); copyWriter.copyChunk(); assertTrue(!copyWriter.isDone()); assertEquals(21L, copyWriter.getTotalBytesCopied()); assertEquals(42L, copyWriter.getBlobSize()); RestorableState<CopyWriter> rewriterState = copyWriter.capture(); CopyWriter restoredRewriter = rewriterState.restore(); assertEquals(result, restoredRewriter.getResult()); assertTrue(restoredRewriter.isDone()); assertEquals(42L, restoredRewriter.getTotalBytesCopied()); assertEquals(42L, restoredRewriter.getBlobSize()); }
@Test public void testSaveAndRestoreWithResult() { EasyMock.expect(storageRpcMock.continueRewrite(RESPONSE_WITH_OBJECT)) .andReturn(RESPONSE_WITH_OBJECT_DONE); EasyMock.replay(storageRpcMock); copyWriter = new CopyWriter(options, RESPONSE_WITH_OBJECT); copyWriter.copyChunk(); assertEquals(result, copyWriter.getResult()); assertTrue(copyWriter.isDone()); assertEquals(42L, copyWriter.getTotalBytesCopied()); assertEquals(42L, copyWriter.getBlobSize()); RestorableState<CopyWriter> rewriterState = copyWriter.capture(); CopyWriter restoredRewriter = rewriterState.restore(); assertEquals(result, restoredRewriter.getResult()); assertTrue(restoredRewriter.isDone()); assertEquals(42L, restoredRewriter.getTotalBytesCopied()); assertEquals(42L, restoredRewriter.getBlobSize()); } }