mirror of
https://github.com/hashicorp/vault.git
synced 2026-01-06 09:11:10 +01:00
58e59e67b2
* license: update headers to IBM Corp. * `make proto` * update offset because source file changed * update licenses in files that were not included in backport Signed-off-by: Ryan Cragun <me@ryan.ec> Co-authored-by: Ryan Cragun <me@ryan.ec>
48 lines
911 B
Go
48 lines
911 B
Go
// Copyright IBM Corp. 2016, 2025
|
|
// 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
|
|
}
|