Fix updating of EventManager.CurrentState

The EventManager methods did not use a pointer as receiver, which caused updated of CurrentState to be lost.
2019-12-09 12:30:37 +01:00
parent 6a3833b27d
commit f8c992a385
2 changed files with 21 additions and 3 deletions
@@ -19,6 +19,24 @@ const (
 	defaultTimeout = 2 * time.Second
 )

+func TestEventManager(t *testing.T) {
+	mgr := EventManager{}
+	mgr.updateState(StateConnected)
+	if mgr.CurrentState != StateConnected {
+		t.Fatal("CurrentState not updated by updateState()")
+	}
+
+	mgr.disconnected(SMState{})
+	if mgr.CurrentState != StateDisconnected {
+		t.Fatalf("CurrentState not reset by disconnected()")
+	}
+
+	mgr.streamError(ErrTLSNotSupported.Error(), "")
+	if mgr.CurrentState != StateStreamError {
+		t.Fatalf("CurrentState not set by streamError()")
+	}
+}
+
 func TestClient_Connect(t *testing.T) {
 	// Setup Mock server
 	mock := ServerMock{}