countenvelope_test.go raw

   1  package countenvelope
   2  
   3  import (
   4  	"testing"
   5  
   6  	"next.orly.dev/pkg/nostr/encoders/envelopes"
   7  	"next.orly.dev/pkg/nostr/encoders/filter"
   8  	"next.orly.dev/pkg/nostr/utils"
   9  	"next.orly.dev/pkg/nostr/utils/bufpool"
  10  	"next.orly.dev/pkg/lol/chk"
  11  	"lukechampine.com/frand"
  12  )
  13  
  14  func TestRequest(t *testing.T) {
  15  	var err error
  16  	for i := range 1000 {
  17  		rb, rb1, rb2 := bufpool.Get(), bufpool.Get(), bufpool.Get()
  18  		var f filter.S
  19  		if f, err = filter.GenFilters(); chk.E(err) {
  20  			t.Fatal(err)
  21  		}
  22  		s := utils.NewSubscription(i)
  23  		req := NewRequest(s, f)
  24  		rb = req.Marshal(rb)
  25  		rb1 = append(rb1, rb...)
  26  		var rem []byte
  27  		var l string
  28  		if l, rb, err = envelopes.Identify(rb); chk.E(err) {
  29  			t.Fatal(err)
  30  		}
  31  		if l != L {
  32  			t.Fatalf("invalid sentinel %s, expect %s", l, L)
  33  		}
  34  		req2 := New()
  35  		if rem, err = req2.Unmarshal(rb); chk.E(err) {
  36  			t.Fatal(err)
  37  		}
  38  		if len(rem) > 0 {
  39  			t.Fatalf(
  40  				"unmarshal failed, remainder\n%d %s",
  41  				len(rem), rem,
  42  			)
  43  		}
  44  		rb2 = req2.Marshal(rb2)
  45  		if !utils.FastEqual(rb1, rb2) {
  46  			if len(rb1) != len(rb2) {
  47  				t.Fatalf(
  48  					"unmarshal failed, different lengths\n%d %s\n%d %s\n",
  49  					len(rb1), rb1, len(rb2), rb2,
  50  				)
  51  			}
  52  			for i := range rb1 {
  53  				if rb1[i] != rb2[i] {
  54  					t.Fatalf(
  55  						"unmarshal failed, difference at position %d\n%d %s\n%s\n%d %s\n%s\n",
  56  						i, len(rb1), rb1[:i], rb1[i:], len(rb2), rb2[:i],
  57  						rb2[i:],
  58  					)
  59  				}
  60  			}
  61  			t.Fatalf(
  62  				"unmarshal failed\n%d %s\n%d %s\n",
  63  				len(rb1), rb1, len(rb2), rb2,
  64  			)
  65  		}
  66  		bufpool.Put(rb1)
  67  		bufpool.Put(rb2)
  68  		bufpool.Put(rb)
  69  	}
  70  }
  71  
  72  func TestResponse(t *testing.T) {
  73  	var err error
  74  	for i := range 1000 {
  75  		rb, rb1, rb2 := bufpool.Get(), bufpool.Get(), bufpool.Get()
  76  		s := utils.NewSubscription(i)
  77  		var res *Response
  78  		if i&2 == 0 {
  79  			if res, err = NewResponseFrom(
  80  				s, frand.Intn(200), true,
  81  			); chk.E(err) {
  82  				t.Fatal(err)
  83  			}
  84  		} else {
  85  			if res, err = NewResponseFrom(s, frand.Intn(200)); chk.E(err) {
  86  				t.Fatal(err)
  87  			}
  88  		}
  89  		rb = res.Marshal(rb)
  90  		rb1 = append(rb1, rb...)
  91  		var rem []byte
  92  		var l string
  93  		if l, rb, err = envelopes.Identify(rb); chk.E(err) {
  94  			t.Fatal(err)
  95  		}
  96  		if l != L {
  97  			t.Fatalf("invalid sentinel %s, expect %s", l, L)
  98  		}
  99  		res2 := NewResponse()
 100  		if rem, err = res2.Unmarshal(rb); chk.E(err) {
 101  			t.Fatal(err)
 102  		}
 103  		if len(rem) > 0 {
 104  			t.Fatalf(
 105  				"unmarshal failed, remainder\n%d %s",
 106  				len(rem), rem,
 107  			)
 108  		}
 109  		rb2 = res2.Marshal(rb2)
 110  		if !utils.FastEqual(rb1, rb2) {
 111  			if len(rb1) != len(rb2) {
 112  				t.Fatalf(
 113  					"unmarshal failed, different lengths\n%d %s\n%d %s\n",
 114  					len(rb1), rb1, len(rb2), rb2,
 115  				)
 116  			}
 117  			for i := range rb1 {
 118  				if rb1[i] != rb2[i] {
 119  					t.Fatalf(
 120  						"unmarshal failed, difference at position %d\n%d %s\n%s\n%d %s\n%s\n",
 121  						i, len(rb1), rb1[:i], rb1[i:], len(rb2), rb2[:i],
 122  						rb2[i:],
 123  					)
 124  				}
 125  			}
 126  			t.Fatalf(
 127  				"unmarshal failed\n%d %s\n%d %s\n",
 128  				len(rb1), rb1, len(rb2), rb2,
 129  			)
 130  		}
 131  		bufpool.Put(rb1)
 132  		bufpool.Put(rb2)
 133  		bufpool.Put(rb)
 134  	}
 135  }
 136