vault/command/token/helper_testing.go
Tom Proctor 9ed00822ea
Move CLI token helper to api module (#25744)
* Move command/config + command/token to api/cliconfig + api/tokenhelper
* Remove unused functions and unused import
* Simplify and inline function copied from SDK
* Delete unused duplicated/forwarding config implementation from command package
* Delete unused code, unexport API surface that's only used internally to the package
* Fix up license headers
* Add changelog
* Tweak .gitignore to track hcl files in testdata/ folders
2024-03-04 18:29:20 +00:00

48 lines
910 B
Go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1
package token
import (
"sync"
"github.com/hashicorp/vault/api/tokenhelper"
)
var _ tokenhelper.TokenHelper = (*TestingTokenHelper)(nil)
// TestingTokenHelper implements token.TokenHelper which runs entirely
// in-memory. This should not be used outside of testing.
type TestingTokenHelper struct {
lock sync.RWMutex
token string
}
func NewTestingTokenHelper() *TestingTokenHelper {
return &TestingTokenHelper{}
}
func (t *TestingTokenHelper) Erase() error {
t.lock.Lock()
defer t.lock.Unlock()
t.token = ""
return nil
}
func (t *TestingTokenHelper) Get() (string, error) {
t.lock.RLock()
defer t.lock.RUnlock()
return t.token, nil
}
func (t *TestingTokenHelper) Path() string {
return ""
}
func (t *TestingTokenHelper) Store(token string) error {
t.lock.Lock()
defer t.lock.Unlock()
t.token = token
return nil
}