public void CreateAndCancelKeepaliveTask()
{
string serverUri = $"http://{_random.Next(256)}.{_random.Next(256)}.{_random.Next(256)}.{_random.Next(256)}:{_random.Next(65000)}";
ushort leaseTtl = Convert.ToUInt16(_random.Next(65000));
long? leaseId = null;
CancellationToken?token = null;
Task keepAliveTask = Task.Run(() => { Trace.WriteLine("x"); });
FQueueConfiguration configuration = new FQueueConfiguration
(
RestNodeConfigurationTests.CreateConfiguration(),
RestConfigurationTests.CreateConfiguration(),
FilesConfigurationTests.CreateConfiguration(),
PerformanceConfigurationTests.CreateConfiguration(),
new LeaderElectionConfiguration(leaseTtl)
);
Mock <IEtcdWrapper> etcdWrapperMock = new Mock <IEtcdWrapper>();
etcdWrapperMock
.Setup(p => p.CreateLease(serverUri, It.IsAny <long>(), It.IsAny <long>()))
.Callback <string, long, long>((uri, id, ttl) => { leaseId = id; })
.Returns(true);
etcdWrapperMock
.Setup(p => p.StartKeepAlive(serverUri, It.IsAny <long>(), It.IsAny <CancellationToken>()))
.Callback <string, long, CancellationToken>((uri, id, ct) => { token = ct; })
.Returns(keepAliveTask);
Mock <IServerUri> serverUriMock = new Mock <IServerUri>();
serverUriMock.Setup(p => p.Uri).Returns(serverUri);
Mock <IConfigurationReader> configurationReaderMock = new Mock <IConfigurationReader>();
configurationReaderMock.Setup(p => p.Configuration).Returns(configuration);
EtcdLeaseBackgroundTask backgroundTask = new EtcdLeaseBackgroundTask(etcdWrapperMock.Object, serverUriMock.Object, configurationReaderMock.Object);
backgroundTask.Start();
Assert.IsNotNull(token);
Assert.IsNotNull(leaseId);
etcdWrapperMock.Verify(p => p.CreateLease(serverUri, It.IsAny <long>(), Convert.ToInt64(leaseTtl)), Times.Once);
etcdWrapperMock.Verify(p => p.StartKeepAlive(serverUri, leaseId.Value, token.Value), Times.Once);
serverUriMock.Verify(p => p.Uri, Times.AtLeastOnce);
backgroundTask.Stop();
Assert.IsTrue(token.Value.IsCancellationRequested);
}
public void RestConfigurationForNodeReturned()
{
Fixture fixture = new Fixture();
ushort port = fixture.Create <ushort>();
FQueueConfiguration configuration = new FQueueConfiguration
(
new RestNodeConfiguration(port, ThrottlingConfigurationTests.CreateConfiguration(), true),
RestConfigurationTests.CreateConfiguration(),
FilesConfigurationTests.CreateConfiguration(),
PerformanceConfigurationTests.CreateConfiguration(),
LeaderElectionConfigurationTests.CreateConfiguration()
);
Mock <IConfigurationReader> configurationReaderMock = new Mock <IConfigurationReader>();
configurationReaderMock.Setup(p => p.Configuration).Returns(configuration);
NodeEngineTester tester = new NodeEngineTester(configurationReaderMock.Object);
Assert.AreEqual(port, tester.GetRestPort());
}
public void CreateLeaseFailed()
{
string serverUri = $"http://{_random.Next(256)}.{_random.Next(256)}.{_random.Next(256)}.{_random.Next(256)}:{_random.Next(65000)}";
ushort leaseTtl = Convert.ToUInt16(_random.Next(65000));
Task keepAliveTask = Task.Run(() => { Trace.WriteLine("x"); });
FQueueConfiguration configuration = new FQueueConfiguration
(
RestNodeConfigurationTests.CreateConfiguration(),
RestConfigurationTests.CreateConfiguration(),
FilesConfigurationTests.CreateConfiguration(),
PerformanceConfigurationTests.CreateConfiguration(),
new LeaderElectionConfiguration(leaseTtl)
);
Mock <IEtcdWrapper> etcdWrapperMock = new Mock <IEtcdWrapper>();
etcdWrapperMock.Setup(p => p.CreateLease(serverUri, It.IsAny <long>(), It.IsAny <long>())).Returns(false);
etcdWrapperMock.Setup(p => p.StartKeepAlive(serverUri, It.IsAny <long>(), It.IsAny <CancellationToken>())).Returns(keepAliveTask);
Mock <IServerUri> serverUriMock = new Mock <IServerUri>();
serverUriMock.Setup(p => p.Uri).Returns(serverUri);
Mock <IConfigurationReader> configurationReaderMock = new Mock <IConfigurationReader>();
configurationReaderMock.Setup(p => p.Configuration).Returns(configuration);
EtcdLeaseBackgroundTask backgroundTask = new EtcdLeaseBackgroundTask(etcdWrapperMock.Object, serverUriMock.Object, configurationReaderMock.Object);
Assert.Throws <IOException>(() => backgroundTask.Start());
etcdWrapperMock.Verify(p => p.CreateLease(It.IsAny <string>(), It.IsAny <long>(), It.IsAny <long>()), Times.Once);
etcdWrapperMock.Verify(p => p.StartKeepAlive(It.IsAny <string>(), It.IsAny <long>(), It.IsAny <CancellationToken>()), Times.Never);
Assert.Throws <InvalidOperationException>(() => backgroundTask.Stop());
}
