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
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package spanner
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"cloud.google.com/go/spanner"
|
|
log "github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/vault/helper/logging"
|
|
"github.com/hashicorp/vault/physical"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func testCleanup(t testing.TB, client *spanner.Client, table string) {
|
|
t.Helper()
|
|
|
|
// Delete all data in the table
|
|
ctx := context.Background()
|
|
m := spanner.Delete(table, spanner.AllKeys())
|
|
if _, err := client.Apply(ctx, []*spanner.Mutation{m}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestBackend(t *testing.T) {
|
|
database := os.Getenv("GOOGLE_SPANNER_DATABASE")
|
|
if database == "" {
|
|
t.Skip("GOOGLE_SPANNER_DATABASE not set")
|
|
}
|
|
|
|
table := os.Getenv("GOOGLE_SPANNER_TABLE")
|
|
if table == "" {
|
|
t.Skip("GOOGLE_SPANNER_TABLE not set")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
client, err := spanner.NewClient(ctx, database)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
testCleanup(t, client, table)
|
|
defer testCleanup(t, client, table)
|
|
|
|
backend, err := NewBackend(map[string]string{
|
|
"database": database,
|
|
"table": table,
|
|
"ha_enabled": "false",
|
|
}, logging.NewVaultLogger(log.Debug))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
physical.ExerciseBackend(t, backend)
|
|
physical.ExerciseBackend_ListPrefix(t, backend)
|
|
physical.ExerciseTransactionalBackend(t, backend)
|
|
}
|