test: add unit test

taosdata · Nov 7, 2024 · 8a82319 · 8a82319
1 parent a96feb5
commit 8a82319
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/taosWS/connector_test.go b/taosWS/connector_test.go
@@ -1,6 +1,7 @@
 package taosWS
 
 import (
+	"context"
 	"database/sql"
 	"fmt"
 	"math/rand"
@@ -456,3 +457,14 @@ func TestBatch(t *testing.T) {
 		})
 	}
 }
+
+func TestConnect(t *testing.T) {
+	conn := connector{
+		cfg: &config{},
+	}
+	db, err := conn.Connect(context.Background())
+	assert.NoError(t, err)
+	db.Close()
+	driver := conn.Driver()
+	assert.Equal(t, &TDengineDriver{}, driver)
+}