Fixed unit tests

Elenpay · Aug 8, 2023 · f0f98ac · f0f98ac
1 parent 3bbf44f
commit f0f98ac
Showing 1 changed file with 40 additions and 6 deletions.
diff --git a/test/NodeGuard.Tests/Services/LightningServiceTests.cs b/test/NodeGuard.Tests/Services/LightningServiceTests.cs
@@ -1625,7 +1625,7 @@ public async Task GetChannelsStatus_SourceNodeIsManaged_SourceIsInitiator()
         [Fact]
         public async Task GetChannelsStatus_SourceNodeIsManaged_SourceIsNotInitiator()
         {
-// Arrange
+            // Arrange
             var nodeRepository = new Mock<INodeRepository>();
             var lightningClientService = new Mock<ILightningClientService>();
 
@@ -1690,7 +1690,7 @@ public async Task GetChannelsStatus_BothNodesAreManaged_SourceIsInitiator()
                     }
                 });
 
-            var listChannelsResponse = new ListChannelsResponse
+            var listChannelsResponse1 = new ListChannelsResponse
             {
                 Channels =
                 {
@@ -1705,7 +1705,24 @@ public async Task GetChannelsStatus_BothNodesAreManaged_SourceIsInitiator()
                 }
             };
 
-            lightningClientService.Setup(x => x.ListChannels(It.IsAny<Node>(), null)).ReturnsAsync(listChannelsResponse);
+            var listChannelsResponse2 = new ListChannelsResponse
+            {
+                Channels =
+                {
+                    new Lnrpc.Channel
+                    {
+                        ChanId = 0,
+                        LocalBalance = 0,
+                        RemoteBalance = 500,
+                        Initiator = false,
+                        RemotePubkey = "managedPubKey1"
+                    }
+                }
+            };
+
+            lightningClientService.SetupSequence(x => x.ListChannels(It.IsAny<Node>(), null))
+                .ReturnsAsync(listChannelsResponse1)
+                .ReturnsAsync(listChannelsResponse2);
             var lightningService = new LightningService(null, null, nodeRepository.Object, null, null, null, null, null ,null, lightningClientService.Object);
 
             // Act
@@ -1740,7 +1757,22 @@ public async Task GetChannelsStatus_BothNodesAreManaged_SourceIsNotInitiator()
                     }
                 });
 
-            var listChannelsResponse = new ListChannelsResponse
+            var listChannelsResponse1 = new ListChannelsResponse
+            {
+                Channels =
+                {
+                    new Lnrpc.Channel
+                    {
+                        ChanId = 0,
+                        LocalBalance = 0,
+                        RemoteBalance = 500,
+                        Initiator = false,
+                        RemotePubkey = "managedPubKey2"
+                    }
+                }
+            };
+
+            var listChannelsResponse2 = new ListChannelsResponse
             {
                 Channels =
                 {
@@ -1750,12 +1782,14 @@ public async Task GetChannelsStatus_BothNodesAreManaged_SourceIsNotInitiator()
                         LocalBalance = 500,
                         RemoteBalance = 0,
                         Initiator = true,
-                        RemotePubkey = "managedPubKey2"
+                        RemotePubkey = "managedPubKey1"
                     }
                 }
             };
 
-            lightningClientService.Setup(x => x.ListChannels(It.IsAny<Node>(), null)).ReturnsAsync(listChannelsResponse);
+            lightningClientService.SetupSequence(x => x.ListChannels(It.IsAny<Node>(), null))
+                .ReturnsAsync(listChannelsResponse1)
+                .ReturnsAsync(listChannelsResponse2);
             var lightningService = new LightningService(null, null, nodeRepository.Object, null, null, null, null, null ,null, lightningClientService.Object);
 
             // Act