コード例 #1
0
        public async Task CreateAndInitTest()
        {
            // Arrange
            ConnectionStatusChangesHandler connectionStatusChangesHandler = (status, reason) => { };
            DesiredPropertyUpdateCallback  desiredPropertyUpdateCallback  = (properties, context) => Task.CompletedTask;

            var moduleClient = new Mock <IModuleClient>();

            moduleClient.Setup(m => m.IsActive).Returns(true);
            var moduleClientProvider = new Mock <IModuleClientProvider>();

            moduleClientProvider.Setup(m => m.Create(connectionStatusChangesHandler))
            .ReturnsAsync(moduleClient.Object);

            var  requestManager      = new Mock <IRequestManager>();
            bool enableSubscriptions = true;

            // Act
            var moduleConnection = new ModuleConnection(moduleClientProvider.Object, requestManager.Object, connectionStatusChangesHandler, desiredPropertyUpdateCallback, enableSubscriptions);
            await Task.Delay(TimeSpan.FromSeconds(5));

            Option <IModuleClient> resultModuleClientOption = moduleConnection.GetModuleClient();

            // Assert
            Assert.True(resultModuleClientOption.HasValue);
            Assert.Equal(moduleClient.Object, resultModuleClientOption.OrDefault());
            moduleClientProvider.Verify(m => m.Create(connectionStatusChangesHandler), Times.Once);
            moduleClient.Verify(m => m.SetDefaultMethodHandlerAsync(It.IsAny <MethodCallback>()), Times.Once);
            moduleClient.Verify(m => m.SetDesiredPropertyUpdateCallbackAsync(desiredPropertyUpdateCallback), Times.Once);

            // Act
            IModuleClient resultModuleClient = await moduleConnection.GetOrCreateModuleClient();

            // Assert
            Assert.NotNull(resultModuleClient);
            Assert.Equal(moduleClient.Object, resultModuleClient);
            moduleClientProvider.Verify(m => m.Create(connectionStatusChangesHandler), Times.Once);
            moduleClient.Verify(m => m.SetDefaultMethodHandlerAsync(It.IsAny <MethodCallback>()), Times.Once);
            moduleClient.Verify(m => m.SetDesiredPropertyUpdateCallbackAsync(desiredPropertyUpdateCallback), Times.Once);
        }
コード例 #2
0
        public async Task CreateAndCloseTest()
        {
            // Arrange
            ConnectionStatusChangesHandler connectionStatusChangesHandler = (status, reason) => { };
            DesiredPropertyUpdateCallback  desiredPropertyUpdateCallback  = (properties, context) => Task.CompletedTask;

            Task <IModuleClient> GetModuleClient() => Task.FromResult(Mock.Of <IModuleClient>(m => m.IsActive));

            var moduleClientProvider = new Mock <IModuleClientProvider>();

            moduleClientProvider.Setup(m => m.Create(connectionStatusChangesHandler))
            .Returns(GetModuleClient);

            var  requestManager      = new Mock <IRequestManager>();
            bool enableSubscriptions = true;

            // Act
            var           moduleConnection   = new ModuleConnection(moduleClientProvider.Object, requestManager.Object, connectionStatusChangesHandler, desiredPropertyUpdateCallback, enableSubscriptions);
            IModuleClient resultModuleClient = await moduleConnection.GetOrCreateModuleClient();

            Option <IModuleClient> optionResultModuleClient = moduleConnection.GetModuleClient();

            // Assert
            Assert.NotNull(resultModuleClient);
            Assert.True(optionResultModuleClient.HasValue);
            moduleClientProvider.Verify(m => m.Create(connectionStatusChangesHandler), Times.Once);
            Mock <IModuleClient> moduleClient = Mock.Get(resultModuleClient);

            moduleClient.Verify(m => m.SetDefaultMethodHandlerAsync(It.IsAny <MethodCallback>()), Times.Once);
            moduleClient.Verify(m => m.SetDesiredPropertyUpdateCallbackAsync(desiredPropertyUpdateCallback), Times.Once);

            // Act - Set the client to not active and try to get a Get a module client
            moduleClient.Setup(m => m.IsActive).Returns(false);
            optionResultModuleClient = moduleConnection.GetModuleClient();

            // Assert
            Assert.False(optionResultModuleClient.HasValue);

            // Act
            resultModuleClient = await moduleConnection.GetOrCreateModuleClient();

            optionResultModuleClient = moduleConnection.GetModuleClient();

            // Assert
            Assert.NotNull(resultModuleClient);
            Assert.True(optionResultModuleClient.HasValue);
            moduleClient = Mock.Get(resultModuleClient);
            moduleClientProvider.Verify(m => m.Create(connectionStatusChangesHandler), Times.Exactly(2));
            moduleClient.Verify(m => m.SetDefaultMethodHandlerAsync(It.IsAny <MethodCallback>()), Times.Once);
            moduleClient.Verify(m => m.SetDesiredPropertyUpdateCallbackAsync(desiredPropertyUpdateCallback), Times.Once);

            // Act - Set the client to not active and raise the client closed event
            moduleClient.Setup(m => m.IsActive).Returns(false);
            moduleClient.Raise(m => m.Closed += null, new EventArgs());

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

            optionResultModuleClient = moduleConnection.GetModuleClient();

            // Assert
            Assert.True(optionResultModuleClient.HasValue);
            moduleClient = Mock.Get(resultModuleClient);
            moduleClientProvider.Verify(m => m.Create(connectionStatusChangesHandler), Times.Exactly(3));
            moduleClient.Verify(m => m.SetDefaultMethodHandlerAsync(It.IsAny <MethodCallback>()), Times.Once);
            moduleClient.Verify(m => m.SetDesiredPropertyUpdateCallbackAsync(desiredPropertyUpdateCallback), Times.Once);

            // Act
            resultModuleClient = await moduleConnection.GetOrCreateModuleClient();

            // Assert
            Assert.NotNull(resultModuleClient);
            moduleClient = Mock.Get(resultModuleClient);
            moduleClientProvider.Verify(m => m.Create(connectionStatusChangesHandler), Times.Exactly(3));
            moduleClient.Verify(m => m.SetDefaultMethodHandlerAsync(It.IsAny <MethodCallback>()), Times.Once);
            moduleClient.Verify(m => m.SetDesiredPropertyUpdateCallbackAsync(desiredPropertyUpdateCallback), Times.Once);
        }