mirror of
https://github.com/hashicorp/vault.git
synced 2025-08-19 13:41:10 +02:00
a54603039d
The result will still pass gofmtcheck and won't trigger additional changes if someone isn't using goimports, but it will avoid the piecemeal imports changes we've been seeing.
86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
package logical
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
"time"
|
|
|
|
testing "github.com/mitchellh/go-testing-interface"
|
|
|
|
log "github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/vault/helper/logging"
|
|
)
|
|
|
|
// TestRequest is a helper to create a purely in-memory Request struct.
|
|
func TestRequest(t testing.T, op Operation, path string) *Request {
|
|
return &Request{
|
|
Operation: op,
|
|
Path: path,
|
|
Data: make(map[string]interface{}),
|
|
Storage: new(InmemStorage),
|
|
}
|
|
}
|
|
|
|
// TestStorage is a helper that can be used from unit tests to verify
|
|
// the behavior of a Storage impl.
|
|
func TestStorage(t testing.T, s Storage) {
|
|
keys, err := s.List(context.Background(), "")
|
|
if err != nil {
|
|
t.Fatalf("list error: %s", err)
|
|
}
|
|
if len(keys) > 0 {
|
|
t.Fatalf("should have no keys to start: %#v", keys)
|
|
}
|
|
|
|
entry := &StorageEntry{Key: "foo", Value: []byte("bar")}
|
|
if err := s.Put(context.Background(), entry); err != nil {
|
|
t.Fatalf("put error: %s", err)
|
|
}
|
|
|
|
actual, err := s.Get(context.Background(), "foo")
|
|
if err != nil {
|
|
t.Fatalf("get error: %s", err)
|
|
}
|
|
if !reflect.DeepEqual(actual, entry) {
|
|
t.Fatalf("wrong value. Expected: %#v\nGot: %#v", entry, actual)
|
|
}
|
|
|
|
keys, err = s.List(context.Background(), "")
|
|
if err != nil {
|
|
t.Fatalf("list error: %s", err)
|
|
}
|
|
if !reflect.DeepEqual(keys, []string{"foo"}) {
|
|
t.Fatalf("bad keys: %#v", keys)
|
|
}
|
|
|
|
if err := s.Delete(context.Background(), "foo"); err != nil {
|
|
t.Fatalf("put error: %s", err)
|
|
}
|
|
|
|
keys, err = s.List(context.Background(), "")
|
|
if err != nil {
|
|
t.Fatalf("list error: %s", err)
|
|
}
|
|
if len(keys) > 0 {
|
|
t.Fatalf("should have no keys to start: %#v", keys)
|
|
}
|
|
}
|
|
|
|
func TestSystemView() *StaticSystemView {
|
|
defaultLeaseTTLVal := time.Hour * 24
|
|
maxLeaseTTLVal := time.Hour * 24 * 2
|
|
return &StaticSystemView{
|
|
DefaultLeaseTTLVal: defaultLeaseTTLVal,
|
|
MaxLeaseTTLVal: maxLeaseTTLVal,
|
|
}
|
|
}
|
|
|
|
func TestBackendConfig() *BackendConfig {
|
|
bc := &BackendConfig{
|
|
Logger: logging.NewVaultLogger(log.Trace),
|
|
System: TestSystemView(),
|
|
}
|
|
|
|
return bc
|
|
}
|