public void RunAsyncBlockingCall()
{
Console.WriteLine("StatefulServiceLifeCycleTests - Test Method: RunAsyncBlockingCall()");
var serviceContext = TestMocksRepository.GetMockStatefulServiceContext();
var testService = new RunAsyncBlockingCallTestService(serviceContext);
IStatefulServiceReplica testServiceReplica = new StatefulServiceReplicaAdapter(serviceContext, testService);
var partition = new Mock <IStatefulServicePartition>();
partition.SetupGet(p => p.WriteStatus).Returns(PartitionAccessStatus.Granted);
testServiceReplica.OpenAsync(ReplicaOpenMode.New, partition.Object, CancellationToken.None).GetAwaiter().GetResult();
Console.WriteLine(@"// U -> P");
var changeRoleTask = testServiceReplica.ChangeRoleAsync(ReplicaRole.Primary, CancellationToken.None);
CancellationTokenSource source = new CancellationTokenSource(10000);
while (!testService.RunAsyncInvoked)
{
Task.Delay(100, source.Token).GetAwaiter().GetResult();
}
Assert.True(changeRoleTask.IsCompleted && !changeRoleTask.IsCanceled && !changeRoleTask.IsFaulted);
((StatefulServiceReplicaAdapter)testServiceReplica).Test_IsRunAsyncTaskRunning().Should().BeTrue();
testServiceReplica.CloseAsync(CancellationToken.None).GetAwaiter().GetResult();
}
public void CancellationDuringWriteStatus()
{
Console.WriteLine("StatefulServiceLifeCycleTests - Test Method: CancellationDuringWriteStatus()");
var serviceContext = TestMocksRepository.GetMockStatefulServiceContext();
IStatefulServiceReplica testServiceReplica =
new StatefulServiceReplicaAdapter(serviceContext, new StatefulBaseTestService(serviceContext));
var partition = new Mock <IStatefulServicePartition>();
// This will make WaitForWriteStatusAsync to keep retrying in loop.
partition.SetupGet(p => p.WriteStatus).Returns(PartitionAccessStatus.ReconfigurationPending);
testServiceReplica.OpenAsync(ReplicaOpenMode.New, partition.Object, CancellationToken.None).GetAwaiter().GetResult();
Console.WriteLine(@"// U -> P");
testServiceReplica.ChangeRoleAsync(ReplicaRole.Primary, CancellationToken.None).GetAwaiter().GetResult();
// This will throw if canceling during write status propagates out, as canceling of executeRunAsyncTask
// (which includes waiting for write status) is awaited during change role away from primary.
Console.WriteLine(@"// P -> S");
testServiceReplica.ChangeRoleAsync(ReplicaRole.ActiveSecondary, CancellationToken.None).GetAwaiter().GetResult();
partition.Verify(p => p.ReportFault(It.IsAny <FaultType>()), Times.Never());
}
public void RunAsyncSlowCancellation()
{
Console.WriteLine("StatefulServiceLifeCycleTests - Test Method: RunAsyncSlowCancellation()");
var serviceContext = TestMocksRepository.GetMockStatefulServiceContext();
var testService = new RunAsyncSlowCancellationTestService(serviceContext);
IStatefulServiceReplica testServiceReplica = new StatefulServiceReplicaAdapter(serviceContext, testService);
var partition = new Mock <IStatefulServicePartition>();
partition.SetupGet(p => p.WriteStatus).Returns(PartitionAccessStatus.Granted);
partition.Setup(p => p.ReportPartitionHealth(It.IsAny <HealthInformation>()));
testServiceReplica.OpenAsync(ReplicaOpenMode.New, partition.Object, CancellationToken.None).GetAwaiter().GetResult();
Console.WriteLine(@"// U -> P");
testServiceReplica.ChangeRoleAsync(ReplicaRole.Primary, CancellationToken.None).GetAwaiter().GetResult();
CancellationTokenSource source = new CancellationTokenSource(10000);
while (!testService.RunAsyncInvoked)
{
Task.Delay(100, source.Token).GetAwaiter().GetResult();
}
Console.WriteLine(@"// P -> S");
testServiceReplica.ChangeRoleAsync(ReplicaRole.ActiveSecondary, CancellationToken.None).GetAwaiter().GetResult();
partition.Verify(p => p.ReportFault(It.IsAny <FaultType>()), Times.Never());
partition.Verify(p => p.ReportPartitionHealth(It.Is <HealthInformation>(hinfo => Utility.IsRunAsyncSlowCancellationHealthInformation(hinfo))), Times.AtLeastOnce);
}
public void RunAsyncCancellation()
{
Console.WriteLine("StatefulServiceLifeCycleTests - Test Method: RunAsyncCancellation()");
var serviceContext = TestMocksRepository.GetMockStatefulServiceContext();
var testService = new RunAsyncCancellationTestService(serviceContext);
IStatefulServiceReplica testServiceReplica = new StatefulServiceReplicaAdapter(serviceContext, testService);
var partition = new Mock <IStatefulServicePartition>();
partition.SetupGet(p => p.WriteStatus).Returns(PartitionAccessStatus.Granted);
testServiceReplica.OpenAsync(ReplicaOpenMode.New, partition.Object, CancellationToken.None).GetAwaiter().GetResult();
Console.WriteLine(@"// U -> P");
testServiceReplica.ChangeRoleAsync(ReplicaRole.Primary, CancellationToken.None).GetAwaiter().GetResult();
CancellationTokenSource source = new CancellationTokenSource(10000);
while (!testService.StartedWaiting)
{
Task.Delay(100, source.Token).GetAwaiter().GetResult();
}
// This will throw if cancellation propagates out, as canceling of RunAsync is awaited during
// change role away from primary.
Console.WriteLine(@"// P -> S");
testServiceReplica.ChangeRoleAsync(ReplicaRole.ActiveSecondary, CancellationToken.None).GetAwaiter().GetResult();
partition.Verify(p => p.ReportFault(It.IsAny <FaultType>()), Times.Never());
}
public void RunAsyncFail()
{
Console.WriteLine("StatefulServiceLifeCycleTests - Test Method: RunAsyncFail()");
var serviceContext = TestMocksRepository.GetMockStatefulServiceContext();
IStatefulServiceReplica testServiceReplica = new StatefulServiceReplicaAdapter(serviceContext, new RunAsyncFailTestService(serviceContext));
var tcs = new TaskCompletionSource <bool>();
var partition = new Mock <IStatefulServicePartition>();
partition.SetupGet(p => p.WriteStatus).Returns(PartitionAccessStatus.Granted);
partition.Setup(p => p.ReportFault(It.IsAny <FaultType>())).Callback(
() =>
{
tcs.SetResult(true);
});
partition.Setup(p => p.ReportPartitionHealth(It.IsAny <HealthInformation>()));
testServiceReplica.OpenAsync(ReplicaOpenMode.New, partition.Object, CancellationToken.None).GetAwaiter().GetResult();
Console.WriteLine(@"// U -> P");
testServiceReplica.ChangeRoleAsync(ReplicaRole.Primary, CancellationToken.None).GetAwaiter().GetResult();
Task.Run(
() =>
{
Task.Delay(10000).GetAwaiter().GetResult();
tcs.SetCanceled();
});
tcs.Task.GetAwaiter().GetResult().Should().BeTrue();
// This will throw if RunAsync exception propagates out
Console.WriteLine(@"// P -> S");
testServiceReplica.ChangeRoleAsync(ReplicaRole.ActiveSecondary, CancellationToken.None).GetAwaiter().GetResult();
partition.Verify(p => p.ReportFault(It.IsNotIn(FaultType.Transient)), Times.Never());
partition.Verify(p => p.ReportFault(FaultType.Transient), Times.Once());
partition.Verify(p => p.ReportPartitionHealth(It.Is <HealthInformation>(hinfo => Utility.IsRunAsyncUnhandledExceptionHealthInformation(hinfo))), Times.Once());
}
public void ListenerExceptionOnAbort()
{
Console.WriteLine("StatefulServiceLifeCycleTests - Test Method: ListenerExceptionOnAbort");
var serviceContext = TestMocksRepository.GetMockStatefulServiceContext();
var testService = new StatefulBaseTestService(serviceContext)
{
EnableListenerExceptionOnAbort = true
};
IStatefulServiceReplica testServiceReplica = new StatefulServiceReplicaAdapter(serviceContext, testService);
var partition = new Mock <IStatefulServicePartition>();
testServiceReplica.OpenAsync(ReplicaOpenMode.New, partition.Object, CancellationToken.None).GetAwaiter().GetResult();
testServiceReplica.ChangeRoleAsync(ReplicaRole.Primary, CancellationToken.None).GetAwaiter().GetResult();
testServiceReplica.Abort();
}
public void CommunicationListenerLifeCycle_P_S_P_N_ListenOnSecondary()
{
Console.WriteLine("StatefulServiceLifeCycleTests - Test Method: CommunicationListenerLifeCycle_P_S_P_N_ListenOnSecondary");
var serviceContext = TestMocksRepository.GetMockStatefulServiceContext();
var testService = new StatefulBaseTestService(serviceContext)
{
ListenOnSecondary = true
};
IStatefulServiceReplica testServiceReplica = new StatefulServiceReplicaAdapter(serviceContext, testService);
var partition = new Mock <IStatefulServicePartition>();
testServiceReplica.OpenAsync(ReplicaOpenMode.New, partition.Object, CancellationToken.None).GetAwaiter().GetResult();
Console.WriteLine(@"// U -> P");
testServiceReplica.ChangeRoleAsync(ReplicaRole.Primary, CancellationToken.None).GetAwaiter().GetResult();
{
const int expectedCount = 1;
int actualCount = testService.Listeners.Count;
actualCount.Should().Be(expectedCount, "listener has been opened only once(U->P)");
testService.Listeners.Last().Should().Be(testService.CurrentListener);
((StatefulServiceReplicaAdapter)testServiceReplica).Test_CommunicationListeners.First().Should().Be(testService.CurrentListener.Object);
testService.CurrentListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
testService.CurrentListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Never());
testService.CurrentListener.Verify(l => l.Abort(), Times.Never());
}
Console.WriteLine(@"// P -> S");
testServiceReplica.ChangeRoleAsync(ReplicaRole.ActiveSecondary, CancellationToken.None).GetAwaiter().GetResult();
{
const int expectedCount = 2;
int actualCount = testService.Listeners.Count;
actualCount.Should().Be(expectedCount, "listener has been opened twice(U->P->S)");
testService.Listeners.Last().Should().Be(testService.CurrentListener);
((StatefulServiceReplicaAdapter)testServiceReplica).Test_CommunicationListeners.First().Should().Be(testService.CurrentListener.Object);
var firstListener = testService.Listeners[0];
firstListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
firstListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Once());
firstListener.Verify(l => l.Abort(), Times.Never());
testService.CurrentListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
testService.CurrentListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Never());
testService.CurrentListener.Verify(l => l.Abort(), Times.Never());
}
Console.WriteLine(@"// S -> P");
testServiceReplica.ChangeRoleAsync(ReplicaRole.Primary, CancellationToken.None).GetAwaiter().GetResult();
{
const int expectedCount = 3;
int actualCount = testService.Listeners.Count;
actualCount.Should().Be(expectedCount, "listener has been opened three times(U->P->S->P)");
testService.Listeners.Last().Should().Be(testService.CurrentListener);
((StatefulServiceReplicaAdapter)testServiceReplica).Test_CommunicationListeners.First().Should().Be(testService.CurrentListener.Object);
var firstListener = testService.Listeners[0];
firstListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
firstListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Once());
firstListener.Verify(l => l.Abort(), Times.Never());
var secondListener = testService.Listeners[1];
secondListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
secondListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Once());
secondListener.Verify(l => l.Abort(), Times.Never());
testService.CurrentListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
testService.CurrentListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Never());
testService.CurrentListener.Verify(l => l.Abort(), Times.Never());
}
Console.WriteLine(@"// P -> N");
testServiceReplica.ChangeRoleAsync(ReplicaRole.None, CancellationToken.None).GetAwaiter().GetResult();
{
const int expectedCount = 3;
int actualCount = testService.Listeners.Count;
actualCount.Should().Be(expectedCount, "listener has been opened three times(U->P->S->P->N)");
testService.Listeners.Last().Should().Be(testService.CurrentListener);
((StatefulServiceReplicaAdapter)testServiceReplica).Test_CommunicationListeners.Should().BeNull();
var firstListener = testService.Listeners[0];
firstListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
firstListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Once());
firstListener.Verify(l => l.Abort(), Times.Never());
var secondListener = testService.Listeners[1];
secondListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
secondListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Once());
secondListener.Verify(l => l.Abort(), Times.Never());
testService.CurrentListener.Verify(l => l.OpenAsync(It.IsAny <CancellationToken>()), Times.Once());
testService.CurrentListener.Verify(l => l.CloseAsync(It.IsAny <CancellationToken>()), Times.Once());
testService.CurrentListener.Verify(l => l.Abort(), Times.Never());
}
partition.Verify(p => p.ReportFault(It.IsAny <FaultType>()), Times.Never());
}