Function TestHandleSearch

modules/frontend/mcp_tools_test.go:45–123

(t *testing.T)

}

func TestHandleSearch(t *testing.T) {
	server, callAndTestResults := testFrontend()

	tests := []struct {
		name     string
		request  mcp.CallToolRequest
		expected expectedResult
	}{
		{
			name: "query only",
			request: callToolRequest(map[string]any{
				"query": "{ span.foo = \"bar\" }",
			}),
			expected: expectedResult{
				path: "/api/search",
				params: map[string]string{
					"q": "{ span.foo = \"bar\" }",
				},
				meta: map[string]any{
					"type":     "search-results",
					"encoding": "json",
					"version":  "1",
				},
			},
		},
		{
			name:    "no query!",
			request: callToolRequest(map[string]any{}),
			expected: expectedResult{
				err: "required argument \"query\" not found",
			},
		},
		{
			name: "query + start + end",
			request: callToolRequest(map[string]any{
				"query": "{ span.foo = \"bar\" }",
				"start": "2022-01-01T00:00:00Z",
				"end":   "2022-01-02T00:00:00Z",
			}),
			expected: expectedResult{
				path: "/api/search",
				params: map[string]string{
					"q":     "{ span.foo = \"bar\" }",
					"start": "1640995200",
					"end":   "1641081600",
				},
				meta: map[string]any{
					"type":     "search-results",
					"encoding": "json",
					"version":  "1",
				},
			},
		},
		{
			name: "bad query",
			request: callToolRequest(map[string]any{
				"query": "{ foo bar baz }",
			}),