CAdd test for connection failed

KacperMalachowski · Mar 1, 2024 · 52ba2ce · 52ba2ce
1 parent 667eeeb
commit 52ba2ce
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/td2/client_test.go b/pkg/td2/client_test.go
@@ -51,6 +51,17 @@ func TestConnectDisconnect(t *testing.T) {
 	}
 }
 
+func TestConnectionFailed(t *testing.T) {
+	ctx := context.Background()
+	stationHash := "exampleHash"
+	client := td2.New(ctx, stationHash)
+
+	err := client.Connect("127.0.0.1:0")
+	if err == nil {
+		t.Error("Expected error on Connect, but got nothing")
+	}
+}
+
 func TestWrite(t *testing.T) {
 	ctx := context.Background()
 	stationHash := "exampleHash"