124 lines
2.3 KiB
Go
124 lines
2.3 KiB
Go
package framework
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hashicorp/vault/logical"
|
|
)
|
|
|
|
func TestPathMap(t *testing.T) {
|
|
p := &PathMap{Name: "foo"}
|
|
storage := new(logical.InmemStorage)
|
|
var b logical.Backend = &Backend{Paths: p.Paths()}
|
|
|
|
// Write via HTTP
|
|
_, err := b.HandleRequest(&logical.Request{
|
|
Operation: logical.WriteOperation,
|
|
Path: "map/foo/a",
|
|
Data: map[string]interface{}{
|
|
"value": "bar",
|
|
},
|
|
Storage: storage,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
|
|
// Read via HTTP
|
|
resp, err := b.HandleRequest(&logical.Request{
|
|
Operation: logical.ReadOperation,
|
|
Path: "map/foo/a",
|
|
Storage: storage,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
if resp.Data["value"] != "bar" {
|
|
t.Fatalf("bad: %#v", resp)
|
|
}
|
|
|
|
// Read via API
|
|
v, err := p.Get(storage, "a")
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
if v["value"] != "bar" {
|
|
t.Fatalf("bad: %#v", v)
|
|
}
|
|
|
|
// Read via API with other casing
|
|
v, err = p.Get(storage, "A")
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
if v["value"] != "bar" {
|
|
t.Fatalf("bad: %#v", v)
|
|
}
|
|
|
|
// Verify List
|
|
keys, err := p.List(storage, "")
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
if len(keys) != 1 || keys[0] != "a" {
|
|
t.Fatalf("bad: %#v", keys)
|
|
}
|
|
|
|
// Delete via HTTP
|
|
resp, err = b.HandleRequest(&logical.Request{
|
|
Operation: logical.DeleteOperation,
|
|
Path: "map/foo/a",
|
|
Storage: storage,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
if resp != nil {
|
|
t.Fatalf("bad: %#v", resp)
|
|
}
|
|
|
|
// Re-read via HTTP
|
|
resp, err = b.HandleRequest(&logical.Request{
|
|
Operation: logical.ReadOperation,
|
|
Path: "map/foo/a",
|
|
Storage: storage,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
if _, ok := resp.Data["value"]; ok {
|
|
t.Fatalf("bad: %#v", resp)
|
|
}
|
|
|
|
// Re-read via API
|
|
v, err = p.Get(storage, "a")
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
if v != nil {
|
|
t.Fatalf("bad: %#v", v)
|
|
}
|
|
}
|
|
|
|
func TestPathMap_getInvalid(t *testing.T) {
|
|
p := &PathMap{Name: "foo"}
|
|
storage := new(logical.InmemStorage)
|
|
|
|
v, err := p.Get(storage, "nope")
|
|
if err != nil {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
if v != nil {
|
|
t.Fatalf("bad: %#v", v)
|
|
}
|
|
}
|
|
|
|
func TestPathMap_routes(t *testing.T) {
|
|
p := &PathMap{Name: "foo"}
|
|
TestBackendRoutes(t, &Backend{Paths: p.Paths()}, []string{
|
|
"map/foo", // Normal
|
|
"map/foo/bar", // Normal
|
|
"map/foo/bar-baz", // Hyphen key
|
|
})
|
|
}
|