130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
package memdb
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/influxdata/chronograf"
|
|
)
|
|
|
|
func TestInterfaceImplementation(t *testing.T) {
|
|
var _ chronograf.ServersStore = &KapacitorStore{}
|
|
var _ chronograf.ServersStore = &MultiKapacitorStore{}
|
|
}
|
|
|
|
func TestKapacitorStoreAll(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
store := KapacitorStore{}
|
|
kaps, err := store.All(ctx)
|
|
if err != nil {
|
|
t.Fatal("All should not throw an error with an empty Store")
|
|
}
|
|
if len(kaps) != 0 {
|
|
t.Fatal("Store should be empty")
|
|
}
|
|
|
|
store.Kapacitor = &chronograf.Server{}
|
|
kaps, err = store.All(ctx)
|
|
if err != nil {
|
|
t.Fatal("All should not throw an error with an empty Store")
|
|
}
|
|
if len(kaps) != 1 {
|
|
t.Fatal("Store should have 1 element")
|
|
}
|
|
}
|
|
|
|
func TestKapacitorStoreAdd(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
store := KapacitorStore{}
|
|
_, err := store.Add(ctx, chronograf.Server{})
|
|
if err == nil {
|
|
t.Fatal("Store should not support adding another source")
|
|
}
|
|
}
|
|
|
|
func TestKapacitorStoreDelete(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
store := KapacitorStore{}
|
|
err := store.Delete(ctx, chronograf.Server{})
|
|
if err == nil {
|
|
t.Fatal("Delete should not operate on an empty Store")
|
|
}
|
|
|
|
store.Kapacitor = &chronograf.Server{
|
|
ID: 9,
|
|
}
|
|
err = store.Delete(ctx, chronograf.Server{
|
|
ID: 8,
|
|
})
|
|
if err == nil {
|
|
t.Fatal("Delete should not remove elements with the wrong ID")
|
|
}
|
|
|
|
err = store.Delete(ctx, chronograf.Server{
|
|
ID: 9,
|
|
})
|
|
if err != nil {
|
|
t.Fatal("Delete should remove an element with a matching ID")
|
|
}
|
|
}
|
|
|
|
func TestKapacitorStoreGet(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
store := KapacitorStore{}
|
|
_, err := store.Get(ctx, 9)
|
|
if err == nil {
|
|
t.Fatal("Get should return an error for an empty Store")
|
|
}
|
|
|
|
store.Kapacitor = &chronograf.Server{
|
|
ID: 9,
|
|
}
|
|
_, err = store.Get(ctx, 8)
|
|
if err == nil {
|
|
t.Fatal("Get should return an error if it finds no matches")
|
|
}
|
|
|
|
store.Kapacitor = &chronograf.Server{
|
|
ID: 9,
|
|
}
|
|
kap, err := store.Get(ctx, 9)
|
|
if err != nil || kap.ID != 9 {
|
|
t.Fatal("Get should find the element with a matching ID")
|
|
}
|
|
}
|
|
|
|
func TestKapacitorStoreUpdate(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
store := KapacitorStore{}
|
|
err := store.Update(ctx, chronograf.Server{})
|
|
if err == nil {
|
|
t.Fatal("Update fhouls return an error for an empty Store")
|
|
}
|
|
|
|
store.Kapacitor = &chronograf.Server{
|
|
ID: 9,
|
|
}
|
|
err = store.Update(ctx, chronograf.Server{
|
|
ID: 8,
|
|
})
|
|
if err == nil {
|
|
t.Fatal("Update should return an error if it finds no matches")
|
|
}
|
|
|
|
store.Kapacitor = &chronograf.Server{
|
|
ID: 9,
|
|
}
|
|
err = store.Update(ctx, chronograf.Server{
|
|
ID: 9,
|
|
URL: "http://crystal.pepsi.com",
|
|
})
|
|
if err != nil || store.Kapacitor.URL != "http://crystal.pepsi.com" {
|
|
t.Fatal("Update should overwrite elements with matching IDs")
|
|
}
|
|
}
|