141 lines
2.9 KiB
Go
141 lines
2.9 KiB
Go
package vault
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/vault/logical"
|
|
)
|
|
|
|
func TestPassthroughBackend_RootPaths(t *testing.T) {
|
|
b := testPassthroughBackend()
|
|
root := b.SpecialPaths()
|
|
if root != nil {
|
|
t.Fatalf("unexpected: %v", root)
|
|
}
|
|
}
|
|
|
|
func TestPassthroughBackend_Write(t *testing.T) {
|
|
b := testPassthroughBackend()
|
|
req := logical.TestRequest(t, logical.WriteOperation, "foo")
|
|
req.Data["raw"] = "test"
|
|
|
|
resp, err := b.HandleRequest(req)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if resp != nil {
|
|
t.Fatalf("bad: %v", resp)
|
|
}
|
|
|
|
out, err := req.Storage.Get("foo")
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if out == nil {
|
|
t.Fatalf("failed to write to view")
|
|
}
|
|
}
|
|
|
|
func TestPassthroughBackend_Read(t *testing.T) {
|
|
b := testPassthroughBackend()
|
|
req := logical.TestRequest(t, logical.WriteOperation, "foo")
|
|
req.Data["raw"] = "test"
|
|
req.Data["lease"] = "1h"
|
|
storage := req.Storage
|
|
|
|
if _, err := b.HandleRequest(req); err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
req = logical.TestRequest(t, logical.ReadOperation, "foo")
|
|
req.Storage = storage
|
|
|
|
resp, err := b.HandleRequest(req)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
expected := &logical.Response{
|
|
Secret: &logical.Secret{
|
|
Renewable: false,
|
|
Lease: time.Hour,
|
|
},
|
|
Data: map[string]interface{}{
|
|
"raw": "test",
|
|
"lease": "1h",
|
|
},
|
|
}
|
|
|
|
resp.Secret.InternalData = nil
|
|
resp.Secret.VaultID = ""
|
|
if !reflect.DeepEqual(resp, expected) {
|
|
t.Fatalf("bad response.\n\nexpected: %#v\n\nGot: %#v", expected, resp)
|
|
}
|
|
}
|
|
|
|
func TestPassthroughBackend_Delete(t *testing.T) {
|
|
b := testPassthroughBackend()
|
|
req := logical.TestRequest(t, logical.WriteOperation, "foo")
|
|
req.Data["raw"] = "test"
|
|
storage := req.Storage
|
|
|
|
if _, err := b.HandleRequest(req); err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
req = logical.TestRequest(t, logical.DeleteOperation, "foo")
|
|
req.Storage = storage
|
|
resp, err := b.HandleRequest(req)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if resp != nil {
|
|
t.Fatalf("bad: %v", resp)
|
|
}
|
|
|
|
req = logical.TestRequest(t, logical.ReadOperation, "foo")
|
|
req.Storage = storage
|
|
resp, err = b.HandleRequest(req)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if resp != nil {
|
|
t.Fatalf("bad: %v", resp)
|
|
}
|
|
}
|
|
|
|
func TestPassthroughBackend_List(t *testing.T) {
|
|
b := testPassthroughBackend()
|
|
req := logical.TestRequest(t, logical.WriteOperation, "foo")
|
|
req.Data["raw"] = "test"
|
|
storage := req.Storage
|
|
|
|
if _, err := b.HandleRequest(req); err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
req = logical.TestRequest(t, logical.ListOperation, "")
|
|
req.Storage = storage
|
|
resp, err := b.HandleRequest(req)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
expected := &logical.Response{
|
|
Data: map[string]interface{}{
|
|
"keys": []string{"foo"},
|
|
},
|
|
}
|
|
|
|
if !reflect.DeepEqual(resp, expected) {
|
|
t.Fatalf("bad response.\n\nexpected: %#v\n\nGot: %#v", expected, resp)
|
|
}
|
|
}
|
|
|
|
func testPassthroughBackend() logical.Backend {
|
|
b, _ := PassthroughBackendFactory(nil)
|
|
return b
|
|
}
|