add test case for concurrent query

antchfx · Jan 26, 2023 · defe049 · defe049
1 parent e655bcc
commit defe049
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/query_test.go b/query_test.go
@@ -7,6 +7,7 @@ import (
 	"net/http/httptest"
 	"os"
 	"strings"
+	"sync"
 	"testing"
 
 	"github.com/antchfx/xpath"
@@ -158,6 +159,22 @@ func TestXPathCdUp(t *testing.T) {
 	}
 }
 
+func TestConcurrentQuery(t *testing.T) {
+	var wg sync.WaitGroup
+	for i := 0; i < 10; i++ {
+		wg.Add(1)
+		go func(i int) {
+			defer wg.Done()
+			s := `<html><head></head><body><div>a</div></body>`
+			doc := loadHTML(s)
+			if n := FindOne(doc, `//div`); n == nil {
+				t.Fatalf("should find one but got nil [%d]", i)
+			}
+		}(i)
+	}
+	wg.Done()
+}
+
 func loadHTML(str string) *html.Node {
 	node, err := Parse(strings.NewReader(str))
 	if err != nil {