Files
nostrlib/types_test.go
2025-11-23 06:57:54 -03:00

98 lines
2.9 KiB
Go

package nostr
import (
"testing"
"github.com/stretchr/testify/require"
"github.com/templexxx/cpu"
)
func TestIDJSONEncoding(t *testing.T) {
id := MustIDFromHex("6348118f31cc19fe6b699fa2db5edff315429f7ebb6cc16d3627fdbc4dcae904")
// test marshaling
b, err := json.Marshal(id)
require.NoError(t, err)
require.Equal(t, `"6348118f31cc19fe6b699fa2db5edff315429f7ebb6cc16d3627fdbc4dcae904"`, string(b))
// test unmarshaling
var id2 ID
err = json.Unmarshal(b, &id2)
require.NoError(t, err)
require.Equal(t, id, id2)
// test unmarshaling invalid json
err = json.Unmarshal([]byte(`"not64chars"`), &id2)
require.Error(t, err)
// test unmarshaling invalid hex
if !cpu.X86.HasAVX2 {
err = json.Unmarshal([]byte(`"zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz"`), &id2)
require.Error(t, err)
}
}
func TestPubKeyJSONEncoding(t *testing.T) {
pk := MustPubKeyFromHex("6348118f31cc19fe6b699fa2db5edff315429f7ebb6cc16d3627fdbc4dcae904")
// test marshaling
b, err := json.Marshal(pk)
require.NoError(t, err)
require.Equal(t, `"6348118f31cc19fe6b699fa2db5edff315429f7ebb6cc16d3627fdbc4dcae904"`, string(b))
// test unmarshaling
var pk2 PubKey
err = json.Unmarshal(b, &pk2)
require.NoError(t, err)
require.Equal(t, pk, pk2)
// test unmarshaling invalid json
err = json.Unmarshal([]byte(`"not64chars"`), &pk2)
require.Error(t, err)
// test unmarshaling invalid hex
err = json.Unmarshal([]byte(`"zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz"`), &pk2)
require.Error(t, err)
}
type TestStruct struct {
ID ID `json:"id"`
PubKey PubKey `json:"pubkey"`
Name string `json:"name"`
}
func TestStructWithIDAndPubKey(t *testing.T) {
ts := TestStruct{
ID: MustIDFromHex("6348118f31cc19fe6b699fa2db5edff315429f7ebb6cc16d3627fdbc4dcae904"),
PubKey: MustPubKeyFromHex("123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0"),
Name: "test",
}
// test marshaling
b, err := json.Marshal(ts)
require.NoError(t, err)
require.Equal(t, `{"id":"6348118f31cc19fe6b699fa2db5edff315429f7ebb6cc16d3627fdbc4dcae904","pubkey":"123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0","name":"test"}`, string(b))
// test unmarshaling
var ts2 TestStruct
err = json.Unmarshal(b, &ts2)
require.NoError(t, err)
require.Equal(t, ts, ts2)
// test unmarshaling with missing fields
var ts3 TestStruct
err = json.Unmarshal([]byte(`{"name":"test"}`), &ts3)
require.NoError(t, err)
require.Equal(t, "test", ts3.Name)
require.Equal(t, ZeroID, ts3.ID)
require.Equal(t, ZeroPK, ts3.PubKey)
// test unmarshaling with invalid ID
err = json.Unmarshal([]byte(`{"id":"invalid","pubkey":"123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0","name":"test"}`), &ts2)
require.Error(t, err)
// test unmarshaling with invalid PubKey
err = json.Unmarshal([]byte(`{"id":"6348118f31cc19fe6b699fa2db5edff315429f7ebb6cc16d3627fdbc4dcae904","pubkey":"invalid","name":"test"}`), &ts2)
require.Error(t, err)
}