FEAT: Add TestPrefixQuery_AllOptions

sdqri · Sep 9, 2024 · 781a2db · 781a2db
1 parent e1e3ce0
commit 781a2db
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/queries/prefixquery/prefix_query_test.go b/queries/prefixquery/prefix_query_test.go
@@ -18,3 +18,30 @@ func TestPrefixQuery_WithNoOptions(t *testing.T) {
 	assert.Nil(t, err)
 	assert.Equal(t, expectedBody, string(jsonBody))
 }
+
+func TestPrefixQuery_AllOptions(t *testing.T) {
+	expectedBody := `{
+		"prefix": {
+			"user.id": {
+				"value": "ki",
+				"rewrite": "top_terms_boost_N",
+				"case_insensitive": true
+			}
+		}
+	}`
+
+	prefixQueryResult := pq.PrefixQuery(
+		"user.id",
+		"ki",
+		pq.WithRewrite(pq.TopTermsBoostN),
+		pq.WithCaseInsensitive(true),
+	)
+
+	err := prefixQueryResult.Err
+	prefixQuery := prefixQueryResult.Ok
+	assert.Nil(t, err)
+	jsonBody, err := json.Marshal(prefixQuery)
+	assert.Nil(t, err)
+	assert.JSONEq(t, expectedBody, string(jsonBody))
+}
+