initial public commit

tools/search/simple_field_resolver_test.go

@@ -0,0 +1,81 @@
+package search_test
+
+import (
+	"testing"
+
+	"github.com/pocketbase/dbx"
+	"github.com/pocketbase/pocketbase/tools/search"
+)
+
+func TestSimpleFieldResolverUpdateQuery(t *testing.T) {
+	r := search.NewSimpleFieldResolver("test")
+
+	scenarios := []struct {
+		fieldName   string
+		expectQuery string
+	}{
+		// missing field (the query shouldn't change)
+		{"", `SELECT "id" FROM "test"`},
+		// unknown field (the query shouldn't change)
+		{"unknown", `SELECT "id" FROM "test"`},
+		// allowed field (the query shouldn't change)
+		{"test", `SELECT "id" FROM "test"`},
+	}
+
+	for i, s := range scenarios {
+		db := dbx.NewFromDB(nil, "")
+		query := db.Select("id").From("test")
+
+		r.Resolve(s.fieldName)
+
+		if err := r.UpdateQuery(nil); err != nil {
+			t.Errorf("(%d) UpdateQuery failed with error %v", i, err)
+			continue
+		}
+
+		rawQuery := query.Build().SQL()
+		// rawQuery := s.expectQuery
+
+		if rawQuery != s.expectQuery {
+			t.Errorf("(%d) Expected query %v, got \n%v", i, s.expectQuery, rawQuery)
+		}
+	}
+}
+
+func TestSimpleFieldResolverResolve(t *testing.T) {
+	r := search.NewSimpleFieldResolver("test", `^test_regex\d+$`, "Test columnify!")
+
+	scenarios := []struct {
+		fieldName   string
+		expectError bool
+		expectName  string
+	}{
+		{"", true, ""},
+		{" ", true, ""},
+		{"unknown", true, ""},
+		{"test", false, "[[test]]"},
+		{"test.sub", true, ""},
+		{"test_regex", true, ""},
+		{"test_regex1", false, "[[test_regex1]]"},
+		{"Test columnify!", false, "[[Testcolumnify]]"},
+	}
+
+	for i, s := range scenarios {
+		name, params, err := r.Resolve(s.fieldName)
+
+		hasErr := err != nil
+		if hasErr != s.expectError {
+			t.Errorf("(%d) Expected hasErr %v, got %v (%v)", i, s.expectError, hasErr, err)
+			continue
+		}
+
+		if name != s.expectName {
+			t.Errorf("(%d) Expected name %q, got %q", i, s.expectName, name)
+		}
+
+		// params should be empty
+		if len(params) != 0 {
+			t.Errorf("(%d) Expected 0 params, got %v", i, params)
+		}
+	}
+}