go-nostr/nip04/nip04_test.go

package nip04

import (
	"strings"
	"testing"

	"github.com/nbd-wtf/go-nostr"
)

func TestEncryptionAndDecryption(t *testing.T) {
	sharedSecret := make([]byte, 32)
	message := "hello hello"

	ciphertext, err := Encrypt(message, sharedSecret)
	if err != nil {
		t.Errorf("failed to encrypt: %s", err.Error())
	}

	plaintext, err := Decrypt(ciphertext, sharedSecret)
	if err != nil {
		t.Errorf("failed to decrypt: %s", err.Error())
	}

	if message != plaintext {
		t.Errorf("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)
		if err != nil {
			t.Errorf("failed to encrypt: %s", err.Error())
		}

		plaintext, err := Decrypt(ciphertext, sharedSecret)
		if err != nil {
			t.Errorf("failed to decrypt: %s", err.Error())
		}

		if message != plaintext {
			t.Errorf("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)
	if plaintext != "hello" {
		t.Fatal("invalid decryption of nostr-tools payload")
	}
}