1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
package filedriver
import (
"io/ioutil"
"os"
"testing"
"github.com/stretchr/testify/require"
)
func setup() (*Driver, error) {
tmppath, err := ioutil.TempDir("", "secretsdata")
if err != nil {
return nil, err
}
return NewDriver(tmppath)
}
func TestStoreAndLookupSecretData(t *testing.T) {
tstdriver, err := setup()
require.NoError(t, err)
defer os.Remove(tstdriver.secretsDataFilePath)
err = tstdriver.Store("unique_id", []byte("somedata"))
require.NoError(t, err)
secretData, err := tstdriver.Lookup("unique_id")
require.NoError(t, err)
require.Equal(t, secretData, []byte("somedata"))
}
func TestStoreDupID(t *testing.T) {
tstdriver, err := setup()
require.NoError(t, err)
defer os.Remove(tstdriver.secretsDataFilePath)
err = tstdriver.Store("unique_id", []byte("somedata"))
require.NoError(t, err)
err = tstdriver.Store("unique_id", []byte("somedata"))
require.Error(t, err)
}
func TestLookupBogus(t *testing.T) {
tstdriver, err := setup()
require.NoError(t, err)
defer os.Remove(tstdriver.secretsDataFilePath)
_, err = tstdriver.Lookup("bogus")
require.Error(t, err)
}
func TestDeleteSecretData(t *testing.T) {
tstdriver, err := setup()
require.NoError(t, err)
defer os.Remove(tstdriver.secretsDataFilePath)
err = tstdriver.Store("unique_id", []byte("somedata"))
require.NoError(t, err)
err = tstdriver.Delete("unique_id")
require.NoError(t, err)
data, err := tstdriver.Lookup("unique_id")
require.Error(t, err)
require.Nil(t, data)
}
func TestDeleteSecretDataNotExist(t *testing.T) {
tstdriver, err := setup()
require.NoError(t, err)
defer os.Remove(tstdriver.secretsDataFilePath)
err = tstdriver.Delete("bogus")
require.Error(t, err)
}
func TestList(t *testing.T) {
tstdriver, err := setup()
require.NoError(t, err)
defer os.Remove(tstdriver.secretsDataFilePath)
err = tstdriver.Store("unique_id", []byte("somedata"))
require.NoError(t, err)
err = tstdriver.Store("unique_id2", []byte("moredata"))
require.NoError(t, err)
data, err := tstdriver.List()
require.NoError(t, err)
require.Len(t, data, 2)
}
|
