357 lines
7.2 KiB
Go
357 lines
7.2 KiB
Go
|
package canned_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
"reflect"
|
||
|
|
"strconv"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/influxdata/mrfusion"
|
||
|
|
"github.com/influxdata/mrfusion/canned"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAll(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
var tests = []struct {
|
||
|
|
Existing []mrfusion.Layout
|
||
|
|
Err error
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{
|
||
|
|
{ID: 1,
|
||
|
|
Application: "howdy",
|
||
|
|
},
|
||
|
|
{ID: 2,
|
||
|
|
Application: "doody",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Err: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{},
|
||
|
|
Err: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: nil,
|
||
|
|
Err: errors.New("Error"),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for i, test := range tests {
|
||
|
|
apps, _ := MockApps(test.Existing, test.Err)
|
||
|
|
layouts, err := apps.All(context.Background())
|
||
|
|
if err != test.Err {
|
||
|
|
t.Errorf("Test %d: Canned all error expected: %v; actual: %v", i, test.Err, err)
|
||
|
|
}
|
||
|
|
if !reflect.DeepEqual(layouts, test.Existing) {
|
||
|
|
t.Errorf("Test %d: Layouts should be equal; expected %v; actual %v", i, test.Existing, layouts)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAdd(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
var tests = []struct {
|
||
|
|
Existing []mrfusion.Layout
|
||
|
|
Add mrfusion.Layout
|
||
|
|
ExpectedID int
|
||
|
|
Err error
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{
|
||
|
|
{ID: 1,
|
||
|
|
Application: "howdy",
|
||
|
|
},
|
||
|
|
{ID: 2,
|
||
|
|
Application: "doody",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Add: mrfusion.Layout{
|
||
|
|
Application: "newbie",
|
||
|
|
},
|
||
|
|
ExpectedID: 3,
|
||
|
|
Err: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{},
|
||
|
|
Add: mrfusion.Layout{
|
||
|
|
Application: "newbie",
|
||
|
|
},
|
||
|
|
ExpectedID: 0,
|
||
|
|
Err: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: nil,
|
||
|
|
Add: mrfusion.Layout{
|
||
|
|
Application: "newbie",
|
||
|
|
},
|
||
|
|
ExpectedID: 0,
|
||
|
|
Err: errors.New("Error"),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for i, test := range tests {
|
||
|
|
apps, _ := MockApps(test.Existing, test.Err)
|
||
|
|
layout, err := apps.Add(context.Background(), test.Add)
|
||
|
|
if err != test.Err {
|
||
|
|
t.Errorf("Test %d: Canned add error expected: %v; actual: %v", i, test.Err, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if layout.ID != test.ExpectedID {
|
||
|
|
t.Errorf("Test %d: Layout ID should be equal; expected %d; actual %d", i, test.ExpectedID, layout.ID)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDelete(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
var tests = []struct {
|
||
|
|
Existing []mrfusion.Layout
|
||
|
|
DeleteID int
|
||
|
|
Expected map[string]mrfusion.Layout
|
||
|
|
Err error
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{
|
||
|
|
{ID: 1,
|
||
|
|
Application: "howdy",
|
||
|
|
},
|
||
|
|
{ID: 2,
|
||
|
|
Application: "doody",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
DeleteID: 1,
|
||
|
|
Expected: map[string]mrfusion.Layout{
|
||
|
|
"dir/2.json": {ID: 2,
|
||
|
|
Application: "doody",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Err: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{},
|
||
|
|
DeleteID: 1,
|
||
|
|
Expected: map[string]mrfusion.Layout{},
|
||
|
|
Err: mrfusion.ErrLayoutNotFound,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: nil,
|
||
|
|
DeleteID: 1,
|
||
|
|
Expected: map[string]mrfusion.Layout{},
|
||
|
|
Err: errors.New("Error"),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for i, test := range tests {
|
||
|
|
apps, actual := MockApps(test.Existing, test.Err)
|
||
|
|
err := apps.Delete(context.Background(), mrfusion.Layout{ID: test.DeleteID})
|
||
|
|
if err != test.Err {
|
||
|
|
t.Errorf("Test %d: Canned delete error expected: %v; actual: %v", i, test.Err, err)
|
||
|
|
}
|
||
|
|
if !reflect.DeepEqual(*actual, test.Expected) {
|
||
|
|
t.Errorf("Test %d: Layouts should be equal; expected %v; actual %v", i, test.Expected, actual)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGet(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
var tests = []struct {
|
||
|
|
Existing []mrfusion.Layout
|
||
|
|
ID int
|
||
|
|
Expected mrfusion.Layout
|
||
|
|
Err error
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{
|
||
|
|
{ID: 1,
|
||
|
|
Application: "howdy",
|
||
|
|
},
|
||
|
|
{ID: 2,
|
||
|
|
Application: "doody",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
ID: 1,
|
||
|
|
Expected: mrfusion.Layout{
|
||
|
|
ID: 1,
|
||
|
|
Application: "howdy",
|
||
|
|
},
|
||
|
|
Err: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{},
|
||
|
|
ID: 1,
|
||
|
|
Expected: mrfusion.Layout{},
|
||
|
|
Err: mrfusion.ErrLayoutNotFound,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: nil,
|
||
|
|
ID: 1,
|
||
|
|
Expected: mrfusion.Layout{},
|
||
|
|
Err: mrfusion.ErrLayoutNotFound,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for i, test := range tests {
|
||
|
|
apps, _ := MockApps(test.Existing, test.Err)
|
||
|
|
layout, err := apps.Get(context.Background(), test.ID)
|
||
|
|
if err != test.Err {
|
||
|
|
t.Errorf("Test %d: Canned get error expected: %v; actual: %v", i, test.Err, err)
|
||
|
|
}
|
||
|
|
if !reflect.DeepEqual(layout, test.Expected) {
|
||
|
|
t.Errorf("Test %d: Layouts should be equal; expected %v; actual %v", i, test.Expected, layout)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUpdate(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
var tests = []struct {
|
||
|
|
Existing []mrfusion.Layout
|
||
|
|
Update mrfusion.Layout
|
||
|
|
Expected map[string]mrfusion.Layout
|
||
|
|
Err error
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{
|
||
|
|
{ID: 1,
|
||
|
|
Application: "howdy",
|
||
|
|
},
|
||
|
|
{ID: 2,
|
||
|
|
Application: "doody",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Update: mrfusion.Layout{
|
||
|
|
ID: 1,
|
||
|
|
Application: "hello",
|
||
|
|
Measurement: "measurement",
|
||
|
|
},
|
||
|
|
Expected: map[string]mrfusion.Layout{
|
||
|
|
"dir/1.json": {ID: 1,
|
||
|
|
Application: "hello",
|
||
|
|
Measurement: "measurement",
|
||
|
|
},
|
||
|
|
"dir/2.json": {ID: 2,
|
||
|
|
Application: "doody",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Err: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: []mrfusion.Layout{},
|
||
|
|
Update: mrfusion.Layout{
|
||
|
|
ID: 1,
|
||
|
|
},
|
||
|
|
Expected: map[string]mrfusion.Layout{},
|
||
|
|
Err: mrfusion.ErrLayoutNotFound,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Existing: nil,
|
||
|
|
Update: mrfusion.Layout{
|
||
|
|
ID: 1,
|
||
|
|
},
|
||
|
|
Expected: map[string]mrfusion.Layout{},
|
||
|
|
Err: mrfusion.ErrLayoutNotFound,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for i, test := range tests {
|
||
|
|
apps, actual := MockApps(test.Existing, test.Err)
|
||
|
|
err := apps.Update(context.Background(), test.Update)
|
||
|
|
if err != test.Err {
|
||
|
|
t.Errorf("Test %d: Canned get error expected: %v; actual: %v", i, test.Err, err)
|
||
|
|
}
|
||
|
|
if !reflect.DeepEqual(*actual, test.Expected) {
|
||
|
|
t.Errorf("Test %d: Layouts should be equal; expected %v; actual %v", i, test.Expected, actual)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type MockFileInfo struct {
|
||
|
|
name string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *MockFileInfo) Name() string {
|
||
|
|
return m.name
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *MockFileInfo) Size() int64 {
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *MockFileInfo) Mode() os.FileMode {
|
||
|
|
return 0666
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *MockFileInfo) ModTime() time.Time {
|
||
|
|
return time.Now()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *MockFileInfo) IsDir() bool {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *MockFileInfo) Sys() interface{} {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func MockApps(existing []mrfusion.Layout, expected error) (canned.Apps, *map[string]mrfusion.Layout) {
|
||
|
|
layouts := map[string]mrfusion.Layout{}
|
||
|
|
fileName := func(dir string, layout mrfusion.Layout) string {
|
||
|
|
return path.Join(dir, strconv.Itoa(layout.ID)+".json")
|
||
|
|
}
|
||
|
|
dir := "dir"
|
||
|
|
for _, l := range existing {
|
||
|
|
layouts[fileName(dir, l)] = l
|
||
|
|
}
|
||
|
|
load := func(file string) (mrfusion.Layout, error) {
|
||
|
|
if expected != nil {
|
||
|
|
return mrfusion.Layout{}, expected
|
||
|
|
}
|
||
|
|
|
||
|
|
if l, ok := layouts[file]; !ok {
|
||
|
|
return mrfusion.Layout{}, mrfusion.ErrLayoutNotFound
|
||
|
|
} else {
|
||
|
|
return l, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
create := func(file string, layout mrfusion.Layout) error {
|
||
|
|
if expected != nil {
|
||
|
|
return expected
|
||
|
|
}
|
||
|
|
layouts[file] = layout
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
readDir := func(dirname string) ([]os.FileInfo, error) {
|
||
|
|
if expected != nil {
|
||
|
|
return nil, expected
|
||
|
|
}
|
||
|
|
info := []os.FileInfo{}
|
||
|
|
for k, _ := range layouts {
|
||
|
|
info = append(info, &MockFileInfo{k})
|
||
|
|
}
|
||
|
|
return info, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
remove := func(name string) error {
|
||
|
|
if expected != nil {
|
||
|
|
return expected
|
||
|
|
}
|
||
|
|
if _, ok := layouts[name]; !ok {
|
||
|
|
return mrfusion.ErrLayoutNotFound
|
||
|
|
}
|
||
|
|
delete(layouts, name)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return canned.Apps{
|
||
|
|
Dir: dir,
|
||
|
|
Load: load,
|
||
|
|
Filename: fileName,
|
||
|
|
Create: create,
|
||
|
|
ReadDir: readDir,
|
||
|
|
Remove: remove,
|
||
|
|
}, &layouts
|
||
|
|
}
|