improve test coverage (#150)

beltran · Jun 30, 2021 · ef83633 · ef83633
1 parent 2d87e3f
commit ef83633
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/hive_test.go b/hive_test.go
@@ -32,6 +32,16 @@ func TestConnectDefault(t *testing.T) {
 	connection.Close()
 }
 
+
+func TestConnectZookeeper(t *testing.T) {
+	configuration := NewConnectConfiguration()
+	configuration.Service = "hive"
+	_, err := ConnectZookeeper("host1:port1,host2:port2", getAuth(), configuration)
+	if err == nil {
+		t.Fatal("error was expected")
+	}
+}
+
 func TestDomainDoesntExist(t *testing.T) {
 	transport := os.Getenv("TRANSPORT")
 	auth := os.Getenv("AUTH")

diff --git a/sasl_transport_test.go b/sasl_transport_test.go
@@ -47,6 +47,7 @@ func TestSaslTransportThrift(t *testing.T) {
 		t.Fatal(err)
 	}
 	trans.Open()
+	trans.SetMaxLength(1638400)
 	socket.Reset()
 	TransportTest(t, trans, trans)
 	if trans.RemainingBytes() != 0 {