49 lines
998 B
Go
49 lines
998 B
Go
|
package memory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/coreos/poke/storage"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestStaticClients(t *testing.T) {
|
||
|
|
s := New()
|
||
|
|
|
||
|
|
c1 := storage.Client{ID: "foo", Secret: "foo_secret"}
|
||
|
|
c2 := storage.Client{ID: "bar", Secret: "bar_secret"}
|
||
|
|
s.CreateClient(c1)
|
||
|
|
s2 := storage.WithStaticClients(s, []storage.Client{c2})
|
||
|
|
|
||
|
|
tests := []struct {
|
||
|
|
id string
|
||
|
|
s storage.Storage
|
||
|
|
wantErr bool
|
||
|
|
wantClient storage.Client
|
||
|
|
}{
|
||
|
|
{"foo", s, false, c1},
|
||
|
|
{"bar", s, true, storage.Client{}},
|
||
|
|
{"foo", s2, true, storage.Client{}},
|
||
|
|
{"bar", s2, false, c2},
|
||
|
|
}
|
||
|
|
|
||
|
|
for i, tc := range tests {
|
||
|
|
gotClient, err := tc.s.GetClient(tc.id)
|
||
|
|
if err != nil {
|
||
|
|
if !tc.wantErr {
|
||
|
|
t.Errorf("case %d: GetClient(%q) %v", i, tc.id, err)
|
||
|
|
}
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if tc.wantErr {
|
||
|
|
t.Errorf("case %d: GetClient(%q) expected error", i, tc.id)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if !reflect.DeepEqual(tc.wantClient, gotClient) {
|
||
|
|
t.Errorf("case %d: expected=%#v got=%#v", i, tc.wantClient, gotClient)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|