Fixed test

net-daemon · Apr 25, 2020 · 88150ac · 88150ac
1 parent 2a53811
commit 88150ac
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 11 deletions.
diff --git a/tests/NetDaemon.Daemon.Tests/Demon/NetDaemonHostTests.cs b/tests/NetDaemon.Daemon.Tests/Demon/NetDaemonHostTests.cs
@@ -155,17 +155,6 @@ public void RunNullReferenceToHassClientShouldThrowException()
                 { var DefaultDaemonHost = new NetDaemonHost(null, null); });
         }
 
-        [Fact]
-        public async Task RunWhenCanceledShouldCompleteWithCanceledException()
-        {
-            // ARRANGE
-
-            // ACT and ASSERT
-            await RunDefauldDaemonUntilCanceled(20);
-
-            Assert.False(DefaultDaemonHost.Connected);
-        }
-
         [Fact]
         public async Task SendEventShouldCallCorrectMethod()
         {

diff --git a/tests/NetDaemon.Daemon.Tests/HassClientMock.cs b/tests/NetDaemon.Daemon.Tests/HassClientMock.cs
@@ -32,6 +32,9 @@ public HassClientMock()
             Setup(x => x.ReadEventAsync())
                 .ReturnsAsync(() => { return FakeEvents.TryDequeue(out var ev) ? ev : null; });
 
+            Setup(x => x.ReadEventAsync(It.IsAny<CancellationToken>()))
+                .ReturnsAsync(() => { return FakeEvents.TryDequeue(out var ev) ? ev : null; });
+
             Setup(x => x.SetState(It.IsAny<string>(), It.IsAny<string>(), It.IsAny<object?>())).Returns<string, string, object>(
                 (entityId, state, attributes) =>
                 {