feat: 支持漏洞关联多条流量证据

evidence/testmain_test.go

@@ -0,0 +1,35 @@
+package evidence
+
+import (
+	"context"
+	"os"
+	"testing"
+
+	"github.com/Autumn-27/artex/db"
+)
+
+// Initialize an explicitly configured fresh database before taking the same
+// suite lock as db, agent and server. Hold it on one pinned connection.
+func TestMain(m *testing.M) {
+	if os.Getenv("ARTEX_PG_DSN") == "" {
+		os.Exit(m.Run())
+	}
+	os.Exit(runEvidenceSuite(m))
+}
+func runEvidenceSuite(m *testing.M) int {
+	pg, err := db.Open(os.Getenv("ARTEX_PG_DSN"))
+	if err != nil {
+		panic(err)
+	}
+	defer pg.Close()
+	conn, err := pg.Conn(context.Background())
+	if err != nil {
+		panic(err)
+	}
+	defer conn.Close()
+	if _, err = conn.ExecContext(context.Background(), `SELECT pg_advisory_lock(7337741002)`); err != nil {
+		panic(err)
+	}
+	defer conn.ExecContext(context.Background(), `SELECT pg_advisory_unlock(7337741002)`)
+	return m.Run()
+}