test: add tests for DecryptBlocks and EncryptBlocks

gotd · Jun 6, 2021 · 05915c8 · 05915c8
1 parent ee9db41
commit 05915c8
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/decrypt_test.go b/decrypt_test.go
@@ -53,6 +53,23 @@ func TestDecrypterCryptBlocks(t *testing.T) {
 	}
 }
 
+func TestDecryptBlocks(t *testing.T) {
+	for a, v := range TestVectors {
+		out := make([]byte, len(v.Ciphertext))
+
+		c, err := aes.NewCipher(v.Key)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		DecryptBlocks(c, v.IV, out, v.Ciphertext)
+
+		if !bytes.Equal(out, v.Plaintext) {
+			t.Fatalf("test vector %d has wrong ciphertext\n", a+1)
+		}
+	}
+}
+
 func TestDecryptCryptBlocksPanicSrc(t *testing.T) {
 	c, err := aes.NewCipher(make([]byte, 16))
 	if err != nil {

diff --git a/encrypt_test.go b/encrypt_test.go
@@ -53,6 +53,23 @@ func TestEncrypterCryptBlocks(t *testing.T) {
 	}
 }
 
+func TestEncryptBlocks(t *testing.T) {
+	for a, v := range TestVectors {
+		out := make([]byte, len(v.Ciphertext))
+
+		c, err := aes.NewCipher(v.Key)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		EncryptBlocks(c, v.IV, out, v.Plaintext)
+
+		if !bytes.Equal(out, v.Ciphertext) {
+			t.Fatalf("test vector %d has wrong ciphertext\n", a+1)
+		}
+	}
+}
+
 func TestEncryptCryptBlocksPanicSrc(t *testing.T) {
 	c, err := aes.NewCipher(make([]byte, 16))
 	if err != nil {