okenvelope_test.go raw

   1  package okenvelope
   2  
   3  import (
   4  	"testing"
   5  
   6  	"next.orly.dev/pkg/nostr/encoders/envelopes"
   7  	"next.orly.dev/pkg/nostr/encoders/envelopes/messages"
   8  	"next.orly.dev/pkg/nostr/utils"
   9  	"github.com/minio/sha256-simd"
  10  	"next.orly.dev/pkg/lol/chk"
  11  	"lukechampine.com/frand"
  12  )
  13  
  14  func TestMarshalUnmarshal(t *testing.T) {
  15  	var err error
  16  	rb, rb1, rb2 := make([]byte, 0, 65535), make([]byte, 0, 65535), make(
  17  		[]byte, 0, 65535,
  18  	)
  19  	for i := range 1000 {
  20  		var randMsg []byte
  21  		ok := i%2 == 1
  22  		if !ok {
  23  			randMsg = messages.RandomMessage()
  24  		}
  25  		req := NewFrom(frand.Bytes(sha256.Size), ok, randMsg)
  26  		rb = req.Marshal(rb)
  27  		rb1 = rb1[:len(rb)]
  28  		copy(rb1, rb)
  29  		var rem []byte
  30  		var l string
  31  		if l, rb, err = envelopes.Identify(rb); chk.E(err) {
  32  			t.Fatal(err)
  33  		}
  34  		if l != L {
  35  			t.Fatalf("invalid sentinel %s, expect %s", l, L)
  36  		}
  37  		req2 := New()
  38  		if rem, err = req2.Unmarshal(rb); chk.E(err) {
  39  			t.Fatal(err)
  40  		}
  41  		if len(rem) > 0 {
  42  			t.Fatalf(
  43  				"unmarshal failed, remainder\n%d %s",
  44  				len(rem), rem,
  45  			)
  46  		}
  47  		rb2 = req2.Marshal(rb2)
  48  		if !utils.FastEqual(rb1, rb2) {
  49  			if len(rb1) != len(rb2) {
  50  				t.Fatalf(
  51  					"unmarshal failed, different lengths\n%d %s\n%d %s\n",
  52  					len(rb1), rb1, len(rb2), rb2,
  53  				)
  54  			}
  55  			for i := range rb1 {
  56  				if rb1[i] != rb2[i] {
  57  					t.Fatalf(
  58  						"unmarshal failed, difference at position %d\n%d %s\n%s\n%d %s\n%s\n",
  59  						i, len(rb1), rb1[:i], rb1[i:], len(rb2), rb2[:i],
  60  						rb2[i:],
  61  					)
  62  				}
  63  			}
  64  			t.Fatalf(
  65  				"unmarshal failed\n%d %s\n%d %s\n",
  66  				len(rb1), rb1, len(rb2), rb2,
  67  			)
  68  		}
  69  		rb, rb1, rb2 = rb[:0], rb1[:0], rb2[:0]
  70  	}
  71  }
  72