public async Task ConnectivityTestFailedTest()
{
// Arrange / act
var deviceIdentity = Mock.Of <IIdentity>(i => i.Id == "d2");
var edgeHubIdentity = Mock.Of <IIdentity>(i => i.Id == "d1/m1");
var deviceConnectivityManager = new DeviceConnectivityManager(TimeSpan.FromSeconds(3), TimeSpan.FromSeconds(3), edgeHubIdentity);
var client = new Mock <IClient>();
client.SetupSequence(c => c.UpdateReportedPropertiesAsync(It.IsAny <TwinCollection>()))
.Returns(Task.CompletedTask)
.Throws <TimeoutException>()
.Returns(Task.CompletedTask)
.Throws <TimeoutException>()
.Returns(Task.CompletedTask);
IClient connectivityAwareClient = new ConnectivityAwareClient(client.Object, deviceConnectivityManager, deviceIdentity);
ICloudProxy cloudProxy = new CloudProxy(connectivityAwareClient, Mock.Of <IMessageConverterProvider>(), "d1/m1", null, Mock.Of <ICloudListener>(), TimeSpan.FromHours(1), true);
var connectionManager = Mock.Of <IConnectionManager>(c => c.GetCloudConnection("d1/m1") == Task.FromResult(Option.Some(cloudProxy)));
deviceConnectivityManager.SetConnectionManager(connectionManager);
int connected = 0;
int disconnected = 0;
deviceConnectivityManager.DeviceConnected += (_, __) => Interlocked.Increment(ref connected);
deviceConnectivityManager.DeviceDisconnected += (_, __) => Interlocked.Increment(ref disconnected);
// Assert
await Task.Delay(TimeSpan.FromSeconds(15));
Assert.Equal(1, connected);
Assert.Equal(0, disconnected);
}
public async Task WithDownstreamEventsTest()
{
// Arrange / act
var deviceIdentity = Mock.Of <IIdentity>(i => i.Id == "d2");
var edgeHubIdentity = Mock.Of <IIdentity>(i => i.Id == "d1/m1");
var deviceConnectivityManager = new DeviceConnectivityManager(TimeSpan.FromSeconds(3), TimeSpan.FromSeconds(3), edgeHubIdentity);
int connectedCallbackCount = 0;
int disconnectedCallbackCount = 0;
void ConnectionStatusChangedHandler(ConnectionStatus status, ConnectionStatusChangeReason reason)
{
if (status == ConnectionStatus.Connected && reason == ConnectionStatusChangeReason.Connection_Ok)
{
Interlocked.Increment(ref connectedCallbackCount);
}
else if (status == ConnectionStatus.Disconnected && reason == ConnectionStatusChangeReason.No_Network)
{
Interlocked.Increment(ref disconnectedCallbackCount);
}
}
var device1UnderlyingClient = new Mock <IClient>();
device1UnderlyingClient.Setup(c => c.SendEventAsync(It.IsAny <Message>()))
.Returns(Task.CompletedTask);
var device1Client = new ConnectivityAwareClient(device1UnderlyingClient.Object, deviceConnectivityManager, deviceIdentity);
device1Client.SetConnectionStatusChangedHandler(ConnectionStatusChangedHandler);
var device2UnderlyingClient = new Mock <IClient>();
device2UnderlyingClient.Setup(c => c.SendEventAsync(It.IsAny <Message>()))
.Throws <TimeoutException>();
var device2Client = new ConnectivityAwareClient(device2UnderlyingClient.Object, deviceConnectivityManager, deviceIdentity);
device2Client.SetConnectionStatusChangedHandler(ConnectionStatusChangedHandler);
var edgeHubUnderlyingClient = new Mock <IClient>();
edgeHubUnderlyingClient.Setup(c => c.UpdateReportedPropertiesAsync(It.IsAny <TwinCollection>()))
.Throws <TimeoutException>();
IClient edgeHubClient = new ConnectivityAwareClient(edgeHubUnderlyingClient.Object, deviceConnectivityManager, deviceIdentity);
edgeHubClient.SetConnectionStatusChangedHandler(ConnectionStatusChangedHandler);
ICloudProxy cloudProxy = new CloudProxy(edgeHubClient, Mock.Of <IMessageConverterProvider>(), "d1/m1", null, Mock.Of <ICloudListener>(), TimeSpan.FromHours(1), true);
var connectionManager = Mock.Of <IConnectionManager>(c => c.GetCloudConnection("d1/m1") == Task.FromResult(Option.Some(cloudProxy)));
deviceConnectivityManager.SetConnectionManager(connectionManager);
bool connected = false;
deviceConnectivityManager.DeviceConnected += (_, __) => connected = true;
deviceConnectivityManager.DeviceDisconnected += (_, __) => connected = false;
// Act
var cts = new CancellationTokenSource();
Task t = Task.Run(
async() =>
{
while (!cts.IsCancellationRequested)
{
await device1Client.SendEventAsync(new Message());
if (!cts.IsCancellationRequested)
{
await Task.Delay(TimeSpan.FromSeconds(1));
}
}
});
await Task.Delay(TimeSpan.FromSeconds(5));
// Assert
Assert.True(connected);
edgeHubUnderlyingClient.Verify(c => c.UpdateReportedPropertiesAsync(It.IsAny <TwinCollection>()), Times.Never);
Assert.Equal(3, connectedCallbackCount);
Assert.Equal(0, disconnectedCallbackCount);
cts.Cancel();
await t;
var cts2 = new CancellationTokenSource();
Task t2 = Task.Run(
async() =>
{
while (!cts2.IsCancellationRequested)
{
try
{
await device2Client.SendEventAsync(new Message());
}
catch (TimeoutException)
{
}
if (!cts2.IsCancellationRequested)
{
await Task.Delay(TimeSpan.FromSeconds(1));
}
}
});
await Task.Delay(TimeSpan.FromSeconds(5));
// Assert
Assert.False(connected);
edgeHubUnderlyingClient.Verify(c => c.UpdateReportedPropertiesAsync(It.IsAny <TwinCollection>()), Times.Once);
Assert.Equal(3, connectedCallbackCount);
Assert.Equal(3, disconnectedCallbackCount);
cts2.Cancel();
await t2;
await device1Client.SendEventAsync(new Message());
// Assert
Assert.True(connected);
edgeHubUnderlyingClient.Verify(c => c.UpdateReportedPropertiesAsync(It.IsAny <TwinCollection>()), Times.Once);
Assert.Equal(6, connectedCallbackCount);
Assert.Equal(3, disconnectedCallbackCount);
}