Merge pull request #613 from jhaals/moretests

Add test cases for errors
jhaals · Sep 29, 2020 · fe52ed4 · fe52ed4
2 parents 00f1056 + cc137d2
commit fe52ed4
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/pkg/yopass/client_test.go b/pkg/yopass/client_test.go
@@ -37,6 +37,12 @@ func TestFetch(t *testing.T) {
 	}
 }
 
+func TestFetchInvalidServer(t *testing.T) {
+	_, err := yopass.Fetch("127.0.0.1:9999/invalid", "1337")
+	if err == nil {
+		t.Error("expected error, got none")
+	}
+}
 func TestStore(t *testing.T) {
 	db := testDB(map[string]string{})
 	y := server.New(&db, 1024, prometheus.NewRegistry())

diff --git a/pkg/yopass/yopass_test.go b/pkg/yopass/yopass_test.go
@@ -3,6 +3,7 @@ package yopass_test
 import (
 	"bytes"
 	"errors"
+	"fmt"
 	"io/ioutil"
 	"regexp"
 	"strings"
@@ -147,6 +148,21 @@ func TestEncrypt(t *testing.T) {
 	}
 }
 
+type invalidFile struct{}
+
+func (invalidFile) Read(p []byte) (n int, err error) { return 0, fmt.Errorf("Broken I/O") }
+
+func TestEncryptWithInvalidFile(t *testing.T) {
+	_, err := yopass.Encrypt(invalidFile{}, "somekey")
+	if err == nil {
+		t.Fatal("expected error, got none")
+	}
+	want := "could not copy data: Broken I/O"
+	if err.Error() != want {
+		t.Fatalf("expected %s, got %v", want, err)
+	}
+}
+
 func TestGenerateKey(t *testing.T) {
 	format := regexp.MustCompile("^[a-zA-Z0-9-_]{22}$")