@Mock public BackupVerificationResult isMetaFileValid(AbstractBackupPath metaBackupPath) { return getBackupVerificationResult(); } }
@Mock NodeProbe instance(IConfiguration config) { return nodeProbe; } };
@RunWith(JMockit.class) public class PriamConfigTest { private @Mocked PriamServer priamServer;
@Mock public boolean isTruncated() { return truncated; } }
@Test public void isReplaceToken(@Mocked final InstanceIdentity identity) { new Expectations() { { priamServer.getInstanceIdentity(); result = identity; identity.isReplace(); result = true; } }; Response response = resource.isReplaceToken(); assertEquals(200, response.getStatus()); assertEquals("true", response.getEntity()); }
@Mock public void setBucketLifecycleConfiguration( String bucketName, BucketLifecycleConfiguration bucketLifecycleConfiguration) { bconf = bucketLifecycleConfiguration; }
@Test public void getReplacedAddress(@Mocked final InstanceIdentity identity) { final String replacedIp = "127.0.0.1"; new Expectations() { { priamServer.getInstanceIdentity(); result = identity; identity.getReplacedIp(); result = replacedIp; } }; Response response = resource.getReplacedIp(); assertEquals(200, response.getStatus()); assertEquals(replacedIp, response.getEntity()); }
@Mock public BackupVerificationResult isMetaFileValid(AbstractBackupPath metaBackupPath) { return getBackupVerificationResult(); } }
@Mock public void $init(AmazonS3 client, DataPart dp, List<PartETag> partETags) { MockS3PartUploader.partETags = partETags; }
@Mock public List<S3ObjectSummary> getObjectSummaries() { if (firstcall) { firstcall = false; if (simfilter) return getObjectSummaryEmpty(); return getObjectSummary(); } else { if (simfilter) { simfilter = false; // reset return getObjectSummaryEmpty(); } else truncated = false; return getNextObjectSummary(); } }
@Mock public void clearSnapshot(String tag, String... keyspaces) { cleanup(new File("target/data")); } }
@Mock public CompleteMultipartUploadResult completeUpload() throws BackupRestoreException { ++compattempts; if (completionFailure) throw new BackupRestoreException("Test exception"); return null; }
@Mock public DeleteObjectsResult deleteObjects(DeleteObjectsRequest var1) throws SdkClientException, AmazonServiceException { if (emulateError) throw new AmazonServiceException("Unable to reach AWS"); return null; } }
@Mock public InitiateMultipartUploadResult initiateMultipartUpload( InitiateMultipartUploadRequest initiateMultipartUploadRequest) throws AmazonClientException { return new InitiateMultipartUploadResult(); }
@Mock public Void retriableCall() throws AmazonClientException, BackupRestoreException { logger.info("MOCK UPLOADING..."); return uploadPart(); }
@Mock private Void uploadPart() throws AmazonClientException, BackupRestoreException { ++partAttempts; if (partFailure) throw new BackupRestoreException("Test exception"); partETags.add(new PartETag(0, null)); return null; }
@Mock public ObjectListing listObjects(ListObjectsRequest listObjectsRequest) throws AmazonClientException { ObjectListing listing = new ObjectListing(); listing.setBucketName(listObjectsRequest.getBucketName()); listing.setPrefix(listObjectsRequest.getPrefix()); return listing; }
@Mock public ObjectListing listNextBatchOfObjects(ObjectListing previousObjectListing) throws AmazonClientException { ObjectListing listing = new ObjectListing(); listing.setBucketName(previousObjectListing.getBucketName()); listing.setPrefix(previousObjectListing.getPrefix()); return new ObjectListing(); } }
@Mock public BucketLifecycleConfiguration getBucketLifecycleConfiguration(String bucketName) { List<BucketLifecycleConfiguration.Rule> rules = Lists.newArrayList(); if (ruleAvailable) { String clusterPath = "casstestbackup/" + region + "/fake-app/"; BucketLifecycleConfiguration.Rule rule = new BucketLifecycleConfiguration.Rule() .withExpirationInDays(5) .withPrefix(clusterPath); rule.setStatus(BucketLifecycleConfiguration.ENABLED); rule.setId(clusterPath); rules.add(rule); } bconf.setRules(rules); return bconf; }