fix state switches

pubnub · Feb 5, 2024 · e03a01f · e03a01f
1 parent eae3301
commit e03a01f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/Api/PubnubApi/EventEngine/Presence/States/CooldownState.cs b/src/Api/PubnubApi/EventEngine/Presence/States/CooldownState.cs
@@ -23,10 +23,6 @@ public override TransitionResult Transition(IEvent ev)
                 },
                 Events.LeftEvent e => HandleLeftEvent(e),
                 Events.LeftAllEvent e => new InactiveState(),
-                Events.HeartbeatSuccessEvent e => new CooldownState()
-                {
-                    Input = this.Input,
-                },
                 Events.DisconnectEvent e => new StoppedState()
                 {
                     Input = this.Input,

diff --git a/src/Api/PubnubApi/EventEngine/Presence/States/HeartBeatingState.cs b/src/Api/PubnubApi/EventEngine/Presence/States/HeartBeatingState.cs
@@ -22,6 +22,10 @@ public override TransitionResult Transition(IEvent ev)
                 },
                 Events.LeftEvent e => HandleLeftEvent(e),
                 Events.LeftAllEvent e => new InactiveState(),
+                Events.HeartbeatSuccessEvent e => new CooldownState()
+                {
+                    Input = this.Input,
+                },
                 Events.HeartbeatFailureEvent e => HandleHeartbeatFailureEvent(e), 
                 Events.DisconnectEvent e => new StoppedState()
                 {