358 lines
8.5 KiB
Go
358 lines
8.5 KiB
Go
package testing
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
platform "github.com/influxdata/influxdb/v2"
|
|
platform2 "github.com/influxdata/influxdb/v2/kit/platform"
|
|
"github.com/influxdata/influxdb/v2/kit/platform/errors"
|
|
"github.com/influxdata/influxdb/v2/mock"
|
|
)
|
|
|
|
const (
|
|
defaultSourceID = "020f755c3c082000"
|
|
defaultSourceOrganizationID = "50616e67652c206c"
|
|
sourceOneID = "020f755c3c082001"
|
|
sourceTwoID = "020f755c3c082002"
|
|
sourceOrgOneID = "61726920617a696f"
|
|
)
|
|
|
|
var sourceCmpOptions = cmp.Options{
|
|
cmp.Comparer(func(x, y []byte) bool {
|
|
return bytes.Equal(x, y)
|
|
}),
|
|
cmp.Transformer("Sort", func(in []*platform.Source) []*platform.Source {
|
|
out := append([]*platform.Source(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
|
|
}),
|
|
}
|
|
|
|
// SourceFields will include the IDGenerator, and sources
|
|
type SourceFields struct {
|
|
IDGenerator platform2.IDGenerator
|
|
Sources []*platform.Source
|
|
}
|
|
|
|
// CreateSource testing
|
|
func CreateSource(
|
|
init func(SourceFields, *testing.T) (platform.SourceService, string, func()),
|
|
t *testing.T,
|
|
) {
|
|
type args struct {
|
|
source *platform.Source
|
|
}
|
|
type wants struct {
|
|
err error
|
|
sources []*platform.Source
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
fields SourceFields
|
|
args args
|
|
wants wants
|
|
}{
|
|
{
|
|
name: "create sources with empty set",
|
|
fields: SourceFields{
|
|
IDGenerator: mock.NewIDGenerator(sourceOneID, t),
|
|
Sources: []*platform.Source{},
|
|
},
|
|
args: args{
|
|
source: &platform.Source{
|
|
Name: "name1",
|
|
},
|
|
},
|
|
wants: wants{
|
|
sources: []*platform.Source{
|
|
{
|
|
Name: "autogen",
|
|
Type: "self",
|
|
ID: MustIDBase16(defaultSourceID),
|
|
OrganizationID: MustIDBase16(defaultSourceOrganizationID),
|
|
Default: true,
|
|
},
|
|
{
|
|
Name: "name1",
|
|
ID: MustIDBase16(sourceOneID),
|
|
OrganizationID: MustIDBase16(sourceOneID),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s, opPrefix, done := init(tt.fields, t)
|
|
defer done()
|
|
ctx := context.Background()
|
|
err := s.CreateSource(ctx, tt.args.source)
|
|
diffPlatformErrors(tt.name, err, tt.wants.err, opPrefix, t)
|
|
defer s.DeleteSource(ctx, tt.args.source.ID)
|
|
|
|
sources, _, err := s.FindSources(ctx, platform.FindOptions{})
|
|
if err != nil {
|
|
t.Fatalf("failed to retrieve sources: %v", err)
|
|
}
|
|
if diff := cmp.Diff(sources, tt.wants.sources, sourceCmpOptions...); diff != "" {
|
|
t.Errorf("sources are different -got/+want\ndiff %s", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// FindSourceByID testing
|
|
func FindSourceByID(
|
|
init func(SourceFields, *testing.T) (platform.SourceService, string, func()),
|
|
t *testing.T,
|
|
) {
|
|
type args struct {
|
|
id platform2.ID
|
|
}
|
|
type wants struct {
|
|
err error
|
|
source *platform.Source
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
fields SourceFields
|
|
args args
|
|
wants wants
|
|
}{
|
|
{
|
|
name: "find default source by ID",
|
|
fields: SourceFields{
|
|
IDGenerator: mock.NewIDGenerator(sourceOneID, t),
|
|
Sources: []*platform.Source{
|
|
{
|
|
Name: "name1",
|
|
ID: MustIDBase16(sourceOneID),
|
|
OrganizationID: MustIDBase16(sourceOrgOneID),
|
|
},
|
|
},
|
|
},
|
|
args: args{
|
|
id: MustIDBase16(sourceOneID),
|
|
},
|
|
wants: wants{
|
|
source: &platform.Source{
|
|
Name: "name1",
|
|
ID: MustIDBase16(sourceOneID),
|
|
OrganizationID: MustIDBase16(sourceOrgOneID),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "find source by ID",
|
|
fields: SourceFields{
|
|
IDGenerator: mock.NewIDGenerator(sourceOneID, t),
|
|
Sources: []*platform.Source{},
|
|
},
|
|
args: args{
|
|
id: MustIDBase16(defaultSourceID),
|
|
},
|
|
wants: wants{
|
|
source: &platform.Source{
|
|
Name: "autogen",
|
|
Type: "self",
|
|
ID: MustIDBase16(defaultSourceID),
|
|
OrganizationID: MustIDBase16(defaultSourceOrganizationID),
|
|
Default: true,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s, opPrefix, done := init(tt.fields, t)
|
|
defer done()
|
|
ctx := context.Background()
|
|
source, err := s.FindSourceByID(ctx, tt.args.id)
|
|
diffPlatformErrors(tt.name, err, tt.wants.err, opPrefix, t)
|
|
|
|
if diff := cmp.Diff(source, tt.wants.source, sourceCmpOptions...); diff != "" {
|
|
t.Errorf("sources are different -got/+want\ndiff %s", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// FindSources testing
|
|
func FindSources(
|
|
init func(SourceFields, *testing.T) (platform.SourceService, string, func()),
|
|
t *testing.T,
|
|
) {
|
|
type args struct {
|
|
opts platform.FindOptions
|
|
}
|
|
type wants struct {
|
|
err error
|
|
sources []*platform.Source
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
fields SourceFields
|
|
args args
|
|
wants wants
|
|
}{
|
|
{
|
|
name: "find all sources",
|
|
fields: SourceFields{
|
|
IDGenerator: mock.NewIDGenerator(sourceOneID, t),
|
|
Sources: []*platform.Source{
|
|
{
|
|
Name: "name1",
|
|
ID: MustIDBase16(sourceOneID),
|
|
OrganizationID: MustIDBase16(sourceOrgOneID),
|
|
},
|
|
{
|
|
Name: "name2",
|
|
ID: MustIDBase16(sourceTwoID),
|
|
OrganizationID: MustIDBase16(sourceOrgOneID),
|
|
},
|
|
},
|
|
},
|
|
args: args{},
|
|
wants: wants{
|
|
sources: []*platform.Source{
|
|
{
|
|
Name: "autogen",
|
|
Type: "self",
|
|
ID: MustIDBase16(defaultSourceID),
|
|
OrganizationID: MustIDBase16(defaultSourceOrganizationID),
|
|
Default: true,
|
|
},
|
|
{
|
|
Name: "name1",
|
|
ID: MustIDBase16(sourceOneID),
|
|
OrganizationID: MustIDBase16(sourceOrgOneID),
|
|
},
|
|
{
|
|
Name: "name2",
|
|
ID: MustIDBase16(sourceTwoID),
|
|
OrganizationID: MustIDBase16(sourceOrgOneID),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s, opPrefix, done := init(tt.fields, t)
|
|
defer done()
|
|
ctx := context.Background()
|
|
sources, _, err := s.FindSources(ctx, tt.args.opts)
|
|
diffPlatformErrors(tt.name, err, tt.wants.err, opPrefix, t)
|
|
|
|
if diff := cmp.Diff(sources, tt.wants.sources, sourceCmpOptions...); diff != "" {
|
|
t.Errorf("sources are different -got/+want\ndiff %s", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// DeleteSource testing
|
|
func DeleteSource(
|
|
init func(SourceFields, *testing.T) (platform.SourceService, string, func()),
|
|
t *testing.T,
|
|
) {
|
|
type args struct {
|
|
id platform2.ID
|
|
}
|
|
type wants struct {
|
|
err error
|
|
sources []*platform.Source
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
fields SourceFields
|
|
args args
|
|
wants wants
|
|
}{
|
|
{
|
|
name: "delete source by ID",
|
|
fields: SourceFields{
|
|
IDGenerator: mock.NewIDGenerator(sourceOneID, t),
|
|
Sources: []*platform.Source{
|
|
{
|
|
Name: "name1",
|
|
ID: MustIDBase16(sourceOneID),
|
|
OrganizationID: MustIDBase16(sourceOrgOneID),
|
|
},
|
|
},
|
|
},
|
|
args: args{
|
|
id: MustIDBase16(sourceOneID),
|
|
},
|
|
wants: wants{
|
|
sources: []*platform.Source{
|
|
{
|
|
Name: "autogen",
|
|
Type: "self",
|
|
ID: MustIDBase16(defaultSourceID),
|
|
OrganizationID: MustIDBase16(defaultSourceOrganizationID),
|
|
Default: true,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "delete default source by ID",
|
|
fields: SourceFields{
|
|
IDGenerator: mock.NewIDGenerator(sourceOneID, t),
|
|
Sources: []*platform.Source{},
|
|
},
|
|
args: args{
|
|
id: MustIDBase16(defaultSourceID),
|
|
},
|
|
wants: wants{
|
|
err: &errors.Error{
|
|
Code: errors.EForbidden,
|
|
Op: platform.OpDeleteSource,
|
|
Msg: "cannot delete autogen source",
|
|
},
|
|
sources: []*platform.Source{
|
|
{
|
|
Name: "autogen",
|
|
Type: "self",
|
|
ID: MustIDBase16(defaultSourceID),
|
|
OrganizationID: MustIDBase16(defaultSourceOrganizationID),
|
|
Default: true,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s, opPrefix, done := init(tt.fields, t)
|
|
defer done()
|
|
ctx := context.Background()
|
|
err := s.DeleteSource(ctx, tt.args.id)
|
|
diffPlatformErrors(tt.name, err, tt.wants.err, opPrefix, t)
|
|
|
|
sources, _, err := s.FindSources(ctx, platform.FindOptions{})
|
|
if err != nil {
|
|
t.Fatalf("failed to retrieve sources: %v", err)
|
|
}
|
|
if diff := cmp.Diff(sources, tt.wants.sources, sourceCmpOptions...); diff != "" {
|
|
t.Errorf("sources are different -got/+want\ndiff %s", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|