aboutsummaryrefslogtreecommitdiff
path: root/ask/ask_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'ask/ask_test.go')
-rw-r--r--ask/ask_test.go28
1 files changed, 28 insertions, 0 deletions
diff --git a/ask/ask_test.go b/ask/ask_test.go
new file mode 100644
index 0000000..71bc663
--- /dev/null
+++ b/ask/ask_test.go
@@ -0,0 +1,28 @@
+package ask
+
+import (
+ "bytes"
+ "testing"
+)
+
+func TestAsk(t *testing.T) {
+ testCases := []struct {
+ input, pass string
+ err error
+ }{
+ {"test", "test", nil},
+ {"", "", ErrNoPassword},
+ }
+ for _, tc := range testCases {
+ t.Run(tc.input, func(t *testing.T) {
+ buf := bytes.NewBufferString(tc.input + "\r")
+ pass, err := ask(buf, "")
+ if err != tc.err {
+ t.Fatalf("got %v, want %v", err, tc.err)
+ }
+ if pass != tc.pass {
+ t.Error("got %v, want %v", pass, tc.pass)
+ }
+ })
+ }
+}