Influxdata
/
influxdb
mirror of https://github.com/influxdata/influxdb.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
influxdb/testing/cells.go

634 lines
14 KiB
Go
Raw Blame History

package testing
import (
"bytes"
"context"
"fmt"
"sort"
"testing"
"github.com/google/go-cmp/cmp"
"github.com/influxdata/platform"
"github.com/influxdata/platform/mock"
)
const (
viewOneID = "020f755c3c082000"
viewTwoID = "020f755c3c082001"
viewThreeID = "020f755c3c082002"
)
var viewCmpOptions = cmp.Options{
cmp.Comparer(func(x, y []byte) bool {
return bytes.Equal(x, y)
}),
cmp.Transformer("Sort", func(in []*platform.View) []*platform.View {
out := append([]*platform.View(nil), in...) // Copy input to avoid mutating it
sort.Slice(out, func(i, j int) bool {
return out[i].ID.String() > out[j].ID.String()
})
return out
}),
}
// ViewFields will include the IDGenerator, and views
type ViewFields struct {
IDGenerator platform.IDGenerator
Views []*platform.View
}
// CreateView testing
func CreateView(
init func(ViewFields, *testing.T) (platform.ViewService, func()),
t *testing.T,
) {
type args struct {
view *platform.View
}
type wants struct {
err error
views []*platform.View
}
tests := []struct {
name string
fields ViewFields
args args
wants wants
}{
{
name: "basic create view",
fields: ViewFields{
IDGenerator: &mock.IDGenerator{
IDFn: func() platform.ID {
return idFromString(t, viewTwoID)
},
},
Views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
},
},
},
args: args{
view: &platform.View{
ViewContents: platform.ViewContents{
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
wants: wants{
views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
s, done := init(tt.fields, t)
defer done()
ctx := context.TODO()
err := s.CreateView(ctx, tt.args.view)
if (err != nil) != (tt.wants.err != nil) {
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
}
if err != nil && tt.wants.err != nil {
if err.Error() != tt.wants.err.Error() {
t.Fatalf("expected error messages to match '%v' got '%v'", tt.wants.err, err.Error())
}
}
defer s.DeleteView(ctx, tt.args.view.ID)
views, _, err := s.FindViews(ctx, platform.ViewFilter{})
if err != nil {
t.Fatalf("failed to retrieve views: %v", err)
}
if diff := cmp.Diff(views, tt.wants.views, viewCmpOptions...); diff != "" {
t.Errorf("views are different -got/+want\ndiff %s", diff)
}
})
}
}
// FindViewByID testing
func FindViewByID(
init func(ViewFields, *testing.T) (platform.ViewService, func()),
t *testing.T,
) {
type args struct {
id platform.ID
}
type wants struct {
err error
view *platform.View
}
tests := []struct {
name string
fields ViewFields
args args
wants wants
}{
{
name: "basic find view by id",
fields: ViewFields{
Views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
args: args{
id: idFromString(t, viewTwoID),
},
wants: wants{
view: &platform.View{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
s, done := init(tt.fields, t)
defer done()
ctx := context.TODO()
view, err := s.FindViewByID(ctx, tt.args.id)
if (err != nil) != (tt.wants.err != nil) {
t.Fatalf("expected errors to be equal '%v' got '%v'", tt.wants.err, err)
}
if err != nil && tt.wants.err != nil {
if err.Error() != tt.wants.err.Error() {
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
}
}
if diff := cmp.Diff(view, tt.wants.view, viewCmpOptions...); diff != "" {
t.Errorf("view is different -got/+want\ndiff %s", diff)
}
})
}
}
// FindViews testing
func FindViews(
init func(ViewFields, *testing.T) (platform.ViewService, func()),
t *testing.T,
) {
type args struct {
ID platform.ID
name string
}
type wants struct {
views []*platform.View
err error
}
tests := []struct {
name string
fields ViewFields
args args
wants wants
}{
{
name: "find all views",
fields: ViewFields{
Views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
args: args{},
wants: wants{
views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
},
{
name: "find view by id",
fields: ViewFields{
Views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
args: args{
ID: idFromString(t, viewTwoID),
},
wants: wants{
views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
s, done := init(tt.fields, t)
defer done()
ctx := context.TODO()
filter := platform.ViewFilter{}
if tt.args.ID != nil {
filter.ID = &tt.args.ID
}
views, _, err := s.FindViews(ctx, filter)
if (err != nil) != (tt.wants.err != nil) {
t.Fatalf("expected errors to be equal '%v' got '%v'", tt.wants.err, err)
}
if err != nil && tt.wants.err != nil {
if err.Error() != tt.wants.err.Error() {
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
}
}
if diff := cmp.Diff(views, tt.wants.views, viewCmpOptions...); diff != "" {
t.Errorf("views are different -got/+want\ndiff %s", diff)
}
})
}
}
// DeleteView testing
func DeleteView(
init func(ViewFields, *testing.T) (platform.ViewService, func()),
t *testing.T,
) {
type args struct {
ID platform.ID
}
type wants struct {
err error
views []*platform.View
}
tests := []struct {
name string
fields ViewFields
args args
wants wants
}{
{
name: "delete views using exist id",
fields: ViewFields{
Views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
args: args{
ID: idFromString(t, viewOneID),
},
wants: wants{
views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
},
{
name: "delete views using id that does not exist",
fields: ViewFields{
Views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
args: args{
ID: idFromString(t, viewThreeID),
},
wants: wants{
err: fmt.Errorf("View not found"),
views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
s, done := init(tt.fields, t)
defer done()
ctx := context.TODO()
err := s.DeleteView(ctx, tt.args.ID)
if (err != nil) != (tt.wants.err != nil) {
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
}
if err != nil && tt.wants.err != nil {
if err.Error() != tt.wants.err.Error() {
t.Fatalf("expected error messages to match '%v' got '%v'", tt.wants.err, err.Error())
}
}
filter := platform.ViewFilter{}
views, _, err := s.FindViews(ctx, filter)
if err != nil {
t.Fatalf("failed to retrieve views: %v", err)
}
if diff := cmp.Diff(views, tt.wants.views, viewCmpOptions...); diff != "" {
t.Errorf("views are different -got/+want\ndiff %s", diff)
}
})
}
}
// UpdateView testing
func UpdateView(
init func(ViewFields, *testing.T) (platform.ViewService, func()),
t *testing.T,
) {
type args struct {
name string
properties platform.ViewProperties
id platform.ID
}
type wants struct {
err error
view *platform.View
}
tests := []struct {
name string
fields ViewFields
args args
wants wants
}{
{
name: "update name",
fields: ViewFields{
Views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
args: args{
id: idFromString(t, viewOneID),
name: "changed",
},
wants: wants{
view: &platform.View{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "changed",
},
Properties: platform.EmptyViewProperties{},
},
},
},
{
name: "update properties",
fields: ViewFields{
Views: []*platform.View{
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.EmptyViewProperties{},
},
{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewTwoID),
Name: "view2",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
args: args{
id: idFromString(t, viewOneID),
properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
wants: wants{
view: &platform.View{
ViewContents: platform.ViewContents{
ID: idFromString(t, viewOneID),
Name: "view1",
},
Properties: platform.TableViewProperties{
Type: "table",
TimeFormat: "rfc3339",
},
},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
s, done := init(tt.fields, t)
defer done()
ctx := context.TODO()
upd := platform.ViewUpdate{}
if tt.args.name != "" {
upd.Name = &tt.args.name
}
if tt.args.properties != nil {
upd.Properties = tt.args.properties
}
view, err := s.UpdateView(ctx, tt.args.id, upd)
if (err != nil) != (tt.wants.err != nil) {
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
}
if err != nil && tt.wants.err != nil {
if err.Error() != tt.wants.err.Error() {
t.Fatalf("expected error messages to match '%v' got '%v'", tt.wants.err, err.Error())
}
}
if diff := cmp.Diff(view, tt.wants.view, viewCmpOptions...); diff != "" {
t.Errorf("view is different -got/+want\ndiff %s", diff)
}
})
}
}
Reference in New Issue View Git Blame Copy Permalink