SnapshotTaskClientImpl implements SnapshotTaskClient { public CompleteRestoreTaskResult completeRestore(String spaceId, int daysToExpire) throws ContentStoreException { CompleteRestoreTaskParameters taskParams = new CompleteRestoreTaskParameters(); taskParams.setSpaceId(spaceId); taskParams.setDaysToExpire(daysToExpire); String taskResult = contentStore.performTask(SnapshotConstants.COMPLETE_RESTORE_TASK_NAME, taskParams.serialize()); return CompleteRestoreTaskResult.deserialize(taskResult); } SnapshotTaskClientImpl(ContentStore contentStore); @Override CreateSnapshotTaskResult createSnapshot(String spaceId,
                                                   String description,
                                                   String userEmail); @Override GetSnapshotTaskResult getSnapshot(String snapshotId); @Override CleanupSnapshotTaskResult cleanupSnapshot(String spaceId); @Override CompleteSnapshotTaskResult completeSnapshot(String spaceId); @Override GetSnapshotListTaskResult getSnapshots(); @Override GetSnapshotContentsTaskResult getSnapshotContents(String snapshotId,
                                                             int pageNumber,
                                                             int pageSize,
                                                             String prefix); @Override GetSnapshotHistoryTaskResult getSnapshotHistory(String snapshotId,
                                                           int pageNumber,
                                                           int pageSize); @Override RestoreSnapshotTaskResult restoreSnapshot(String snapshotId,
                                                     String userEmail); @Override RequestRestoreSnapshotTaskResult requestRestoreSnapshot(String snapshotId,
                                                                   String userEmail); CompleteRestoreTaskResult completeRestore(String spaceId,
                                                     int daysToExpire); @Override GetRestoreTaskResult getRestore(String restoreId); @Override GetRestoreTaskResult getRestoreBySnapshot(String snapshotId);  }
@Test public void testCompleteRestore() throws Exception { String taskName = SnapshotConstants.COMPLETE_RESTORE_TASK_NAME; CompleteRestoreTaskResult preparedResult = new CompleteRestoreTaskResult(); preparedResult.setResult(completionResult); setupMock(taskName, preparedResult.serialize()); replayMocks(); CompleteRestoreTaskResult result = taskClient.completeRestore(spaceId, daysToExpire); assertThat(completionResult, equalTo(result.getResult())); }