test(server): add cell http handler tests
Michael Desa
parent
203d234464
commit
c0a10c4cb1
|
|
@ -0,0 +1,553 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/bouk/httprouter"
|
||||
"github.com/influxdata/chronograf/log"
|
||||
"github.com/influxdata/chronograf/mocks"
|
||||
chronograf "github.com/influxdata/chronograf/v2"
|
||||
)
|
||||
|
||||
func TestService_CellsV2(t *testing.T) {
|
||||
type fields struct {
|
||||
CellService chronograf.CellService
|
||||
}
|
||||
type args struct {
|
||||
queryParams map[string][]string
|
||||
}
|
||||
type wants struct {
|
||||
statusCode int
|
||||
contentType string
|
||||
body string
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
fields fields
|
||||
args args
|
||||
wants wants
|
||||
}{
|
||||
{
|
||||
name: "get all cells",
|
||||
fields: fields{
|
||||
&mocks.CellService{
|
||||
FindCellsF: func(ctx context.Context, filter chronograf.CellFilter) ([]*chronograf.Cell, int, error) {
|
||||
return []*chronograf.Cell{
|
||||
{
|
||||
CellContents: chronograf.CellContents{
|
||||
ID: chronograf.ID("0"),
|
||||
Name: "hello",
|
||||
},
|
||||
Visualization: chronograf.V1Visualization{
|
||||
Type: "line",
|
||||
},
|
||||
},
|
||||
{
|
||||
CellContents: chronograf.CellContents{
|
||||
ID: chronograf.ID("2"),
|
||||
Name: "example",
|
||||
},
|
||||
},
|
||||
}, 2, nil
|
||||
},
|
||||
},
|
||||
},
|
||||
args: args{},
|
||||
wants: wants{
|
||||
statusCode: http.StatusOK,
|
||||
contentType: "application/json",
|
||||
body: `
|
||||
{
|
||||
"links": {
|
||||
"self": "/chronograf/v2/cells"
|
||||
},
|
||||
"cells": [
|
||||
{
|
||||
"id": "0",
|
||||
"name": "hello",
|
||||
"links": {
|
||||
"self": "/chronograf/v1/cells/0"
|
||||
},
|
||||
"visualization": {
|
||||
"type": "chronograf-v1",
|
||||
"queries": null,
|
||||
"axes": null,
|
||||
"visualizationType": "line",
|
||||
"colors": null,
|
||||
"legend": {},
|
||||
"tableOptions": {
|
||||
"verticalTimeAxis": false,
|
||||
"sortBy": {
|
||||
"internalName": "",
|
||||
"displayName": "",
|
||||
"visible": false
|
||||
},
|
||||
"wrapping": "",
|
||||
"fixFirstColumn": false
|
||||
},
|
||||
"fieldOptions": null,
|
||||
"timeFormat": "",
|
||||
"decimalPlaces": {
|
||||
"isEnforced": false,
|
||||
"digits": 0
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"id": "2",
|
||||
"name": "example",
|
||||
"links": {
|
||||
"self": "/chronograf/v1/cells/2"
|
||||
},
|
||||
"visualization": {
|
||||
"type": "empty"
|
||||
}
|
||||
}
|
||||
]
|
||||
}`,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
s := &Service{
|
||||
Store: &mocks.Store{
|
||||
CellService: tt.fields.CellService,
|
||||
},
|
||||
Logger: log.New(log.DebugLevel),
|
||||
}
|
||||
|
||||
r := httptest.NewRequest("GET", "http://any.url", nil)
|
||||
|
||||
qp := r.URL.Query()
|
||||
for k, vs := range tt.args.queryParams {
|
||||
for _, v := range vs {
|
||||
qp.Add(k, v)
|
||||
}
|
||||
}
|
||||
r.URL.RawQuery = qp.Encode()
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
|
||||
s.CellsV2(w, r)
|
||||
|
||||
res := w.Result()
|
||||
content := res.Header.Get("Content-Type")
|
||||
body, _ := ioutil.ReadAll(res.Body)
|
||||
|
||||
if res.StatusCode != tt.wants.statusCode {
|
||||
t.Errorf("%q. CellsV2() = %v, want %v", tt.name, res.StatusCode, tt.wants.statusCode)
|
||||
}
|
||||
if tt.wants.contentType != "" && content != tt.wants.contentType {
|
||||
t.Errorf("%q. CellsV2() = %v, want %v", tt.name, content, tt.wants.contentType)
|
||||
}
|
||||
if eq, _ := jsonEqual(string(body), tt.wants.body); tt.wants.body != "" && !eq {
|
||||
t.Errorf("%q. CellsV2() = \n***%v***\n,\nwant\n***%v***", tt.name, string(body), tt.wants.body)
|
||||
}
|
||||
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestService_CellIDV2(t *testing.T) {
|
||||
type fields struct {
|
||||
CellService chronograf.CellService
|
||||
}
|
||||
type args struct {
|
||||
id string
|
||||
}
|
||||
type wants struct {
|
||||
statusCode int
|
||||
contentType string
|
||||
body string
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
fields fields
|
||||
args args
|
||||
wants wants
|
||||
}{
|
||||
{
|
||||
name: "get a cell by id",
|
||||
fields: fields{
|
||||
&mocks.CellService{
|
||||
FindCellByIDF: func(ctx context.Context, id chronograf.ID) (*chronograf.Cell, error) {
|
||||
if id == "2" {
|
||||
return &chronograf.Cell{
|
||||
CellContents: chronograf.CellContents{
|
||||
ID: chronograf.ID("2"),
|
||||
Name: "example",
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
return nil, fmt.Errorf("not found")
|
||||
},
|
||||
},
|
||||
},
|
||||
args: args{
|
||||
id: "2",
|
||||
},
|
||||
wants: wants{
|
||||
statusCode: http.StatusOK,
|
||||
contentType: "application/json",
|
||||
body: `
|
||||
{
|
||||
"id": "2",
|
||||
"name": "example",
|
||||
"links": {
|
||||
"self": "/chronograf/v1/cells/2"
|
||||
},
|
||||
"visualization": {
|
||||
"type": "empty"
|
||||
}
|
||||
}
|
||||
`,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
s := &Service{
|
||||
Store: &mocks.Store{
|
||||
CellService: tt.fields.CellService,
|
||||
},
|
||||
Logger: log.New(log.DebugLevel),
|
||||
}
|
||||
|
||||
r := httptest.NewRequest("GET", "http://any.url", nil)
|
||||
|
||||
r = r.WithContext(httprouter.WithParams(
|
||||
context.Background(),
|
||||
httprouter.Params{
|
||||
{
|
||||
Key: "id",
|
||||
Value: tt.args.id,
|
||||
},
|
||||
}))
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
|
||||
s.CellIDV2(w, r)
|
||||
|
||||
res := w.Result()
|
||||
content := res.Header.Get("Content-Type")
|
||||
body, _ := ioutil.ReadAll(res.Body)
|
||||
|
||||
if res.StatusCode != tt.wants.statusCode {
|
||||
t.Errorf("%q. CellIDV2() = %v, want %v", tt.name, res.StatusCode, tt.wants.statusCode)
|
||||
}
|
||||
if tt.wants.contentType != "" && content != tt.wants.contentType {
|
||||
t.Errorf("%q. CellIDV2() = %v, want %v", tt.name, content, tt.wants.contentType)
|
||||
}
|
||||
if eq, _ := jsonEqual(string(body), tt.wants.body); tt.wants.body != "" && !eq {
|
||||
t.Errorf("%q. CellIDV2() = \n***%v***\n,\nwant\n***%v***", tt.name, string(body), tt.wants.body)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestService_NewCellV2(t *testing.T) {
|
||||
type fields struct {
|
||||
CellService chronograf.CellService
|
||||
}
|
||||
type args struct {
|
||||
cell *chronograf.Cell
|
||||
}
|
||||
type wants struct {
|
||||
statusCode int
|
||||
contentType string
|
||||
body string
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
fields fields
|
||||
args args
|
||||
wants wants
|
||||
}{
|
||||
{
|
||||
name: "create a new cell",
|
||||
fields: fields{
|
||||
&mocks.CellService{
|
||||
CreateCellF: func(ctx context.Context, c *chronograf.Cell) error {
|
||||
c.ID = "2"
|
||||
return nil
|
||||
},
|
||||
},
|
||||
},
|
||||
args: args{
|
||||
cell: &chronograf.Cell{
|
||||
CellContents: chronograf.CellContents{
|
||||
Name: "hello",
|
||||
},
|
||||
Visualization: chronograf.V1Visualization{
|
||||
Type: "line",
|
||||
},
|
||||
},
|
||||
},
|
||||
wants: wants{
|
||||
statusCode: http.StatusCreated,
|
||||
contentType: "application/json",
|
||||
body: `
|
||||
{
|
||||
"id": "2",
|
||||
"name": "hello",
|
||||
"links": {
|
||||
"self": "/chronograf/v1/cells/2"
|
||||
},
|
||||
"visualization": {
|
||||
"type": "chronograf-v1",
|
||||
"queries": null,
|
||||
"axes": null,
|
||||
"visualizationType": "line",
|
||||
"colors": null,
|
||||
"legend": {},
|
||||
"tableOptions": {
|
||||
"verticalTimeAxis": false,
|
||||
"sortBy": {
|
||||
"internalName": "",
|
||||
"displayName": "",
|
||||
"visible": false
|
||||
},
|
||||
"wrapping": "",
|
||||
"fixFirstColumn": false
|
||||
},
|
||||
"fieldOptions": null,
|
||||
"timeFormat": "",
|
||||
"decimalPlaces": {
|
||||
"isEnforced": false,
|
||||
"digits": 0
|
||||
}
|
||||
}
|
||||
}
|
||||
`,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
s := &Service{
|
||||
Store: &mocks.Store{
|
||||
CellService: tt.fields.CellService,
|
||||
},
|
||||
Logger: log.New(log.DebugLevel),
|
||||
}
|
||||
|
||||
b, err := json.Marshal(tt.args.cell)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to unmarshal cell: %v", err)
|
||||
}
|
||||
|
||||
r := httptest.NewRequest("GET", "http://any.url", bytes.NewReader(b))
|
||||
w := httptest.NewRecorder()
|
||||
|
||||
s.NewCellV2(w, r)
|
||||
|
||||
res := w.Result()
|
||||
content := res.Header.Get("Content-Type")
|
||||
body, _ := ioutil.ReadAll(res.Body)
|
||||
|
||||
if res.StatusCode != tt.wants.statusCode {
|
||||
t.Errorf("%q. CellIDV2() = %v, want %v", tt.name, res.StatusCode, tt.wants.statusCode)
|
||||
}
|
||||
if tt.wants.contentType != "" && content != tt.wants.contentType {
|
||||
t.Errorf("%q. CellIDV2() = %v, want %v", tt.name, content, tt.wants.contentType)
|
||||
}
|
||||
if eq, _ := jsonEqual(string(body), tt.wants.body); tt.wants.body != "" && !eq {
|
||||
t.Errorf("%q. CellIDV2() = \n***%v***\n,\nwant\n***%v***", tt.name, string(body), tt.wants.body)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestService_RemoveCellV2(t *testing.T) {
|
||||
type fields struct {
|
||||
CellService chronograf.CellService
|
||||
}
|
||||
type args struct {
|
||||
id string
|
||||
}
|
||||
type wants struct {
|
||||
statusCode int
|
||||
contentType string
|
||||
body string
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
fields fields
|
||||
args args
|
||||
wants wants
|
||||
}{
|
||||
{
|
||||
name: "remove a cell by id",
|
||||
fields: fields{
|
||||
&mocks.CellService{
|
||||
DeleteCellF: func(ctx context.Context, id chronograf.ID) error {
|
||||
if id == "2" {
|
||||
return nil
|
||||
}
|
||||
|
||||
return fmt.Errorf("wrong id")
|
||||
},
|
||||
},
|
||||
},
|
||||
args: args{
|
||||
id: "2",
|
||||
},
|
||||
wants: wants{
|
||||
statusCode: http.StatusNoContent,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
s := &Service{
|
||||
Store: &mocks.Store{
|
||||
CellService: tt.fields.CellService,
|
||||
},
|
||||
Logger: log.New(log.DebugLevel),
|
||||
}
|
||||
|
||||
r := httptest.NewRequest("GET", "http://any.url", nil)
|
||||
|
||||
r = r.WithContext(httprouter.WithParams(
|
||||
context.Background(),
|
||||
httprouter.Params{
|
||||
{
|
||||
Key: "id",
|
||||
Value: tt.args.id,
|
||||
},
|
||||
}))
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
|
||||
s.RemoveCellV2(w, r)
|
||||
|
||||
res := w.Result()
|
||||
content := res.Header.Get("Content-Type")
|
||||
body, _ := ioutil.ReadAll(res.Body)
|
||||
|
||||
if res.StatusCode != tt.wants.statusCode {
|
||||
t.Errorf("%q. RemoveCellV2() = %v, want %v", tt.name, res.StatusCode, tt.wants.statusCode)
|
||||
}
|
||||
if tt.wants.contentType != "" && content != tt.wants.contentType {
|
||||
t.Errorf("%q. RemoveCellV2() = %v, want %v", tt.name, content, tt.wants.contentType)
|
||||
}
|
||||
if eq, _ := jsonEqual(string(body), tt.wants.body); tt.wants.body != "" && !eq {
|
||||
t.Errorf("%q. RemoveCellV2() = \n***%v***\n,\nwant\n***%v***", tt.name, string(body), tt.wants.body)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestService_UpdateCellV2(t *testing.T) {
|
||||
type fields struct {
|
||||
CellService chronograf.CellService
|
||||
}
|
||||
type args struct {
|
||||
id string
|
||||
upd chronograf.CellUpdate
|
||||
}
|
||||
type wants struct {
|
||||
statusCode int
|
||||
contentType string
|
||||
body string
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
fields fields
|
||||
args args
|
||||
wants wants
|
||||
}{
|
||||
{
|
||||
name: "update a cell",
|
||||
fields: fields{
|
||||
&mocks.CellService{
|
||||
UpdateCellF: func(ctx context.Context, id chronograf.ID, upd chronograf.CellUpdate) (*chronograf.Cell, error) {
|
||||
if id == "2" {
|
||||
return &chronograf.Cell{
|
||||
CellContents: chronograf.CellContents{
|
||||
ID: chronograf.ID("2"),
|
||||
Name: "example",
|
||||
},
|
||||
Visualization: chronograf.V1Visualization{
|
||||
Type: "line",
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
return nil, fmt.Errorf("not found")
|
||||
},
|
||||
},
|
||||
},
|
||||
args: args{
|
||||
id: "2",
|
||||
},
|
||||
wants: wants{
|
||||
statusCode: http.StatusOK,
|
||||
contentType: "application/json",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
s := &Service{
|
||||
Store: &mocks.Store{
|
||||
CellService: tt.fields.CellService,
|
||||
},
|
||||
Logger: log.New(log.DebugLevel),
|
||||
}
|
||||
|
||||
b, err := json.Marshal(tt.args.upd)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to unmarshal cell update: %v", err)
|
||||
}
|
||||
|
||||
r := httptest.NewRequest("GET", "http://any.url", bytes.NewReader(b))
|
||||
|
||||
r = r.WithContext(httprouter.WithParams(
|
||||
context.Background(),
|
||||
httprouter.Params{
|
||||
{
|
||||
Key: "id",
|
||||
Value: tt.args.id,
|
||||
},
|
||||
}))
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
|
||||
s.UpdateCellV2(w, r)
|
||||
|
||||
res := w.Result()
|
||||
content := res.Header.Get("Content-Type")
|
||||
body, _ := ioutil.ReadAll(res.Body)
|
||||
|
||||
if res.StatusCode != tt.wants.statusCode {
|
||||
t.Errorf("%q. RemoveCellV2() = %v, want %v", tt.name, res.StatusCode, tt.wants.statusCode)
|
||||
}
|
||||
if tt.wants.contentType != "" && content != tt.wants.contentType {
|
||||
t.Errorf("%q. RemoveCellV2() = %v, want %v", tt.name, content, tt.wants.contentType)
|
||||
}
|
||||
if eq, _ := jsonEqual(string(body), tt.wants.body); tt.wants.body != "" && !eq {
|
||||
t.Errorf("%q. RemoveCellV2() = \n***%v***\n,\nwant\n***%v***", tt.name, string(body), tt.wants.body)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
20
v2/cell.go
20
v2/cell.go
|
|
@ -119,16 +119,14 @@ func MarshalVisualizationJSON(v Visualization) ([]byte, error) {
|
|||
Type: "chronograf-v1",
|
||||
V1Visualization: vis,
|
||||
}
|
||||
case EmptyVisualization:
|
||||
default:
|
||||
s = struct {
|
||||
Type string `json:"type"`
|
||||
EmptyVisualization
|
||||
}{
|
||||
Type: "empty",
|
||||
EmptyVisualization: vis,
|
||||
EmptyVisualization: EmptyVisualization{},
|
||||
}
|
||||
default:
|
||||
return nil, fmt.Errorf("unsupported type")
|
||||
}
|
||||
return json.Marshal(s)
|
||||
}
|
||||
|
|
@ -173,6 +171,20 @@ func (u *CellUpdate) UnmarshalJSON(b []byte) error {
|
|||
u.Visualization = v
|
||||
return nil
|
||||
}
|
||||
func (u CellUpdate) MarshalJSON() ([]byte, error) {
|
||||
vis, err := MarshalVisualizationJSON(u.Visualization)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return json.Marshal(struct {
|
||||
CellContentsUpdate
|
||||
Visualization json.RawMessage `json:"visualization,omitempty"`
|
||||
}{
|
||||
CellContentsUpdate: u.CellContentsUpdate,
|
||||
Visualization: vis,
|
||||
})
|
||||
}
|
||||
|
||||
type V1Visualization struct {
|
||||
Queries []DashboardQuery `json:"queries"`
|
||||
|
|
|
|||
Loading…
Reference in New Issue