03cf302e9a
* logbridge with hclog and identical output * Initial search & replace This compiles, but there is a fair amount of TODO and commented out code, especially around the plugin logclient/logserver code. * strip logbridge * fix majority of tests * update logxi aliases * WIP fixing tests * more test fixes * Update test to hclog * Fix format * Rename hclog -> log * WIP making hclog and logxi love each other * update logger_test.go * clean up merged comments * Replace RawLogger interface with a Logger * Add some logger names * Replace Trace with Debug * update builtin logical logging patterns * Fix build errors * More log updates * update log approach in command and builtin * More log updates * update helper, http, and logical directories * Update loggers * Log updates * Update logging * Update logging * Update logging * Update logging * update logging in physical * prefixing and lowercase * Update logging * Move phyisical logging name to server command * Fix som tests * address jims feedback so far * incorporate brians feedback so far * strip comments * move vault.go to logging package * update Debug to Trace * Update go-plugin deps * Update logging based on review comments * Updates from review * Unvendor logxi * Remove null_logger.go
103 lines
2.1 KiB
Go
103 lines
2.1 KiB
Go
package zookeeper
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
log "github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/vault/helper/logging"
|
|
"github.com/hashicorp/vault/physical"
|
|
|
|
"github.com/samuel/go-zookeeper/zk"
|
|
)
|
|
|
|
func TestZooKeeperBackend(t *testing.T) {
|
|
addr := os.Getenv("ZOOKEEPER_ADDR")
|
|
if addr == "" {
|
|
t.SkipNow()
|
|
}
|
|
|
|
client, _, err := zk.Connect([]string{addr}, time.Second)
|
|
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
randPath := fmt.Sprintf("/vault-%d", time.Now().Unix())
|
|
acl := zk.WorldACL(zk.PermAll)
|
|
_, err = client.Create(randPath, []byte("hi"), int32(0), acl)
|
|
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
defer func() {
|
|
client.Delete(randPath+"/foo/nested1/nested2/nested3", -1)
|
|
client.Delete(randPath+"/foo/nested1/nested2", -1)
|
|
client.Delete(randPath+"/foo/nested1", -1)
|
|
client.Delete(randPath+"/foo/bar/baz", -1)
|
|
client.Delete(randPath+"/foo/bar", -1)
|
|
client.Delete(randPath+"/foo", -1)
|
|
client.Delete(randPath, -1)
|
|
client.Close()
|
|
}()
|
|
|
|
logger := logging.NewVaultLogger(log.Debug)
|
|
|
|
b, err := NewZooKeeperBackend(map[string]string{
|
|
"address": addr + "," + addr,
|
|
"path": randPath,
|
|
}, logger)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
physical.ExerciseBackend(t, b)
|
|
physical.ExerciseBackend_ListPrefix(t, b)
|
|
}
|
|
|
|
func TestZooKeeperHABackend(t *testing.T) {
|
|
addr := os.Getenv("ZOOKEEPER_ADDR")
|
|
if addr == "" {
|
|
t.SkipNow()
|
|
}
|
|
|
|
client, _, err := zk.Connect([]string{addr}, time.Second)
|
|
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
randPath := fmt.Sprintf("/vault-ha-%d", time.Now().Unix())
|
|
acl := zk.WorldACL(zk.PermAll)
|
|
_, err = client.Create(randPath, []byte("hi"), int32(0), acl)
|
|
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
defer func() {
|
|
client.Delete(randPath+"/foo", -1)
|
|
client.Delete(randPath, -1)
|
|
client.Close()
|
|
}()
|
|
|
|
logger := logging.NewVaultLogger(log.Debug)
|
|
|
|
b, err := NewZooKeeperBackend(map[string]string{
|
|
"address": addr + "," + addr,
|
|
"path": randPath,
|
|
}, logger)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
ha, ok := b.(physical.HABackend)
|
|
if !ok {
|
|
t.Fatalf("zookeeper does not implement HABackend")
|
|
}
|
|
physical.ExerciseHABackend(t, ha, ha)
|
|
}
|