mirror of
https://github.com/nbd-wtf/go-nostr.git
synced 2025-03-17 13:22:56 +01:00
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package nip04
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/nbd-wtf/go-nostr"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSharedKeysAreTheSame(t *testing.T) {
|
|
for i := 0; i < 100; i++ {
|
|
sk1 := nostr.GeneratePrivateKey()
|
|
sk2 := nostr.GeneratePrivateKey()
|
|
|
|
pk1, err := nostr.GetPublicKey(sk1)
|
|
require.NoError(t, err)
|
|
pk2, err := nostr.GetPublicKey(sk2)
|
|
require.NoError(t, err)
|
|
|
|
ss1, err := ComputeSharedSecret(pk2, sk1)
|
|
require.NoError(t, err)
|
|
ss2, err := ComputeSharedSecret(pk1, sk2)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, ss1, ss2)
|
|
}
|
|
}
|
|
|
|
func TestEncryptionAndDecryption(t *testing.T) {
|
|
sharedSecret := make([]byte, 32)
|
|
message := "hello hello"
|
|
|
|
ciphertext, err := Encrypt(message, sharedSecret)
|
|
require.NoError(t, err)
|
|
|
|
plaintext, err := Decrypt(ciphertext, sharedSecret)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, plaintext, message, "original '%s' and decrypted '%s' messages differ", message, plaintext)
|
|
}
|
|
|
|
func TestEncryptionAndDecryptionWithMultipleLengths(t *testing.T) {
|
|
sharedSecret := make([]byte, 32)
|
|
|
|
for i := 0; i < 150; i++ {
|
|
message := strings.Repeat("a", i)
|
|
|
|
ciphertext, err := Encrypt(message, sharedSecret)
|
|
require.NoError(t, err)
|
|
|
|
plaintext, err := Decrypt(ciphertext, sharedSecret)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, plaintext, message, "original '%s' and decrypted '%s' messages differ", message, plaintext)
|
|
}
|
|
}
|
|
|
|
func TestNostrToolsCompatibility(t *testing.T) {
|
|
sk1 := "92996316beebf94171065a714cbf164d1f56d7ad9b35b329d9fc97535bf25352"
|
|
sk2 := "591c0c249adfb9346f8d37dfeed65725e2eea1d7a6e99fa503342f367138de84"
|
|
pk2, _ := nostr.GetPublicKey(sk2)
|
|
shared, _ := ComputeSharedSecret(pk2, sk1)
|
|
ciphertext := "A+fRnU4aXS4kbTLfowqAww==?iv=QFYUrl5or/n/qamY79ze0A=="
|
|
plaintext, _ := Decrypt(ciphertext, shared)
|
|
require.Equal(t, "hello", plaintext, "invalid decryption of nostr-tools payload")
|
|
}
|