forked from cyberark/summon-aws-secrets
-
Notifications
You must be signed in to change notification settings - Fork 0
/
package_test.go
62 lines (50 loc) · 1.52 KB
/
package_test.go
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
package main
import (
"bytes"
"os"
"os/exec"
"testing"
. "github.com/smartystreets/goconvey/convey"
)
func RunCommand(name string, arg ...string) (bytes.Buffer, bytes.Buffer, error) {
cmd := exec.Command(name, arg...)
var stdout bytes.Buffer
var stderr bytes.Buffer
cmd.Stdout = &stdout
cmd.Stderr = &stderr
err := cmd.Run()
return stdout, stderr, err
}
const PackageName = "summon-aws-secrets"
func TestPackage(t *testing.T) {
Path := os.Getenv("PATH")
Convey("Given a compiled summon-aws-secrets package", t, func() {
Convey("Given no configuration information", func() {
e := ClearEnv()
defer e.RestoreEnv()
os.Setenv("PATH", Path)
Convey("Given summon-aws-secrets is run with no arguments", func() {
_, stderr, err := RunCommand(PackageName)
Convey("Returns with error", func() {
So(err, ShouldNotBeNil)
So(stderr.String(), ShouldStartWith, "A variable ID or version flag must be given as the first and only argument!")
})
})
})
})
}
func Test_getValueByKey(t *testing.T) {
Convey("Given a valid JSON format stored secret", t, func() {
secret := `{ "username": "USERNAME", "password": "PASSWORD", "port": 8000 }`
Convey("Returns the value of the key", func() {
stdout, err := getValueByKey("username", []byte(secret))
So(err, ShouldBeNil)
So(string(stdout), ShouldEqual, "USERNAME")
})
Convey("Always returns as a string", func() {
stdout, err := getValueByKey("port", []byte(secret))
So(err, ShouldBeNil)
So(string(stdout), ShouldEqual, "8000")
})
})
}