48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/sha256"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestShouldEncryptAndDecriptUsingAES(t *testing.T) {
|
||
|
|
var key [32]byte = sha256.Sum256([]byte("the key"))
|
||
|
|
|
||
|
|
var secret = "the secret"
|
||
|
|
|
||
|
|
encryptedSecret, err := Encrypt([]byte(secret), &key)
|
||
|
|
assert.NoError(t, err, "")
|
||
|
|
|
||
|
|
decryptedSecret, err := Decrypt(encryptedSecret, &key)
|
||
|
|
|
||
|
|
assert.NoError(t, err, "")
|
||
|
|
assert.Equal(t, secret, string(decryptedSecret))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestShouldFailDecryptOnInvalidKey(t *testing.T) {
|
||
|
|
var key [32]byte = sha256.Sum256([]byte("the key"))
|
||
|
|
|
||
|
|
var secret = "the secret"
|
||
|
|
|
||
|
|
encryptedSecret, err := Encrypt([]byte(secret), &key)
|
||
|
|
assert.NoError(t, err, "")
|
||
|
|
|
||
|
|
key = sha256.Sum256([]byte("the key 2"))
|
||
|
|
|
||
|
|
_, err = Decrypt(encryptedSecret, &key)
|
||
|
|
|
||
|
|
assert.Error(t, err, "message authentication failed")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestShouldFailDecryptOnInvalidCypherText(t *testing.T) {
|
||
|
|
var key [32]byte = sha256.Sum256([]byte("the key"))
|
||
|
|
|
||
|
|
encryptedSecret := []byte("abc123")
|
||
|
|
|
||
|
|
_, err := Decrypt(encryptedSecret, &key)
|
||
|
|
|
||
|
|
assert.Error(t, err, "message authentication failed")
|
||
|
|
}
|