From a838dac9c0f7f9c8240c33662ef57841f1ca144a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dawid=20Chyrzy=C5=84ski?= <dev@chyrzynski.pl>
Date: Fri, 9 Feb 2024 20:31:15 +0100
Subject: [PATCH] fix tests (#227)

---
 src/HAMqtt.cpp                 | 4 +++-
 src/mocks/PubSubClientMock.cpp | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/HAMqtt.cpp b/src/HAMqtt.cpp
index 526a8dc..ed1c15c 100644
--- a/src/HAMqtt.cpp
+++ b/src/HAMqtt.cpp
@@ -312,7 +312,9 @@ void HAMqtt::connectToServer()
         true
     );
 
-    if (!isConnected()) {
+    if (isConnected()) {
+        setState(StateConnected);
+    } else {
         ARDUINOHA_DEBUG_PRINTLN(F("AHA: failed to connect"))
     }
 }
diff --git a/src/mocks/PubSubClientMock.cpp b/src/mocks/PubSubClientMock.cpp
index 6140871..4562eea 100644
--- a/src/mocks/PubSubClientMock.cpp
+++ b/src/mocks/PubSubClientMock.cpp
@@ -8,7 +8,7 @@ PubSubClientMock::PubSubClientMock() :
     _flushedMessages(nullptr),
     _keepAlive(15),
     _bufferSize(256),
-    _state(0),
+    _state(-1),
     _flushedMessagesNb(0),
     _subscriptions(nullptr),
     _subscriptionsNb(0),