62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package httpd_test
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/influxdb/influxql"
|
|
"github.com/influxdata/influxdb/models"
|
|
"github.com/influxdata/influxdb/services/httpd"
|
|
)
|
|
|
|
func TestResponseWriter_CSV(t *testing.T) {
|
|
header := make(http.Header)
|
|
header.Set("Accept", "text/csv")
|
|
r := &http.Request{
|
|
Header: header,
|
|
URL: &url.URL{},
|
|
}
|
|
w := httptest.NewRecorder()
|
|
|
|
writer := httpd.NewResponseWriter(w, r)
|
|
writer.WriteResponse(httpd.Response{
|
|
Results: []*influxql.Result{
|
|
{
|
|
StatementID: 0,
|
|
Series: []*models.Row{
|
|
{
|
|
Name: "cpu",
|
|
Tags: map[string]string{
|
|
"host": "server01",
|
|
"region": "uswest",
|
|
},
|
|
Columns: []string{"time", "value"},
|
|
Values: [][]interface{}{
|
|
{time.Unix(0, 10), float64(2.5)},
|
|
{time.Unix(0, 20), int64(5)},
|
|
{time.Unix(0, 30), nil},
|
|
{time.Unix(0, 40), "foobar"},
|
|
{time.Unix(0, 50), true},
|
|
{time.Unix(0, 60), false},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
if got, want := w.Body.String(), `name,tags,time,value
|
|
cpu,"host=server01,region=uswest",10,2.5
|
|
cpu,"host=server01,region=uswest",20,5
|
|
cpu,"host=server01,region=uswest",30,
|
|
cpu,"host=server01,region=uswest",40,foobar
|
|
cpu,"host=server01,region=uswest",50,true
|
|
cpu,"host=server01,region=uswest",60,false
|
|
`; got != want {
|
|
t.Errorf("unexpected output:\n\ngot=%v\nwant=%s", got, want)
|
|
}
|
|
}
|