fix: unit test using wrong sql

taosdata · Dec 2, 2024 · 3b36a59 · 3b36a59
1 parent 7b550ca
commit 3b36a59
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 11 deletions.
diff --git a/plugin/nodeexporter/plugin_test.go b/plugin/nodeexporter/plugin_test.go
@@ -60,7 +60,7 @@ func TestNodeExporter_Gather(t *testing.T) {
 	defer func() {
 		wrapper.TaosClose(conn)
 	}()
-	err = exec(conn, "drop database if exists node_exporter")
+	err = exec(conn, "create database if not exists node_exporter precision 'ns'")
 	assert.NoError(t, err)
 	err = exec(conn, "use node_exporter")
 	assert.NoError(t, err)
@@ -70,7 +70,7 @@ func TestNodeExporter_Gather(t *testing.T) {
 	err = n.Start()
 	assert.NoError(t, err)
 	time.Sleep(time.Second * 2)
-	values, err := query(conn, "select last(`value`) as `value` from node_exporter.go_gc_duration_seconds;")
+	values, err := query(conn, "select last(`value`) as `value` from node_exporter.test_metric;")
 	assert.NoError(t, err)
 	assert.Equal(t, float64(1), values[0][0])
 	err = n.Stop()

diff --git a/plugin/opentsdbtelnet/plugin_test.go b/plugin/opentsdbtelnet/plugin_test.go
@@ -41,7 +41,7 @@ func TestPlugin(t *testing.T) {
 	defer func() {
 		wrapper.TaosClose(conn)
 	}()
-	err = exec(conn, "drop database if exists opentsdb_telnet")
+	err = exec(conn, "create database if not exists opentsdb_telnet")
 	assert.NoError(t, err)
 	err = p.Init(nil)
 	assert.NoError(t, err)

diff --git a/plugin/statsd/plugin_test.go b/plugin/statsd/plugin_test.go
@@ -40,7 +40,7 @@ func TestStatsd(t *testing.T) {
 	defer func() {
 		wrapper.TaosClose(conn)
 	}()
-	err = exec(conn, "drop database if exists statsd")
+	err = exec(conn, "create database if not exists statsd")
 	assert.NoError(t, err)
 	err = p.Init(nil)
 	assert.NoError(t, err)
@@ -58,13 +58,8 @@ func TestStatsd(t *testing.T) {
 	assert.NoError(t, err)
 	time.Sleep(time.Second)
 	defer func() {
-		r := wrapper.TaosQuery(conn, "drop database if exists statsd")
-		code := wrapper.TaosError(r)
-		if code != 0 {
-			errStr := wrapper.TaosErrorStr(r)
-			t.Error(errors.NewError(code, errStr))
-		}
-		wrapper.TaosFreeResult(r)
+		err = exec(conn, "drop database if exists statsd")
+		assert.NoError(t, err)
 	}()
 	values, err := query(conn, "select last(`value`) from statsd.`foo`")
 	assert.NoError(t, err)