Added expect test for putBlob with reduced redundancy

This commit is contained in:
andreisavu 2012-01-14 14:05:17 +02:00
parent f27bed524c
commit 20f5205727
3 changed files with 68 additions and 3 deletions

View File

@ -33,6 +33,7 @@ import org.jclouds.aws.s3.binders.AssignCorrectHostnameAndBindAsHostPrefixIfConf
import org.jclouds.http.RequiresHttp; import org.jclouds.http.RequiresHttp;
import org.jclouds.location.Region; import org.jclouds.location.Region;
import org.jclouds.rest.ConfiguresRestClient; import org.jclouds.rest.ConfiguresRestClient;
import org.jclouds.rest.RestContext;
import org.jclouds.s3.Bucket; import org.jclouds.s3.Bucket;
import org.jclouds.s3.S3AsyncClient; import org.jclouds.s3.S3AsyncClient;
import org.jclouds.s3.S3Client; import org.jclouds.s3.S3Client;
@ -84,4 +85,15 @@ public class AWSS3RestClientModule extends S3RestClientModule<AWSS3Client, AWSS3
return in; return in;
} }
/**
* so that we can inject RestContext<S3Client, S3AsyncClient>
*/
@SuppressWarnings("unchecked")
@Singleton
@Provides
RestContext<S3Client, S3AsyncClient>
provideBaseContext(RestContext<AWSS3Client, AWSS3AsyncClient> in) {
return (RestContext) in;
}
} }

View File

@ -18,9 +18,13 @@
*/ */
package org.jclouds.aws.s3; package org.jclouds.aws.s3;
import com.google.common.collect.ImmutableMultimap;
import org.jclouds.aws.s3.internal.BaseAWSS3ClientExpectTest; import org.jclouds.aws.s3.internal.BaseAWSS3ClientExpectTest;
import org.jclouds.blobstore.domain.Blob;
import org.jclouds.http.HttpRequest; import org.jclouds.http.HttpRequest;
import org.jclouds.http.HttpResponse; import org.jclouds.http.HttpResponse;
import org.jclouds.io.payloads.StringPayload;
import org.jclouds.s3.blobstore.functions.BlobToObject;
import org.testng.annotations.Test; import org.testng.annotations.Test;
import java.net.URI; import java.net.URI;
@ -31,6 +35,7 @@ import static org.jclouds.s3.domain.ObjectMetadata.StorageClass;
/** /**
* @author Andrei Savu * @author Andrei Savu
*/ */
@Test
public class AWSS3ClientExpectTest extends BaseAWSS3ClientExpectTest { public class AWSS3ClientExpectTest extends BaseAWSS3ClientExpectTest {
@Test @Test
@ -38,13 +43,35 @@ public class AWSS3ClientExpectTest extends BaseAWSS3ClientExpectTest {
AWSS3Client client = requestSendsResponse( AWSS3Client client = requestSendsResponse(
HttpRequest.builder() HttpRequest.builder()
.method("PUT") .method("PUT")
.endpoint(URI.create("http://test.s3.amazon.com/")) .endpoint(URI.create("https://test.s3.amazonaws.com/test"))
.headers(ImmutableMultimap.of(
"x-amz-storage-class", "REDUCED_REDUNDANCY",
"Host", "test.s3.amazonaws.com",
"Date", CONSTANT_DATE,
"Authorization", "AWS identity:1mJrW85/mqZpYTFIK5Ebtt2MM6E="
))
.payload(new StringPayload("content"))
.build(), .build(),
HttpResponse.builder() HttpResponse.builder()
.statusCode(200) .statusCode(200)
.headers(ImmutableMultimap.of(
"x-amz-id-2", "w0rL+9fALQiCOToesVQefs8WalIgn+ZhMD7hHMKYud/xv7MyKkAWQOtFNEfK97Ri",
"x-amz-request-id", "7A84C3CD4437A4C0",
"Date", CONSTANT_DATE,
"ETag", "437b930db84b8079c2dd804a71936b5f",
"Server", "AmazonS3"
))
.build() .build()
); );
client.putObject("test", null, storageClass(StorageClass.REDUCED_REDUNDANCY)); Blob blob = blobStore.blobBuilder("test").payload("content").build();
BlobToObject blobToObject = getInstance(BlobToObject.class);
client.putObject("test", blobToObject.apply(blob),
storageClass(StorageClass.REDUCED_REDUNDANCY));
}
public <T> T getInstance(Class<T> klass) {
return blobStoreContext.utils().injector().getInstance(klass);
} }
} }

View File

@ -18,15 +18,25 @@
*/ */
package org.jclouds.aws.s3.internal; package org.jclouds.aws.s3.internal;
import com.google.common.base.Function;
import com.google.common.base.Supplier; import com.google.common.base.Supplier;
import com.google.common.collect.ImmutableSet;
import com.google.inject.Module; import com.google.inject.Module;
import org.jclouds.aws.s3.AWSS3Client; import org.jclouds.aws.s3.AWSS3Client;
import org.jclouds.aws.s3.config.AWSS3RestClientModule; import org.jclouds.aws.s3.config.AWSS3RestClientModule;
import org.jclouds.blobstore.BlobStore;
import org.jclouds.blobstore.BlobStoreContext;
import org.jclouds.blobstore.BlobStoreContextFactory;
import org.jclouds.date.TimeStamp; import org.jclouds.date.TimeStamp;
import org.jclouds.http.HttpRequest;
import org.jclouds.http.HttpResponse;
import org.jclouds.http.RequiresHttp; import org.jclouds.http.RequiresHttp;
import org.jclouds.logging.config.NullLoggingModule;
import org.jclouds.rest.BaseRestClientExpectTest; import org.jclouds.rest.BaseRestClientExpectTest;
import org.jclouds.rest.ConfiguresRestClient; import org.jclouds.rest.ConfiguresRestClient;
import java.util.Properties;
/** /**
* Base class for writing Expect tests for AWS-S3 * Base class for writing Expect tests for AWS-S3
* *
@ -36,6 +46,9 @@ public class BaseAWSS3ClientExpectTest extends BaseRestClientExpectTest<AWSS3Cli
protected static final String CONSTANT_DATE = "2009-11-08T15:54:08.897Z"; protected static final String CONSTANT_DATE = "2009-11-08T15:54:08.897Z";
protected BlobStoreContext blobStoreContext;
protected BlobStore blobStore;
public BaseAWSS3ClientExpectTest() { public BaseAWSS3ClientExpectTest() {
provider = "aws-s3"; provider = "aws-s3";
} }
@ -54,4 +67,17 @@ public class BaseAWSS3ClientExpectTest extends BaseRestClientExpectTest<AWSS3Cli
return new TestAWSS3RestClientModule(); return new TestAWSS3RestClientModule();
} }
@Override
public AWSS3Client createClient(Function<HttpRequest, HttpResponse> fn, Module module, Properties props) {
return clientFrom(BlobStoreContext.class.cast(new BlobStoreContextFactory(setupRestProperties())
.createContext(provider, "identity", "credential", ImmutableSet.<Module>of(new ExpectModule(fn),
new NullLoggingModule(), module), props)));
}
protected AWSS3Client clientFrom(BlobStoreContext context) {
blobStoreContext = context;
blobStore = context.getBlobStore();
return AWSS3Client.class.cast(context.getProviderSpecificContext().getApi());
}
} }