Exemple #1
0
        public async Task HandleExceptionsTest(Type exception)
        {
            // Arrange
            MethodCallback testMethodCallback = (request, context) => Task.FromResult(new MethodResponse(200));

            var sdkModuleClient = new Mock <ISdkModuleClient>(MockBehavior.Strict);

            sdkModuleClient.Setup(s => s.CloseAsync())
            .Returns(Task.CompletedTask);
            sdkModuleClient.Setup(s => s.SetDefaultMethodHandlerAsync(testMethodCallback))
            .ThrowsAsync((Exception)Activator.CreateInstance(exception, "Dummy exception"));

            TimeSpan idleTimeout        = TimeSpan.FromMinutes(3);
            bool     closeOnIdleTimeout = false;

            // Act
            var moduleClient = new ModuleClient(sdkModuleClient.Object, idleTimeout, closeOnIdleTimeout, Core.UpstreamProtocol.Mqtt);

            Assert.Equal(Core.UpstreamProtocol.Mqtt, moduleClient.UpstreamProtocol);
            Assert.True(moduleClient.IsActive);
            await Assert.ThrowsAsync(exception, () => moduleClient.SetDefaultMethodHandlerAsync(testMethodCallback));

            // Assert
            sdkModuleClient.Verify(s => s.SetDefaultMethodHandlerAsync(testMethodCallback), Times.Once);
            sdkModuleClient.Verify(s => s.CloseAsync(), Times.Once);
        }
Exemple #2
0
        public async Task CloseOnInactivityGetsResetTest()
        {
            // Arrange
            MethodCallback testMethodCallback = (request, context) => Task.FromResult(new MethodResponse(200));

            var sdkModuleClient = new Mock <ISdkModuleClient>(MockBehavior.Strict);

            sdkModuleClient.Setup(s => s.CloseAsync())
            .Returns(Task.CompletedTask);
            sdkModuleClient.Setup(s => s.SetDefaultMethodHandlerAsync(testMethodCallback))
            .Returns(Task.CompletedTask);

            TimeSpan idleTimeout        = TimeSpan.FromSeconds(3);
            bool     closeOnIdleTimeout = true;

            // Act
            var moduleClient = new ModuleClient(sdkModuleClient.Object, idleTimeout, closeOnIdleTimeout, Core.UpstreamProtocol.Amqp);

            Assert.True(moduleClient.IsActive);
            Assert.Equal(Core.UpstreamProtocol.Amqp, moduleClient.UpstreamProtocol);

            // Assert
            await Task.Delay(TimeSpan.FromSeconds(2));

            await moduleClient.SetDefaultMethodHandlerAsync(testMethodCallback);

            Assert.True(moduleClient.IsActive);

            await Task.Delay(TimeSpan.FromSeconds(2));

            await moduleClient.SetDefaultMethodHandlerAsync(testMethodCallback);

            Assert.True(moduleClient.IsActive);

            await Task.Delay(TimeSpan.FromSeconds(2));

            await moduleClient.SetDefaultMethodHandlerAsync(testMethodCallback);

            Assert.True(moduleClient.IsActive);

            Assert.True(moduleClient.IsActive);
            await Task.Delay(TimeSpan.FromSeconds(5));

            Assert.False(moduleClient.IsActive);
            sdkModuleClient.Verify(s => s.CloseAsync(), Times.Once);
        }