Fix some warnings in tesproject (#400)

* Fix some warnings in tesproject
net-daemon · Jul 5, 2021 · 6b42d11 · 6b42d11
1 parent 5660244
commit 6b42d11
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/Fakes/NetDaemon.Fakes/DaemonHostTestBase.cs b/src/Fakes/NetDaemon.Fakes/DaemonHostTestBase.cs
@@ -142,7 +142,7 @@ public void SetEntityState(string entityId, object? state = null, string? area =
             DefaultDaemonHost.StateManager.Store(new HassState()
             {
                 EntityId = entityId,
-                State = state.ToString()
+                State = state?.ToString()
             });
 
             if (area == null) return;

diff --git a/src/Fakes/NetDaemon.Fakes/HassClientMock.cs b/src/Fakes/NetDaemon.Fakes/HassClientMock.cs
@@ -170,7 +170,7 @@ public void AddChangedEvent(string entityId, object fromState, object toState, D
                     EntityId = entityId,
                     NewState = new HassState
                     {
-                        State = toState.ToString(),
+                        State = toState?.ToString(),
                         Attributes = new Dictionary<string, object>
                         {
                             ["device_class"] = "motion"
@@ -180,7 +180,7 @@ public void AddChangedEvent(string entityId, object fromState, object toState, D
                     },
                     OldState = new HassState
                     {
-                        State = fromState.ToString(),
+                        State = fromState?.ToString(),
                         Attributes = new Dictionary<string, object>
                         {
                             ["device_class"] = "motion"
@@ -207,7 +207,7 @@ public void AddChangedEvent(string entityId, object fromState, object toState)
                     NewState = new HassState
                     {
                         EntityId = entityId,
-                        State = toState.ToString(),
+                        State = toState?.ToString(),
                         Attributes = new Dictionary<string, object>
                         {
                             ["device_class"] = "motion"
@@ -218,7 +218,7 @@ public void AddChangedEvent(string entityId, object fromState, object toState)
                     OldState = new HassState
                     {
                         EntityId = entityId,
-                        State = fromState.ToString(),
+                        State = fromState?.ToString(),
                         Attributes = new Dictionary<string, object>
                         {
                             ["device_class"] = "motion"