summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/keybase/go-keybase-chat-bot/kbchat/test_utils.go
diff options
context:
space:
mode:
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.go54
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
+}