From cf5796a4f337a663e8e9e50209a84e758a2b9b70 Mon Sep 17 00:00:00 2001
From: Vitaliy Vasylenko <me@vitalvas.com>
Date: Sun, 31 Mar 2024 16:26:23 -0400
Subject: [PATCH] feat: add tests

---
 internal/version/version_test.go | 17 +++++++++++++++++
 pkg/encryptor/key_test.go        | 16 ++++++++++++++++
 pkg/encryptor/value_test.go      | 16 ++++++++++++++++
 3 files changed, 49 insertions(+)
 create mode 100644 internal/version/version_test.go

diff --git a/internal/version/version_test.go b/internal/version/version_test.go
new file mode 100644
index 0000000..7ef8f43
--- /dev/null
+++ b/internal/version/version_test.go
@@ -0,0 +1,17 @@
+package version
+
+import (
+	"runtime/debug"
+	"testing"
+)
+
+func TestVersion(t *testing.T) {
+	bi, ok := debug.ReadBuildInfo()
+	if !ok {
+		t.Fatal("ReadBuildInfo failed")
+	}
+
+	if got, want := Version(), bi.Main.Version; got != want {
+		t.Errorf("Version() = %q; want %q", got, want)
+	}
+}
diff --git a/pkg/encryptor/key_test.go b/pkg/encryptor/key_test.go
index 6946c94..5153b06 100644
--- a/pkg/encryptor/key_test.go
+++ b/pkg/encryptor/key_test.go
@@ -11,6 +11,14 @@ func TestEncryptKey(t *testing.T) {
 		t.Fatalf("unexpected error: %v", err)
 	}
 
+	t.Run("invalid keyAead", func(t *testing.T) {
+		enc := &Encryptor{}
+
+		if _, err := enc.EncryptKey("test"); err == nil {
+			t.Fatalf("expected error, got nil")
+		}
+	})
+
 	for _, tc := range []struct {
 		name  string
 		text  string
@@ -45,6 +53,14 @@ func TestDecryptKey(t *testing.T) {
 		t.Fatalf("unexpected error: %v", err)
 	}
 
+	t.Run("invalid keyAead", func(t *testing.T) {
+		enc := &Encryptor{}
+
+		if _, err := enc.DecryptKey(nil); err == nil {
+			t.Fatalf("expected error, got nil")
+		}
+	})
+
 	for _, tc := range []struct {
 		name  string
 		text  string
diff --git a/pkg/encryptor/value_test.go b/pkg/encryptor/value_test.go
index c73fd01..1d2deaf 100644
--- a/pkg/encryptor/value_test.go
+++ b/pkg/encryptor/value_test.go
@@ -12,6 +12,14 @@ func TestEncryptValue(t *testing.T) {
 		t.Fatalf("unexpected error: %v", err)
 	}
 
+	t.Run("invalid keyAead", func(t *testing.T) {
+		enc := &Encryptor{}
+
+		if _, err := enc.EncryptValue("test", "test"); err == nil {
+			t.Fatalf("expected error, got nil")
+		}
+	})
+
 	for _, tc := range []struct {
 		name  string
 		key   string
@@ -48,6 +56,14 @@ func TestDecryptValue(t *testing.T) {
 		t.Fatalf("unexpected error: %v", err)
 	}
 
+	t.Run("invalid keyAead", func(t *testing.T) {
+		enc := &Encryptor{}
+
+		if _, err := enc.DecryptValue("test", nil); err == nil {
+			t.Fatalf("expected error, got nil")
+		}
+	})
+
 	for _, tc := range []struct {
 		name  string
 		key   string