d0e2badbae
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.
121 lines
2.4 KiB
Go
121 lines
2.4 KiB
Go
package vault
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
log "github.com/hashicorp/go-hclog"
|
|
uuid "github.com/hashicorp/go-uuid"
|
|
|
|
"github.com/hashicorp/vault/helper/logging"
|
|
"github.com/hashicorp/vault/helper/namespace"
|
|
)
|
|
|
|
// mockRollback returns a mock rollback manager
|
|
func mockRollback(t *testing.T) (*RollbackManager, *NoopBackend) {
|
|
backend := new(NoopBackend)
|
|
mounts := new(MountTable)
|
|
router := NewRouter()
|
|
core, _, _ := TestCoreUnsealed(t)
|
|
|
|
_, barrier, _ := mockBarrier(t)
|
|
view := NewBarrierView(barrier, "logical/")
|
|
|
|
mounts.Entries = []*MountEntry{
|
|
&MountEntry{
|
|
Path: "foo",
|
|
NamespaceID: namespace.RootNamespaceID,
|
|
namespace: namespace.RootNamespace,
|
|
},
|
|
}
|
|
meUUID, err := uuid.GenerateUUID()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := router.Mount(backend, "foo", &MountEntry{UUID: meUUID, Accessor: "noopaccessor", NamespaceID: namespace.RootNamespaceID, namespace: namespace.RootNamespace}, view); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
mountsFunc := func() []*MountEntry {
|
|
return mounts.Entries
|
|
}
|
|
|
|
logger := logging.NewVaultLogger(log.Trace)
|
|
|
|
rb := NewRollbackManager(context.Background(), logger, mountsFunc, router, core)
|
|
rb.period = 10 * time.Millisecond
|
|
return rb, backend
|
|
}
|
|
|
|
func TestRollbackManager(t *testing.T) {
|
|
m, backend := mockRollback(t)
|
|
if len(backend.Paths) > 0 {
|
|
t.Fatalf("bad: %#v", backend)
|
|
}
|
|
|
|
m.Start()
|
|
time.Sleep(50 * time.Millisecond)
|
|
m.Stop()
|
|
|
|
count := len(backend.Paths)
|
|
if count == 0 {
|
|
t.Fatalf("bad: %#v", backend)
|
|
}
|
|
if backend.Paths[0] != "" {
|
|
t.Fatalf("bad: %#v", backend)
|
|
}
|
|
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
if count != len(backend.Paths) {
|
|
t.Fatalf("should stop requests: %#v", backend)
|
|
}
|
|
}
|
|
|
|
func TestRollbackManager_Join(t *testing.T) {
|
|
m, backend := mockRollback(t)
|
|
if len(backend.Paths) > 0 {
|
|
t.Fatalf("bad: %#v", backend)
|
|
}
|
|
|
|
m.Start()
|
|
defer m.Stop()
|
|
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(3)
|
|
|
|
errCh := make(chan error, 3)
|
|
go func() {
|
|
defer wg.Done()
|
|
err := m.Rollback(namespace.RootContext(nil), "foo")
|
|
if err != nil {
|
|
errCh <- err
|
|
}
|
|
}()
|
|
|
|
go func() {
|
|
defer wg.Done()
|
|
err := m.Rollback(namespace.RootContext(nil), "foo")
|
|
if err != nil {
|
|
errCh <- err
|
|
}
|
|
}()
|
|
|
|
go func() {
|
|
defer wg.Done()
|
|
err := m.Rollback(namespace.RootContext(nil), "foo")
|
|
if err != nil {
|
|
errCh <- err
|
|
}
|
|
}()
|
|
wg.Wait()
|
|
close(errCh)
|
|
err := <-errCh
|
|
if err != nil {
|
|
t.Fatalf("Error on rollback:%v", err)
|
|
}
|
|
}
|