public void StopsTransmissionSendingWhenTransmissionTimesOut()
{
var policyApplied = new AutoResetEvent(false);
var transmitter = new StubTransmitter(new TestableBackoffLogicManager(TimeSpan.FromSeconds(10)));
transmitter.OnApplyPolicies = () =>
{
policyApplied.Set();
};
var policy = new ErrorHandlingTransmissionPolicy();
policy.Initialize(transmitter);
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(new StubTransmission(), CreateException(statusCode: 408)));
Assert.IsTrue(policyApplied.WaitOne(100));
Assert.AreEqual(0, policy.MaxSenderCapacity);
}
public void InitializeCallsTransmitterInitialize()
{
var transmitterInitialized = new ManualResetEvent(false);
var transmitter = new StubTransmitter();
transmitter.OnInitialize = () =>
{
transmitterInitialized.Set();
};
var channel = new ServerTelemetryChannel {
Transmitter = transmitter
};
var initializedConfiguration = new TelemetryConfiguration();
channel.Initialize(initializedConfiguration);
Assert.IsTrue(transmitterInitialized.WaitOne(1000));
}
public void AsksTransmitterToApplyPoliciesWhenNetworkAvailabilityChanges()
{
bool policiesApplied = false;
var transmitter = new StubTransmitter();
transmitter.OnApplyPolicies = () =>
{
policiesApplied = true;
};
var network = new StubNetwork();
var policy = new NetworkAvailabilityTransmissionPolicy(network);
policy.Initialize(transmitter);
network.OnStatusChanged(EventArgs.Empty);
Assert.True(policiesApplied);
}
public void KeepsTransmissionSenderPausedWhenAdditionalTransmissionsFail()
{
var transmitter = new StubTransmitter(new TestableBackoffLogicManager(TimeSpan.FromMinutes(1)));
var policy = new ErrorHandlingTransmissionPolicy();
policy.Initialize(transmitter);
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(new StubTransmission(), new Exception("Error"), new HttpWebResponseWrapper()
{
StatusCode = ResponseStatusCodes.InternalServerError
}));
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(new StubTransmission(), new Exception("Error"), new HttpWebResponseWrapper()
{
StatusCode = ResponseStatusCodes.InternalServerError
}));
Thread.Sleep(TimeSpan.FromMilliseconds(30));
Assert.AreEqual(0, policy.MaxSenderCapacity);
}
public void ResumesTransmissionSenderAfterPauseDuration()
{
var policyApplied = new AutoResetEvent(false);
var transmitter = new StubTransmitter(new TestableBackoffLogicManager(TimeSpan.FromMilliseconds(1)));
transmitter.OnApplyPolicies = () =>
{
policyApplied.Set();
};
var policy = new ErrorHandlingTransmissionPolicy();
policy.Initialize(transmitter);
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(new StubTransmission(), CreateException(statusCode: 408)));
Assert.True(policyApplied.WaitOne(100));
Assert.True(policyApplied.WaitOne(100));
Assert.Null(policy.MaxSenderCapacity);
}
public void RetriesFailedTransmissionIfItsNumberOfAttemptsDidNotReachMaximum()
{
Transmission enqueuedTransmission = null;
var transmitter = new StubTransmitter();
transmitter.OnEnqueue = transmission =>
{
enqueuedTransmission = transmission;
};
var policy = new ErrorHandlingTransmissionPolicy();
policy.Initialize(transmitter);
var failedTransmission = new StubTransmission();
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(failedTransmission, CreateException(statusCode: 408)));
Assert.AreSame(failedTransmission, enqueuedTransmission);
}
public void NewTransmissionCreatedByIndexesFromResponse()
{
IList <Transmission> enqueuedTransmissions = new List <Transmission>();
var transmitter = new StubTransmitter
{
OnEnqueue = t => { enqueuedTransmissions.Add(t); }
};
var policy = new PartialSuccessTransmissionPolicy();
policy.Initialize(transmitter);
var items = new List <ITelemetry>
{
new EventTelemetry("1"),
new EventTelemetry("2"),
new EventTelemetry("3"),
};
Transmission transmission = new Transmission(new Uri("http://uri"), items, "type", "encoding");
string response = BackendResponseHelper.CreateBackendResponse(
itemsReceived: 3,
itemsAccepted: 1,
errorCodes: new[] { "439", "439" });
var wrapper = new HttpWebResponseWrapper
{
StatusCode = 206,
Content = response
};
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(transmission, null, wrapper));
string[] newItems = JsonSerializer
.Deserialize(enqueuedTransmissions[0].Content)
.Split(new[] { Environment.NewLine }, StringSplitOptions.RemoveEmptyEntries);
Assert.AreEqual(2, newItems.Length);
Assert.IsTrue(newItems[0].Contains("\"name\":\"1\""));
Assert.IsTrue(newItems[1].Contains("\"name\":\"2\""));
}
public void AppliesTransmissionPoliciesToBeginSendingStoredTelemetry()
{
var transmissionPoliciesApplied = new ManualResetEvent(false);
var transmitter = new StubTransmitter();
transmitter.OnApplyPolicies = () =>
{
transmissionPoliciesApplied.Set();
};
var channel = new ServerTelemetryChannel {
Transmitter = transmitter
};
channel.Initialize(TelemetryConfiguration.CreateDefault());
var initializedConfiguration = new TelemetryConfiguration();
channel.Initialize(initializedConfiguration);
Assert.IsTrue(transmissionPoliciesApplied.WaitOne(1000));
}
public void DoesNotRetrySuccessfulTransmission()
{
Transmission enqueuedTransmission = null;
var transmitter = new StubTransmitter();
transmitter.OnEnqueue = transmission =>
{
enqueuedTransmission = transmission;
};
var policy = new ErrorHandlingTransmissionPolicy();
policy.Initialize(transmitter);
var successfulTransmission = new StubTransmission();
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(successfulTransmission));
Assert.IsNull(enqueuedTransmission);
Assert.AreEqual(0, transmitter.BackoffLogicManager.ConsecutiveErrors);
}
private void PositiveTest(int responseCode, int?expectedSenderCapacity, int?expectedBufferCapacity, int?expectedStorageCapacity)
{
const int RetryAfterSeconds = 2;
string retryAfter = DateTime.Now.ToUniversalTime().AddSeconds(RetryAfterSeconds).ToString("R", CultureInfo.InvariantCulture);
const int WaitForTheFirstApplyAsync = 100;
int waitForTheSecondApplyAsync = (RetryAfterSeconds * 1000) /*to miliseconds*/ +
500 /**magic number to wait for other code before/after
* timer which calls 2nd ApplyAsync
**/;
var policyApplied = new AutoResetEvent(false);
var transmitter = new StubTransmitter();
transmitter.OnApplyPolicies = () =>
{
policyApplied.Set();
};
var policy = new ThrottlingTransmissionPolicy();
policy.Initialize(transmitter);
transmitter.OnTransmissionSent(
new TransmissionProcessedEventArgs(
new StubTransmission(),
CreateThrottledResponse(responseCode, retryAfter)));
Assert.True(policyApplied.WaitOne(WaitForTheFirstApplyAsync));
Assert.Equal(expectedSenderCapacity, policy.MaxSenderCapacity);
Assert.Equal(expectedBufferCapacity, policy.MaxBufferCapacity);
Assert.Equal(expectedStorageCapacity, policy.MaxStorageCapacity);
Assert.True(policyApplied.WaitOne(waitForTheSecondApplyAsync));
// Check that it resets after retry-after interval
Assert.Null(policy.MaxSenderCapacity);
Assert.Null(policy.MaxBufferCapacity);
Assert.Null(policy.MaxStorageCapacity);
}
public void AssertUnsupportedResponseCodeDoesntChangeCapacity()
{
var transmitter = new StubTransmitter();
transmitter.OnApplyPolicies = () =>
{
throw new Exception("Apply shouldn't be called because unsupported response code was passed");
};
var policy = new ThrottlingTransmissionPolicy();
policy.Initialize(transmitter);
transmitter.OnTransmissionSent(
new TransmissionProcessedEventArgs(
new StubTransmission(),
CreateThrottledResponse(ResponseCodeUnsupported, 1)));
Assert.Null(policy.MaxSenderCapacity);
Assert.Null(policy.MaxBufferCapacity);
Assert.Null(policy.MaxStorageCapacity);
}
public void EnqueuesTransmissionWithExpectedProperties()
{
Transmission transmission = null;
var transmitter = new StubTransmitter();
transmitter.OnEnqueue = t =>
{
transmission = t;
};
var serializer = new TelemetrySerializer(transmitter)
{
EndpointAddress = new Uri("http://expected.uri")
};
serializer.Serialize(new[] { new StubTelemetry() });
Assert.Equal(serializer.EndpointAddress, transmission.EndpointAddress);
Assert.Equal("application/x-json-stream", transmission.ContentType);
Assert.Equal("gzip", transmission.ContentEncoding);
Assert.Equal(string.Empty, Unzip(transmission.Content));
}
public void RetriesFailedTransmissionInfinitely()
{
Transmission enqueuedTransmission = null;
var transmitter = new StubTransmitter(new TestableBackoffLogicManager(TimeSpan.FromMilliseconds(10)));
transmitter.OnEnqueue = transmission =>
{
enqueuedTransmission = transmission;
};
var policy = new ErrorHandlingTransmissionPolicy();
policy.Initialize(transmitter);
var failedTransmission = new StubTransmission();
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(failedTransmission, CreateException(statusCode: 408)));
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(failedTransmission, CreateException(statusCode: 408)));
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(failedTransmission, CreateException(statusCode: 408)));
Assert.AreSame(failedTransmission, enqueuedTransmission);
}
private void ResumesTransmissionSenderAfterPauseDuration(int responseStatusCode)
{
var policyApplied = new AutoResetEvent(false);
var transmitter = new StubTransmitter(new TestableBackoffLogicManager(TimeSpan.FromMilliseconds(1)));
transmitter.OnApplyPolicies = () => { policyApplied.Set(); };
var policy = new ErrorHandlingTransmissionPolicy();
policy.Initialize(transmitter);
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(new StubTransmission(),
new Exception("Error"),
new HttpWebResponseWrapper()
{
StatusCode = responseStatusCode
}));
Assert.IsTrue(policyApplied.WaitOne(100));
Assert.IsTrue(policyApplied.WaitOne(100));
Assert.IsNull(policy.MaxSenderCapacity);
}
public void ReturnsBoolenTaskWhenTelemetryIsNullOrEmpty()
{
Transmission transmission = null;
var transmitter = new StubTransmitter();
transmitter.OnEnqueue = t =>
{
transmission = t;
};
var serializer = new TelemetrySerializer(transmitter)
{
EndpointAddress = new Uri("http://expected.uri")
};
var result = serializer.SerializeAsync(null, default);
Assert.IsInstanceOfType(result, typeof(Task <bool>));
Assert.IsTrue(result.Result);
result = serializer.SerializeAsync(new List <ITelemetry>(), default);
Assert.IsInstanceOfType(result, typeof(Task <bool>));
Assert.IsTrue(result.Result);
}
public void DoesNotContinueAsyncOperationsOnCapturedSynchronizationContextToImprovePerformance()
{
var transmitter = new StubTransmitter()
{
OnEnqueue = transmission => TaskEx.Run(() => { })
};
var serializer = new TelemetrySerializer(transmitter);
bool postedBack = false;
using (var context = new StubSynchronizationContext())
{
context.OnPost = (callback, state) =>
{
postedBack = true;
callback(state);
};
serializer.Serialize(new[] { new StubTelemetry() });
}
Assert.IsFalse(postedBack);
}
public void CannotParseRetryAfterWritesToEventSource()
{
const string UnparsableDate = "no one can parse me! :)";
var transmitter = new StubTransmitter();
var policy = new ThrottlingTransmissionPolicy();
policy.Initialize(transmitter);
using (var listener = new TestEventListener())
{
const long AllKeyword = -1;
listener.EnableEvents(TelemetryChannelEventSource.Log, EventLevel.LogAlways, (EventKeywords)AllKeyword);
transmitter.OnTransmissionSent(
new TransmissionProcessedEventArgs(
new StubTransmission(),
CreateThrottledResponse(ResponseCodeTooManyRequests, UnparsableDate)));
EventWrittenEventArgs trace = listener.Messages.First(args => args.EventId == 24);
Assert.Equal(UnparsableDate, (string)trace.Payload[0]);
}
}
public void RetriesFailedTransmissionIfItsNumberOfAttemptsDidNotReachMaximum()
{
Transmission enqueuedTransmission = null;
var transmitter = new StubTransmitter();
transmitter.OnEnqueue = transmission =>
{
enqueuedTransmission = transmission;
};
var policy = new ErrorHandlingTransmissionPolicy();
policy.Initialize(transmitter);
var failedTransmission = new StubTransmission();
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(failedTransmission, new Exception("Error"), new HttpWebResponseWrapper()
{
StatusCode = ResponseStatusCodes.InternalServerError
}));
Assert.AreSame(failedTransmission, enqueuedTransmission);
}
public void AssertPaymentRequiredDoesntChangeCapacity()
{
var transmitter = new StubTransmitter();
transmitter.OnApplyPolicies = () =>
{
throw new Exception("Apply shouldn't be called because unsupported response code was passed");
};
var policy = new ThrottlingTransmissionPolicy();
policy.Initialize(transmitter);
transmitter.OnTransmissionSent(
new TransmissionProcessedEventArgs(
new StubTransmission(), null, new HttpWebResponseWrapper()
{
StatusCode = ResponseCodePaymentRequired
}));
Assert.IsNull(policy.MaxSenderCapacity);
Assert.IsNull(policy.MaxBufferCapacity);
Assert.IsNull(policy.MaxStorageCapacity);
}
public void ItemsAreEnqueuedOnFlushAsync()
{
IList <Transmission> enqueuedTransmissions = new List <Transmission>();
var transmitter = new StubTransmitter
{
OnEnqueue = t => { enqueuedTransmissions.Add(t); }
};
var policy = new PartialSuccessTransmissionPolicy();
policy.Initialize(transmitter);
var items = new List <ITelemetry> {
new EventTelemetry(), new EventTelemetry()
};
Transmission transmission = new Transmission(new Uri("http://uri"), items, "type", "encoding")
{
IsFlushAsyncInProgress = true
};
string response = BackendResponseHelper.CreateBackendResponse(
itemsReceived: 2,
itemsAccepted: 1,
errorCodes: new[] { "429" });
var wrapper = new HttpWebResponseWrapper
{
StatusCode = 206,
Content = response
};
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(transmission, null, wrapper));
Assert.AreEqual(1, enqueuedTransmissions.Count);
Assert.IsTrue(transmission.IsFlushAsyncInProgress);
}
public void AssertIfDateParseErrorCausesDefaultDelay()
{
var policyApplied = new AutoResetEvent(false);
var transmitter = new StubTransmitter();
transmitter.OnApplyPolicies = () =>
{
policyApplied.Set();
};
var policy = new ThrottlingTransmissionPolicy();
policy.Initialize(transmitter);
transmitter.OnTransmissionSent(
new TransmissionProcessedEventArgs(
new StubTransmission(),
CreateThrottledResponse(ResponseCodeTooManyRequests, "no one can parse me! :)")));
Assert.True(policyApplied.WaitOne(100));
Assert.Null(policy.MaxSenderCapacity);
Assert.Null(policy.MaxBufferCapacity);
Assert.Null(policy.MaxStorageCapacity);
}
private static void CatchesAndLogsExceptionThrownByTransmitter(ErrorHandlingTransmissionPolicy policy, StubTransmitter transmitter, Exception exception)
{
policy.Initialize(transmitter);
using (var listener = new TestEventListener())
{
const long AllKeywords = -1;
listener.EnableEvents(TelemetryChannelEventSource.Log, EventLevel.Warning, (EventKeywords)AllKeywords);
transmitter.OnTransmissionSent(new TransmissionProcessedEventArgs(new StubTransmission(), CreateException(statusCode: 408)));
EventWrittenEventArgs error = listener.Messages.First(args => args.EventId == 23);
AssertEx.Contains(exception.Message, (string)error.Payload[1], StringComparison.Ordinal);
}
}
