Search API (#350)

2026-01-17 21:34:56 +00:00 · 2022-01-07 09:58:10 +00:00
parent 0f00267922
commit a093abaf5e
16 changed files with 2388 additions and 134 deletions
--- a/search/handler/parser_test.go
+++ b/search/handler/parser_test.go
@@ -0,0 +1,130 @@
+package handler
+
+import (
+	"fmt"
+	"testing"
+
+	. "github.com/onsi/gomega"
+)
+
+func TestParsing(t *testing.T) {
+
+	tcs := []struct {
+		name   string
+		input  string
+		output string
+		err    error
+	}{
+		{
+			name:   "basic",
+			input:  `foo == "bar"`,
+			output: `{"query":{"bool":{"must":[{"match":{"foo":"bar"}}]}}}`,
+		},
+		{
+			name:   "basic",
+			input:  `first_name == 'Dom'`,
+			output: `{"query":{"bool":{"must":[{"match":{"first_name":"Dom"}}]}}}`,
+		},
+		{
+			name:   "basic bool",
+			input:  `foo == true`,
+			output: `{"query":{"bool":{"must":[{"match":{"foo":"true"}}]}}}`,
+		},
+		{
+			name:   "basic bool false",
+			input:  `foo == false`,
+			output: `{"query":{"bool":{"must":[{"match":{"foo":"false"}}]}}}`,
+		},
+		{
+			name:   "basic with spaces",
+			input:  `foo == "hello there"`,
+			output: `{"query":{"bool":{"must":[{"match":{"foo":"hello there"}}]}}}`,
+		},
+		{
+			name:   "basic number",
+			input:  `foo == 123987`,
+			output: `{"query":{"bool":{"must":[{"match":{"foo":"123987"}}]}}}`,
+		},
+		{
+			name:   "basic gt number",
+			input:  `foo >= 123987`,
+			output: `{"query":{"bool":{"must":[{"range":{"foo":{"gte":"123987"}}}]}}}`,
+		},
+		{
+			name:   "basic lt number",
+			input:  `foo <= 123987`,
+			output: `{"query":{"bool":{"must":[{"range":{"foo":{"lte":"123987"}}}]}}}`,
+		},
+		{
+			name:   "AND bool",
+			input:  `foo == 'bar' AND baz == 'hello'`,
+			output: `{"query":{"bool":{"must":[{"match":{"foo":"bar"}},{"match":{"baz":"hello"}}]}}}`,
+		},
+		{
+			name:   "and bool",
+			input:  `foo == 'bar' and baz == 'hello'`,
+			output: `{"query":{"bool":{"must":[{"match":{"foo":"bar"}},{"match":{"baz":"hello"}}]}}}`,
+		},
+		{
+			name:   "OR bool",
+			input:  `foo == 'bar' OR baz == 'hello'`,
+			output: `{"query":{"bool":{"should":[{"match":{"foo":"bar"}},{"match":{"baz":"hello"}}]}}}`,
+		},
+		{
+			name:   "or bool",
+			input:  `foo == 'bar' or baz == 'hello'`,
+			output: `{"query":{"bool":{"should":[{"match":{"foo":"bar"}},{"match":{"baz":"hello"}}]}}}`,
+		},
+		{
+			name:  "bad val",
+			input: `foo == bar`,
+			err:   fmt.Errorf("blah"),
+		},
+		{
+			name:  "gibberish",
+			input: `123onddlqkjn oajsldkj`,
+			err:   fmt.Errorf("blah"),
+		},
+		{
+			name:  "gibberish",
+			input: `123onddlqkjn`,
+			err:   fmt.Errorf("blah"),
+		},
+		{
+			name:   "brackets",
+			input:  `foo == 'bar' and (baz == 'hello' or customer.name == 'john doe')`,
+			output: `{"query":{"bool":{"must":[{"match":{"foo":"bar"}},{"bool":{"should":[{"match":{"baz":"hello"}},{"match":{"customer.name":"john doe"}}]}}]}}}`,
+		},
+		{
+			name:   "gte",
+			input:  `foo >= 6`,
+			output: `{"query":{"bool":{"must":[{"range":{"foo":{"gte":"6"}}}]}}}`,
+		},
+		{
+			name:   "lte",
+			input:  `foo <= 6`,
+			output: `{"query":{"bool":{"must":[{"range":{"foo":{"lte":"6"}}}]}}}`,
+		},
+		{
+			name:   "wildcard",
+			input:  `foo == "ba*"`,
+			output: `{"query":{"bool":{"must":[{"wildcard":{"foo":{"value":"ba*"}}}]}}}`,
+		},
+	}
+	for _, tc := range tcs {
+		t.Run(tc.name, func(t *testing.T) {
+			g := NewWithT(t)
+			js, err := parseQueryString(tc.input)
+			if tc.err != nil {
+				g.Expect(err).To(Not(BeNil()))
+			} else {
+				b, _ := js.MarshalJSON()
+				t.Logf("%+v", string(b))
+				g.Expect(err).To(BeNil())
+				g.Expect(string(b)).To(Equal(tc.output))
+			}
+
+		})
+	}
+
+}