add test

go/vt/topo/zk2topo/zk_conn_test.go

@@ -21,7 +21,6 @@ import (
 	"testing"
 
 	"github.com/z-division/go-zookeeper/zk"
-
 	"vitess.io/vitess/go/testfiles"
 	"vitess.io/vitess/go/vt/zkctl"
 )
@@ -36,7 +35,7 @@ func TestZkConnClosedOnDisconnect(t *testing.T) {
 		t.Fatalf("Get() failed: %v", err)
 	}
 
-	if conn.conn.State() != zk.StateConnected {
+	if !conn.conn.State().IsConnected() {
 		t.Fatalf("Connection not connected: %v", conn.conn.State())
 	}
 
@@ -52,7 +51,12 @@ func TestZkConnClosedOnDisconnect(t *testing.T) {
 		t.Fatalf("Get() failed: %v", err)
 	}
 
-	// check that the old connection is closed
+	// Check that old connection is closed
+	_, _, err = oldConn.Get("/")
+	if err == nil {
+		t.Fatalf("Get() should have failed: %v", err)
+	}
+
 	if oldConn.State() != zk.StateDisconnected {
 		t.Fatalf("Connection not closed: %v", oldConn.State())
 	}