From 4d9a42aa20c2fa91e38bd09ecd53e14b99530d93 Mon Sep 17 00:00:00 2001
From: Seth Vargo <seth@sethvargo.com>
Date: Mon, 4 Sep 2017 23:50:59 -0400
Subject: [PATCH] Add an in-mem token helper for testing

---
 command/token/helper_testing.go | 42 +++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 command/token/helper_testing.go

diff --git a/command/token/helper_testing.go b/command/token/helper_testing.go
new file mode 100644
index 0000000000..93465931b7
--- /dev/null
+++ b/command/token/helper_testing.go
@@ -0,0 +1,42 @@
+package token
+
+import (
+	"sync"
+)
+
+var _ 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
+}