public void TestSignerWithBasicCredentials()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new BasicAWSCredentials("accessKey", "secretKey")));
var context = CreateTestContext();
var signer = new MockSigner();
((RequestContext)context.RequestContext).Signer = signer;
pipeline.InvokeSync(context);
Assert.IsTrue(context.RequestContext.IsSigned);
Assert.AreEqual(1, signer.SignCount);
}
public async Task TestSignerWithBasicCredentialsAsync()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new BasicAWSCredentials("accessKey", "secretKey")));
var context = CreateTestContext();
var signer = new MockSigner();
((RequestContext)context.RequestContext).Signer = signer;
await pipeline.InvokeAsync<AmazonWebServiceResponse>(context);
Assert.IsTrue(context.RequestContext.IsSigned);
Assert.AreEqual(1, signer.SignCount);
}
public async Task TestSignerWithBasicCredentialsAsync()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new BasicAWSCredentials("accessKey", "secretKey")));
var signer = new MockSigner();
var context = CreateTestContext(signer);
await pipeline.InvokeAsync <AmazonWebServiceResponse>(context);
Assert.IsTrue(context.RequestContext.IsSigned);
Assert.AreEqual(1, signer.SignCount);
}
public void TestSignerWithAnonymousCredentials()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new AnonymousAWSCredentials()));
var context = CreateTestContext();
var signer = new MockSigner();
((RequestContext)context.RequestContext).Signer = signer;
pipeline.InvokeSync(context);
Assert.IsTrue(context.RequestContext.IsSigned);
Assert.AreEqual(0, signer.SignCount);
}
public void TestSignerWithBasicCredentials()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new BasicAWSCredentials("accessKey", "secretKey")));
var context = CreateTestContext();
var signer = new MockSigner();
((RequestContext)context.RequestContext).Signer = signer;
pipeline.InvokeSync(context);
Assert.IsTrue(context.RequestContext.IsSigned);
Assert.AreEqual(1, signer.SignCount);
}
public void TestSignerWithAnonymousCredentials()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new AnonymousAWSCredentials()));
var context = CreateTestContext();
var signer = new MockSigner();
((RequestContext)context.RequestContext).Signer = signer;
pipeline.InvokeSync(context);
Assert.IsTrue(context.RequestContext.IsSigned);
Assert.AreEqual(0, signer.SignCount);
}
public void RunRetryTest(Action <IExecutionContext, MockAdaptiveRetryPolicy> DoAction, AmazonS3Config config, CapacityManager capacityManager = null)
{
try
{
if (capacityManager != null)
{
MockAdaptiveRetryPolicy.SetCapacityManagerInstance(capacityManager);
}
var retryPolicy = new MockAdaptiveRetryPolicy(config);
Handler = new RetryHandler(retryPolicy);
if (RuntimePipeline.Handlers.Find(h => h is RetryHandler) != null)
{
RuntimePipeline.ReplaceHandler <RetryHandler>(Handler);
}
else
{
RuntimePipeline.AddHandler(Handler);
}
var executionContext = CreateTestContext(null, null, config);
DoAction(executionContext, retryPolicy);
}
finally
{
if (capacityManager != null)
{
MockAdaptiveRetryPolicy.RestoreManagers();
}
}
}
public void TestSignerWithBasicCredentialsAsync()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new BasicAWSCredentials("accessKey", "secretKey")));
var signer = new MockSigner();
var context = CreateAsyncTestContext(signer);
var asyncResult = pipeline.InvokeAsync(context);
asyncResult.AsyncWaitHandle.WaitOne();
Assert.IsTrue(context.RequestContext.IsSigned);
Assert.AreEqual(1, signer.SignCount);
}
public static void Initialize(TestContext t)
{
var logger =
Amazon.Runtime.Internal.Util.Logger.GetLogger(new object().GetType());
Handler = new ErrorHandler(logger);
RuntimePipeline.AddHandler(Handler);
}
public void ReplaceHandlerTest()
{
var handlerC = new TestHandlerC();
var pipeline = new RuntimePipeline(handlerC);
var handlerB = new TestHandlerB();
pipeline.AddHandler(handlerB);
var handlerA = new TestHandlerA();
//A->B->C
pipeline.AddHandler(handlerA);
ValidatePipeline(pipeline, handlerA, handlerB, handlerC);
var handlerD = new TestHandlerD();
//A->D->C
pipeline.ReplaceHandler<TestHandlerB>(handlerD);
ValidatePipeline(pipeline, handlerA, handlerD, handlerC);
Assert.IsNull(handlerB.OuterHandler);
Assert.IsNull(handlerB.InnerHandler);
}
public static void Initialize(TestContext t)
{
logger = Logger.GetLogger(typeof(RetryAndErrorHandlerTests));
var errorHandler = new ErrorHandler(logger);
RuntimePipeline.AddHandler(errorHandler);
Handler = new RetryHandler(new DefaultRetryPolicy(MAX_RETRIES));
RuntimePipeline.AddHandler(Handler);
}
public void AddHandlerTest()
{
var handlerA = new TestHandlerA();
var pipeline = new RuntimePipeline(handlerA);
var handlerB = new TestHandlerB();
// B->A
pipeline.AddHandler(handlerB);
ValidatePipeline(pipeline, handlerB, handlerA);
}
public void TestSignerWithMutableHeader()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new BasicAWSCredentials("accessKey", "secretKey")));
var context = CreateTestContext();
var signer = new AWS4Signer();
((RequestContext)context.RequestContext).Signer = signer;
// inject a mutable header that the signer should strip out
context.RequestContext.Request.Headers[HeaderKeys.XAmznTraceIdHeader] = "stuff";
pipeline.InvokeSync(context);
// verify that the header is not in the signature
var t = context.RequestContext.Request.Headers[HeaderKeys.AuthorizationHeader];
Assert.IsFalse(t.Contains(HeaderKeys.XAmznTraceIdHeader));
Assert.IsTrue(context.RequestContext.Request.Headers.ContainsKey(HeaderKeys.XAmznTraceIdHeader));
}
public void AddHandlerTest()
{
var handlerA = new TestHandlerA();
var pipeline = new RuntimePipeline(handlerA);
var handlerB = new TestHandlerB();
// B->A
pipeline.AddHandler(handlerB);
ValidatePipeline(pipeline, handlerB, handlerA);
}
public static void Initialize(TestContext t)
{
ClientConfig config = new AmazonS3Config
{
ServiceURL = @"https://s3.amazonaws.com",
MaxErrorRetry = MAX_RETRIES
};
Handler = new RetryHandler(new DefaultRetryPolicy(config));
RuntimePipeline.AddHandler(Handler);
}
public void TestSignerWithMutableHeader()
{
var pipeline = new RuntimePipeline(new MockHandler());
pipeline.AddHandler(new Signer());
pipeline.AddHandler(new CredentialsRetriever(new BasicAWSCredentials("accessKey", "secretKey")));
var signer = new AWS4Signer();
var context = CreateTestContext(signer);
// inject a mutable header that the signer should strip out
context.RequestContext.Request.Headers[HeaderKeys.XAmznTraceIdHeader] = "stuff";
pipeline.InvokeSync(context);
// verify that the header is not in the signature
var t = context.RequestContext.Request.Headers[HeaderKeys.AuthorizationHeader];
Assert.IsFalse(t.Contains(HeaderKeys.XAmznTraceIdHeader));
Assert.IsTrue(context.RequestContext.Request.Headers.ContainsKey(HeaderKeys.XAmznTraceIdHeader));
}
public void ReplaceHandlerTest()
{
var handlerC = new TestHandlerC();
var pipeline = new RuntimePipeline(handlerC);
var handlerB = new TestHandlerB();
pipeline.AddHandler(handlerB);
var handlerA = new TestHandlerA();
//A->B->C
pipeline.AddHandler(handlerA);
ValidatePipeline(pipeline, handlerA, handlerB, handlerC);
var handlerD = new TestHandlerD();
//A->D->C
pipeline.ReplaceHandler <TestHandlerB>(handlerD);
ValidatePipeline(pipeline, handlerA, handlerD, handlerC);
Assert.IsNull(handlerB.OuterHandler);
Assert.IsNull(handlerB.InnerHandler);
}
public static void Initialize(TestContext t)
{
logger = Logger.GetLogger(typeof(RetryAndErrorHandlerTests));
var errorHandler = new ErrorHandler(logger);
RuntimePipeline.AddHandler(errorHandler);
ClientConfig config = new AmazonS3Config
{
ServiceURL = @"https://s3.amazonaws.com",
MaxErrorRetry = MAX_RETRIES
};
Handler = new RetryHandler(new DefaultRetryPolicy(config));
RuntimePipeline.AddHandler(Handler);
}
public void AddHandlerBeforeTest()
{
var handlerD = new TestHandlerD();
var pipeline = new RuntimePipeline(handlerD);
var handlerB = new TestHandlerB();
// B->D
pipeline.AddHandler(handlerB);
var handlerA = new TestHandlerA();
// A->B->D
pipeline.AddHandlerBefore<TestHandlerB>(handlerA);
ValidatePipeline(pipeline, handlerA, handlerB, handlerD);
var handlerC = new TestHandlerC();
// A->B->C->D
pipeline.AddHandlerBefore<TestHandlerD>(handlerC);
ValidatePipeline(pipeline, handlerA, handlerB, handlerC, handlerD);
}
public void AddHandlerBeforeTest()
{
var handlerD = new TestHandlerD();
var pipeline = new RuntimePipeline(handlerD);
var handlerB = new TestHandlerB();
// B->D
pipeline.AddHandler(handlerB);
var handlerA = new TestHandlerA();
// A->B->D
pipeline.AddHandlerBefore <TestHandlerB>(handlerA);
ValidatePipeline(pipeline, handlerA, handlerB, handlerD);
var handlerC = new TestHandlerC();
// A->B->C->D
pipeline.AddHandlerBefore <TestHandlerD>(handlerC);
ValidatePipeline(pipeline, handlerA, handlerB, handlerC, handlerD);
}
public static void Initialize(TestContext t)
{
Handler = new RedirectHandler();
RuntimePipeline.AddHandler(Handler);
}
public void Customize(Type type, RuntimePipeline pipeline)
{
pipeline.AddHandler(new SpecialPipelineHandler());
}
public static void Initialize(TestContext t)
{
Handler = new RetryHandler(new DefaultRetryPolicy(MAX_RETRIES));
RuntimePipeline.AddHandler(Handler);
}
public void Customize(Type type, RuntimePipeline pipeline)
{
var handler = new HandlerFactory().GetInstace(type);
pipeline.AddHandler(handler);
}
public static void Initialize(TestContext t)
{
Handler = new Unmarshaller(true);
RuntimePipeline.AddHandler(Handler);
}