cmd/trace-agent: add test case

cmd/trace-agent/agent_test.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"net/http"
+	"regexp"
 	"runtime"
 	"strings"
 	"testing"
@@ -107,6 +108,33 @@ func TestFormatTrace(t *testing.T) {
 	assert.Contains(result.Meta["sql.query"], "SELECT name FROM people WHERE age = ?")
 }
 
+func TestSQLResourceReplacer(t *testing.T) {
+	cfg := config.New()
+	cfg.APIKey = "test"
+	cfg.Ignore["resource"] = []string{}
+	cfg.ReplaceTags = []*config.ReplaceRule{{
+		Name: "resource.name",
+		Re:   regexp.MustCompile("AND.*"),
+		Repl: "...",
+	}}
+	ctx, cancel := context.WithCancel(context.Background())
+	agent := NewAgent(ctx, cfg)
+	defer cancel()
+
+	now := time.Now()
+	span := &model.Span{
+		Resource: "SELECT name FROM people WHERE age = 42 AND extra = 55",
+		Type:     "sql",
+		Start:    now.Add(-time.Second).UnixNano(),
+		Duration: (500 * time.Millisecond).Nanoseconds(),
+	}
+	agent.Process(model.Trace{span})
+
+	assert := assert.New(t)
+	assert.Equal("SELECT name FROM people WHERE age = ? ...", span.Resource)
+	assert.Equal("SELECT name FROM people WHERE age = ? AND extra = ?", span.Meta["sql.query"])
+}
+
 func BenchmarkAgentTraceProcessing(b *testing.B) {
 	c := config.New()
 	c.APIKey = "test"