test: add some tests

Author: leeqvip

cbc_test.go

@@ -0,0 +1,45 @@
+package openssl
+
+import (
+	"bytes"
+	"crypto/aes"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCBCEncryptAndDecrypt(t *testing.T) {
+	key := []byte("12345678901234567890123456789012") // 32 bytes for AES-256
+	block, err := aes.NewCipher(key)
+	assert.NoError(t, err)
+
+	src := []byte("test data")
+	iv := []byte("1234567890123456") // 16 bytes for AES
+
+	// Test encryption
+	encrypted, err := CBCEncrypt(block, src, iv, PKCS7_PADDING)
+	assert.NoError(t, err)
+	assert.NotEmpty(t, encrypted)
+
+	// Test decryption
+	decrypted, err := CBCDecrypt(block, encrypted, iv, PKCS7_PADDING)
+	assert.NoError(t, err)
+	assert.Equal(t, src, decrypted)
+}
+
+func TestCBCIVPending(t *testing.T) {
+	blockSize := 16
+	testCases := []struct {
+		iv       []byte
+		expected []byte
+	}{
+		{[]byte("1234"), append([]byte("1234"), bytes.Repeat([]byte{0}, 12)...)},
+		{[]byte("12345678901234567890"), []byte("1234567890123456")},
+		{[]byte("1234567890123456"), []byte("1234567890123456")},
+	}
+
+	for _, tc := range testCases {
+		result := cbcIVPending(tc.iv, blockSize)
+		assert.Equal(t, tc.expected, result)
+	}
+}

ecb_test.go

@@ -0,0 +1,51 @@
+package openssl
+
+import (
+	"crypto/aes"
+	"testing"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestECBEncryptAndDecrypt(t *testing.T) {
+	key := []byte("12345678901234567890123456789012") // 32 bytes for AES-256
+	block, err := aes.NewCipher(key)
+	assert.NoError(t, err)
+
+	src := []byte("test data")
+
+	// Test encryption
+	encrypted, err := ECBEncrypt(block, src, PKCS7_PADDING)
+	assert.NoError(t, err)
+	assert.NotEmpty(t, encrypted)
+
+	// Test decryption
+	decrypted, err := ECBDecrypt(block, encrypted, PKCS7_PADDING)
+	assert.NoError(t, err)
+	assert.Equal(t, src, decrypted)
+}
+
+func TestECBEncrypterCryptBlocks(t *testing.T) {
+	key := []byte("1234567890123456") // 16 bytes for AES-128
+	block, err := aes.NewCipher(key)
+	assert.NoError(t, err)
+
+	encrypter := NewECBEncrypter(block)
+	src := make([]byte, encrypter.BlockSize()*2)
+	dst := make([]byte, len(src))
+
+	encrypter.CryptBlocks(dst, src)
+	assert.Equal(t, len(src), len(dst))
+}
+
+func TestECBDecrypterCryptBlocks(t *testing.T) {
+	key := []byte("1234567890123456") // 16 bytes for AES-128
+	block, err := aes.NewCipher(key)
+	assert.NoError(t, err)
+
+	decrypter := NewECBDecrypter(block)
+	src := make([]byte, decrypter.BlockSize()*2)
+	dst := make([]byte, len(src))
+
+	decrypter.CryptBlocks(dst, src)
+	assert.Equal(t, len(src), len(dst))
+}

key_test.go

@@ -0,0 +1,33 @@
+package openssl
+
+import (
+	"crypto/sha1"
+	"testing"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSHA1(t *testing.T) {
+	testData := []byte("test")
+	h := sha1.New()
+	_, _ = h.Write(testData)
+	expected := h.Sum(nil)
+	result := SHA1(testData)
+	assert.Equal(t, expected, result, "SHA1 function output should match expected hash")
+}
+
+func TestKeyGenerator(t *testing.T) {
+	testCases := []struct {
+		data      []byte
+		blockSize int
+		expected  []byte
+	}{
+		{[]byte("test"), 10, KeyGenerator([]byte("test"), 10)},
+		{[]byte("test"), 20, KeyGenerator([]byte("test"), 20)},
+		{[]byte("test"), 30, []byte("test")},
+	}
+
+	for _, tc := range testCases {
+		result := KeyGenerator(tc.data, tc.blockSize)
+		assert.Equal(t, tc.expected, result, "KeyGenerator output should match expected key")
+	}
+}