42 lines
839 B
Go
42 lines
839 B
Go
|
package tunnel
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEncrypt(t *testing.T) {
|
||
|
|
key := "tokenpassphrase"
|
||
|
|
data := []byte("supersecret")
|
||
|
|
|
||
|
|
cipherText, err := Encrypt(data, key)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("failed to encrypt data: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// verify the cipherText is not the same as data
|
||
|
|
if bytes.Equal(data, cipherText) {
|
||
|
|
t.Error("encrypted data are the same as plaintext")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDecrypt(t *testing.T) {
|
||
|
|
key := "tokenpassphrase"
|
||
|
|
data := []byte("supersecret")
|
||
|
|
|
||
|
|
cipherText, err := Encrypt(data, key)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("failed to encrypt data: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
plainText, err := Decrypt(cipherText, key)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("failed to decrypt data: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// verify the plainText is the same as data
|
||
|
|
if !bytes.Equal(data, plainText) {
|
||
|
|
t.Error("decrypted data not the same as plaintext")
|
||
|
|
}
|
||
|
|
}
|