closeenvelope_test.go raw

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