54 lines
2.0 KiB
Go
54 lines
2.0 KiB
Go
package crypto
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"encoding/hex"
|
|
"github.com/stretchr/testify/require"
|
|
"log"
|
|
"testing"
|
|
)
|
|
|
|
func TestEncryptDecrypt(t *testing.T) {
|
|
message := "this is a message or is it?"
|
|
ciphertext, err := Encrypt(message, []byte("AES256Key-32Characters1234567890"))
|
|
require.Nil(t, err)
|
|
plaintext, err := Decrypt(ciphertext, []byte("AES256Key-32Characters1234567890"))
|
|
require.Nil(t, err)
|
|
log.Println(ciphertext)
|
|
require.Equal(t, message, plaintext)
|
|
}
|
|
|
|
func TestEncryptDecryptJWE(t *testing.T) {
|
|
message := "this is a message or is it?"
|
|
ciphertext, err := EncryptJWE(message, []byte("AES256Key-32Characters1234567890"))
|
|
require.Nil(t, err)
|
|
plaintext, err := DecryptJWE(ciphertext, []byte("AES256Key-32Characters1234567890"))
|
|
require.Nil(t, err)
|
|
log.Println(ciphertext)
|
|
require.Equal(t, message, plaintext)
|
|
}
|
|
|
|
func TestEncryptExpectedOutputxxxxx(t *testing.T) {
|
|
// These values are taken from https://docs.pushbullet.com/#encryption
|
|
// The following expected ciphertext from the site was used as a baseline:
|
|
// MQS2K9l3G8YoLccJooY64kDeWjbkI3fAx4WcrYNtbz4p8Q==
|
|
// 31 04b62bd9771bc6282dc709a2863ae240 de5a36e42377c0c7859cad83 6d6f3e29f1
|
|
// v tag nonce ciphertext
|
|
message := "meow!"
|
|
key, _ := base64.StdEncoding.DecodeString("1sW28zp7CWv5TtGjlQpDHHG4Cbr9v36fG5o4f74LsKg=")
|
|
nonce, _ := hex.DecodeString("de5a36e42377c0c7859cad83")
|
|
ciphertext, err := encryptWithNonce(message, nonce, key)
|
|
require.Nil(t, err)
|
|
require.Equal(t, "MQS2K9l3G8YoLccJooY64kDeWjbkI3fAx4WcrYNtbz4p8Q==", ciphertext)
|
|
}
|
|
|
|
func TestEncryptExpectedOutput(t *testing.T) {
|
|
// These values are taken from https://docs.pushbullet.com/#encryption, meaning that
|
|
// all of this is compatible with how Pushbullet encrypts
|
|
encryptedMessage := "MSfJxxY5YdjttlfUkCaKA57qU9SuCN8+ZhYg/xieI+lDnQ=="
|
|
key, _ := base64.StdEncoding.DecodeString("1sW28zp7CWv5TtGjlQpDHHG4Cbr9v36fG5o4f74LsKg=")
|
|
plaintext, err := Decrypt(encryptedMessage, key)
|
|
require.Nil(t, err)
|
|
require.Equal(t, "meow!", plaintext)
|
|
}
|