SnapshotTaskClientImpl implements SnapshotTaskClient { @Override public RequestRestoreSnapshotTaskResult requestRestoreSnapshot(String snapshotId, String userEmail) throws ContentStoreException { RestoreSnapshotTaskParameters taskParams = new RestoreSnapshotTaskParameters(); taskParams.setSnapshotId(snapshotId); taskParams.setUserEmail(userEmail); String taskResult = contentStore.performTask(SnapshotConstants.REQUEST_RESTORE_SNAPSHOT_TASK_NAME, taskParams.serialize()); return RequestRestoreSnapshotTaskResult.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 testRequestRestoreSnapshot() throws Exception { String taskName = SnapshotConstants.REQUEST_RESTORE_SNAPSHOT_TASK_NAME; RequestRestoreSnapshotTaskResult preparedResult = new RequestRestoreSnapshotTaskResult(); preparedResult.setDescription("success"); setupMock(taskName, preparedResult.serialize()); replayMocks(); RequestRestoreSnapshotTaskResult result = taskClient.requestRestoreSnapshot(snapshotId, userEmail); assertThat("success", equalTo(result.getDescription())); }