aboutsummaryrefslogtreecommitdiffstats
path: root/client/go/src/cmd/query_test.go
blob: e062e9c718af8f13b56581ce844e34eef0beb3c5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// query command tests
// Author: bratseth

package cmd

import (
    "github.com/stretchr/testify/assert"
    "testing"
)

func TestQuery(t *testing.T) {
    assertQuery(t,
                "?yql=select+from+sources+%2A+where+title+contains+%27foo%27",
                "select from sources * where title contains 'foo'")
}

func TestQueryWithMultipleParameters(t *testing.T) {
    assertQuery(t,
                "?hits=5&yql=select+from+sources+%2A+where+title+contains+%27foo%27",
                "select from sources * where title contains 'foo'", "hits=5")
}

func TestQueryWithExplicitYqlParameter(t *testing.T) {
    assertQuery(t,
                "?yql=select+from+sources+%2A+where+title+contains+%27foo%27",
                "yql=select from sources * where title contains 'foo'")
}

func assertQuery(t *testing.T, expectedQuery string, query ...string) {
    client := &mockHttpClient{ nextBody: "query result", }
	assert.Equal(t,
	             "query result\n",
	             executeCommand(t, client, []string{"query"}, query),
	             "query output")
    assert.Equal(t, getTarget(queryContext).query + "/search/" + expectedQuery, client.lastRequest.URL.String())
}