Function TestAnswersRows

main_test.go:154–204

(t *testing.T)

}

func TestAnswersRows(t *testing.T) {
	log.SetOutput(&bytes.Buffer{})

	reqString := "github.com\none.one.one.one\ngoogle-public-dns-a.google.com"

	req := httptest.NewRequest("POST",
		"http://example.org/query",
		strings.NewReader(reqString))
	w := httptest.NewRecorder()
	queryHandler, err := initQueryHandler(testDbPath, "SELECT * FROM ip_dns WHERE dns = ?", 0)
	if err != nil {
		t.Fatal(err)
	}
	queryHandler(w, req)

	resp := w.Result()
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		t.Fatalf(`resp.StatusCode (%d) != http.StatusOK (%d)`, resp.StatusCode, http.StatusOK)
	}

	if resp.Header.Get("Content-Type") != "application/json" {
		t.Fatalf(`resp.Header.Get("Content-Type") (%s) != "application/json"`, resp.Header.Get("Content-Type"))
	}

	var fullResponse []queryResult
	decoder := json.NewDecoder(resp.Body)
	err = decoder.Decode(&fullResponse)
	if err != nil {
		t.Fatal(err)
	}

	expectedResponse := []queryResult{
		queryResult{
			Out: [][]interface{}{
				[]interface{}{"192.30.253.112", "github.com"},
				[]interface{}{"192.30.253.113", "github.com"},
			}},
		queryResult{
			Out: [][]interface{}{
				[]interface{}{"1.1.1.1", "one.one.one.one"},
			}},
		queryResult{
			Out: [][]interface{}{
				[]interface{}{"8.8.8.8", "google-public-dns-a.google.com"},
			}},
	}

	compare(t, fullResponse, expectedResponse)
}

func TestMoreThanOneParam(t *testing.T) {
	log.SetOutput(&bytes.Buffer{})