@Test public void testUploadPart() throws IOException { Entity entity = mock(Entity.class); Response response = mock(Response.class); when(response.getHeaders()).thenReturn(ImmutableMap.of("ETag", "etag1234567890")); when(invoker.invoke( Request.builder() .method(Method.PUT) .endpoint(endpoint) .bucket(bucket) .key(key) .parameter("uploadId", uploadId) .parameter("partNumber", "10") .header("Content-MD5", "MD") .header("Date", HttpDateTimeFormatter.formatDateTime(now)) .entity(entity) .build(), Response.class)) .thenReturn(response); UploadPartResult actual = client.uploadPart( bucket, key, uploadId, 10, new UploadPartRequest().withContentMD5("MD").withEntity(entity)); UploadPartResult expected = new UploadPartResult("etag1234567890"); assertEquals(expected, actual); verify(response).close(); }
@Test public void testInitiateMultipartUpload() { ObjectCreation objectCreation = new ObjectCreation() .withStorageClass(StorageClass.REDUCED_REDUNDANCY) .withContentType("test/aaa") .withExpires("xyz"); InitiateMultipartUploadResult expected = mock(InitiateMultipartUploadResult.class); when(invoker.invoke( Request.builder() .method(Method.POST) .endpoint(endpoint) .bucket(bucket) .key(key) .subResource("uploads") .header("x-snda-storage-class", "REDUCED_REDUNDANCY") .header("Content-Type", "test/aaa") .header("Expires", "xyz") .header("Date", HttpDateTimeFormatter.formatDateTime(now)) .build(), InitiateMultipartUploadResult.class)) .thenReturn(expected); InitiateMultipartUploadResult actual = client.initiateMultipartUpload(bucket, key, objectCreation); assertSame(expected, actual); }
@Test public void testCopyPart() throws IOException { CopyPartResult expected = mock(CopyPartResult.class); when(invoker.invoke( Request.builder() .method(Method.PUT) .endpoint(endpoint) .bucket(bucket) .key(key) .parameter("uploadId", uploadId) .parameter("partNumber", "250") .header("x-snda-copy-source", "mybucket/key1") .header("x-snda-copy-source-range", "bytes=200-500") .header("Date", HttpDateTimeFormatter.formatDateTime(now)) .build(), CopyPartResult.class)) .thenReturn(expected); CopyPartResult actual = client.copyPart( bucket, key, uploadId, 250, new CopyPartRequest() .withCopySource(new CopySource("mybucket", "key1")) .withCopySourceRange(new Range(200, 500))); assertSame(expected, actual); }
@Test public void testListParts() { ListPartsResult expected = mock(ListPartsResult.class); when(invoker.invoke( Request.builder() .method(Method.GET) .endpoint(endpoint) .bucket(bucket) .key(key) .parameter("uploadId", uploadId) .parameter("part-number-marker", 255) .header("Date", HttpDateTimeFormatter.formatDateTime(now)) .build(), ListPartsResult.class)) .thenReturn(expected); ListPartsResult actual = client.listParts(bucket, key, uploadId, new ListPartsCriteria().withPartNumberMarker(255)); assertSame(expected, actual); }
@Test @SuppressWarnings("unchecked") public void testCompleteMultipartUpload() { List<Part> parts = mock(List.class); CompleteMultipartUploadResult expected = mock(CompleteMultipartUploadResult.class); when(invoker.invoke( Request.builder() .method(Method.POST) .endpoint(endpoint) .bucket(bucket) .key(key) .parameter("uploadId", uploadId) .header("Date", HttpDateTimeFormatter.formatDateTime(now)) .entity(new CompleteMultipartUpload(parts)) .build(), CompleteMultipartUploadResult.class)) .thenReturn(expected); CompleteMultipartUploadResult actual = client.completeMultipartUpload(bucket, key, uploadId, parts); assertSame(expected, actual); }