Add advanced state change tests

net-daemon · Dec 13, 2020 · fb58559 · fb58559
1 parent bb1568c
commit fb58559
Show file tree

Hide file tree

Showing 4 changed files with 69 additions and 3 deletions.
diff --git a/src/Fakes/NetDaemon.Fakes/DaemonHostTestBase.cs b/src/Fakes/NetDaemon.Fakes/DaemonHostTestBase.cs
@@ -164,10 +164,11 @@ public void AddChangedEvent(string entityId, object fromState, object toState)
         /// <summary>
         ///     Adds a full home assistant fake event
         /// </summary>
-        /// <param name="hassEvent">Event to fake</param>
-        public void AddChangedEvent(HassEvent hassEvent)
+        /// <param name="fromState"></param>
+        /// <param name="toState"></param>
+        public void AddChangeEvent(HassState? fromState, HassState? toState)
         {
-            DefaultHassClientMock.FakeEvents.Enqueue(hassEvent);
+            DefaultHassClientMock.AddChangeEventFull(fromState, toState);
         }
 
         /// <summary>

diff --git a/src/Fakes/NetDaemon.Fakes/HassClientMock.cs b/src/Fakes/NetDaemon.Fakes/HassClientMock.cs
@@ -216,6 +216,25 @@ public void AddChangedEvent(string entityId, object fromState, object toState)
             });
         }
 
+        /// <summary>
+        ///     Adds a changed event for entity
+        /// </summary>
+        /// <param name="fromState">The from state advanced details</param>
+        /// <param name="toState">The to state advanced details</param>
+        public void AddChangeEventFull(HassState? fromState, HassState? toState)
+        {
+            FakeEvents.Enqueue(new HassEvent
+            {
+                EventType = "state_changed",
+                Data = new HassStateChangedEventData
+                {
+                    EntityId = toState?.EntityId ?? fromState?.EntityId ?? "",
+                    NewState = toState,
+                    OldState = fromState
+                },
+            });
+        }
+
         /// <summary>
         ///     Adds a custom event
         /// </summary>

diff --git a/tests/NetDaemon.Daemon.Tests/FakesTests/FakeApp.cs b/tests/NetDaemon.Daemon.Tests/FakesTests/FakeApp.cs
@@ -18,6 +18,14 @@ public override void Initialize()
                 {
                     Entity("light.kitchen").TurnOn();
                 });
+
+            Entity("sensor.temperature")
+                .StateAllChanges
+                .Where(e => e.New?.Attribute?.battery_level < 15)
+                .Subscribe(s =>
+                {
+                    this.CallService("notify", "notify", new { title = "Hello from Home Assistant" });
+                });
         }
     }
 }
diff --git a/tests/NetDaemon.Daemon.Tests/FakesTests/FakeTests.cs b/tests/NetDaemon.Daemon.Tests/FakesTests/FakeTests.cs
@@ -3,6 +3,7 @@
 using System.Linq;
 using System.Reactive.Linq;
 using System.Threading.Tasks;
+using JoySoftware.HomeAssistant.Client;
 using Moq;
 using NetDaemon.Common.Fluent;
 using NetDaemon.Common.Reactive;
@@ -462,5 +463,42 @@ public async Task TestFakeAppTurnOnCorrectLight()
             // Verify that netdaemon called light.turn_on 
             VerifyCallService("light", "turn_on", "light.kitchen");
         }
+
+
+        [Fact]
+        public async Task TestFakeAppCallNoteWhenBatteryLevelBelowValue()
+        {
+            // Add the app to test
+            await AddAppInstance(new FakeApp());
+
+            // Init NetDaemon core runtime
+            await FakeDaemonInit().ConfigureAwait(false);
+
+            // Fake a changed event from en entity
+            AddChangeEvent(new()
+            {
+                EntityId = "sensor.temperature",
+                State = 10.0,
+                Attributes = new()
+                {
+                    ["battery_level"] = 18.2
+                }
+            }
+            , new()
+            {
+                EntityId = "sensor.temperature",
+                State = 10.0,
+                Attributes = new()
+                {
+                    ["battery_level"] = 12.0
+                }
+            });
+
+            // Run the NetDemon Core to process the messages
+            await FakeRunDaemonCoreUntilTimeout().ConfigureAwait(false);
+
+            // Verify that netdaemon called light.turn_on 
+            VerifyCallService("notify", "notify", new { title = "Hello from Home Assistant" });
+        }
     }
 }