69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package miniredis
|
|
|
|
// 'Fail' methods.
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
"reflect"
|
|
"runtime"
|
|
"sort"
|
|
)
|
|
|
|
// T is implemented by Testing.T
|
|
type T interface {
|
|
Fail()
|
|
}
|
|
|
|
// CheckGet does not call Errorf() iff there is a string key with the
|
|
// expected value. Normal use case is `m.CheckGet(t, "username", "theking")`.
|
|
func (m *Miniredis) CheckGet(t T, key, expected string) {
|
|
found, err := m.Get(key)
|
|
if err != nil {
|
|
lError(t, "GET error, key %#v: %v", key, err)
|
|
return
|
|
}
|
|
if found != expected {
|
|
lError(t, "GET error, key %#v: Expected %#v, got %#v", key, expected, found)
|
|
return
|
|
}
|
|
}
|
|
|
|
// CheckList does not call Errorf() iff there is a list key with the
|
|
// expected values.
|
|
// Normal use case is `m.CheckGet(t, "favorite_colors", "red", "green", "infrared")`.
|
|
func (m *Miniredis) CheckList(t T, key string, expected ...string) {
|
|
found, err := m.List(key)
|
|
if err != nil {
|
|
lError(t, "List error, key %#v: %v", key, err)
|
|
return
|
|
}
|
|
if !reflect.DeepEqual(expected, found) {
|
|
lError(t, "List error, key %#v: Expected %#v, got %#v", key, expected, found)
|
|
return
|
|
}
|
|
}
|
|
|
|
// CheckSet does not call Errorf() iff there is a set key with the
|
|
// expected values.
|
|
// Normal use case is `m.CheckSet(t, "visited", "Rome", "Stockholm", "Dublin")`.
|
|
func (m *Miniredis) CheckSet(t T, key string, expected ...string) {
|
|
found, err := m.Members(key)
|
|
if err != nil {
|
|
lError(t, "Set error, key %#v: %v", key, err)
|
|
return
|
|
}
|
|
sort.Strings(expected)
|
|
if !reflect.DeepEqual(expected, found) {
|
|
lError(t, "Set error, key %#v: Expected %#v, got %#v", key, expected, found)
|
|
return
|
|
}
|
|
}
|
|
|
|
func lError(t T, format string, args ...interface{}) {
|
|
_, file, line, _ := runtime.Caller(2)
|
|
prefix := fmt.Sprintf("%s:%d: ", filepath.Base(file), line)
|
|
fmt.Printf(prefix+format+"\n", args...)
|
|
t.Fail()
|
|
}
|