diff options
Diffstat (limited to 'vendor/github.com/keybase/go-keybase-chat-bot/kbchat/test_utils.go')
-rw-r--r-- | vendor/github.com/keybase/go-keybase-chat-bot/kbchat/test_utils.go | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/vendor/github.com/keybase/go-keybase-chat-bot/kbchat/test_utils.go b/vendor/github.com/keybase/go-keybase-chat-bot/kbchat/test_utils.go new file mode 100644 index 00000000..1a163951 --- /dev/null +++ b/vendor/github.com/keybase/go-keybase-chat-bot/kbchat/test_utils.go @@ -0,0 +1,54 @@ +package kbchat + +import ( + "crypto/rand" + "encoding/hex" + "io/ioutil" + "os" + "os/exec" + "path" + "strings" + "testing" + + "github.com/stretchr/testify/require" +) + +func randomString(t *testing.T) string { + bytes := make([]byte, 16) + _, err := rand.Read(bytes) + require.NoError(t, err) + return hex.EncodeToString(bytes) +} + +func randomTempDir(t *testing.T) string { + return path.Join(os.TempDir(), "keybase_bot_"+randomString(t)) +} + +func whichKeybase(t *testing.T) string { + cmd := exec.Command("which", "keybase") + out, err := cmd.Output() + require.NoError(t, err) + location := strings.TrimSpace(string(out)) + return location +} + +func copyFile(t *testing.T, source, dest string) { + sourceData, err := ioutil.ReadFile(source) + require.NoError(t, err) + err = ioutil.WriteFile(dest, sourceData, 0777) + require.NoError(t, err) +} + +// Creates the working directory and copies over the keybase binary in PATH. +// We do this to avoid any version mismatch issues. +func prepWorkingDir(t *testing.T, workingDir string) string { + kbLocation := whichKeybase(t) + + err := os.Mkdir(workingDir, 0777) + require.NoError(t, err) + kbDestination := path.Join(workingDir, "keybase") + + copyFile(t, kbLocation, kbDestination) + + return kbDestination +} |