mirror of
https://github.com/Mic92/sops-nix.git
synced 2025-03-06 08:37:21 +00:00
55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"path"
|
||
|
|
"path/filepath"
|
||
|
|
"runtime"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ok fails the test if an err is not nil.
|
||
|
|
func ok(tb testing.TB, err error) {
|
||
|
|
if err != nil {
|
||
|
|
_, file, line, _ := runtime.Caller(1)
|
||
|
|
fmt.Printf("\033[31m%s:%d: unexpected error: %s\033[39m\n\n", filepath.Base(file), line, err.Error())
|
||
|
|
tb.FailNow()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestShellHook(t *testing.T) {
|
||
|
|
_, filename, _, _ := runtime.Caller(0)
|
||
|
|
assets := path.Join(path.Dir(filename), "test-assets")
|
||
|
|
tempdir, err := ioutil.TempDir("", "testdir")
|
||
|
|
ok(t, err)
|
||
|
|
defer os.RemoveAll(tempdir)
|
||
|
|
|
||
|
|
cmd := exec.Command("nix-shell", "shell.nix", "--run", "echo SOPS_PGP_FP=$SOPS_PGP_FP")
|
||
|
|
cmd.Env = append(os.Environ(), fmt.Sprintf("GNUPGHOME=%s", tempdir))
|
||
|
|
var stdoutBuf, stderrBuf bytes.Buffer
|
||
|
|
cmd.Stdout = &stdoutBuf
|
||
|
|
cmd.Stderr = &stderrBuf
|
||
|
|
cmd.Dir = assets
|
||
|
|
err = cmd.Run()
|
||
|
|
stdout := string(stdoutBuf.Bytes())
|
||
|
|
stderr := string(stderrBuf.Bytes())
|
||
|
|
fmt.Printf("$ %s\nstdout: \n%s\nstderr: \n%s\n", strings.Join(cmd.Args, " "), stdout, stderr)
|
||
|
|
ok(t, err)
|
||
|
|
|
||
|
|
expectedStdout := "SOPS_PGP_FP=C6DA56E69A7C756564A8AFEB4A6B05B714D13EFD,4EC40F8E04A945339F7F7C0032C5225271038E3F,7FB89715AADA920D65D25E63F9BA9DEBD03F57C0"
|
||
|
|
if strings.Index(stdout, expectedStdout) == -1 {
|
||
|
|
t.Fatalf("'%v' not in '%v'", expectedStdout, stdout)
|
||
|
|
}
|
||
|
|
|
||
|
|
expectedStderr := "./non-existing-key.gpg does not exists"
|
||
|
|
if strings.Index(stderr, expectedStderr) == -1 {
|
||
|
|
t.Fatalf("'%v' not in '%v'", expectedStderr, stdout)
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|