github.com/badrootd/nibiru-cometbft@v0.37.5-0.20240307173500-2a75559eee9b/test/fuzz/p2p/addrbook/fuzz.go (about)

     1  package addr
     2  
     3  import (
     4  	"encoding/json"
     5  	"fmt"
     6  	"math/rand"
     7  
     8  	"github.com/badrootd/nibiru-cometbft/p2p"
     9  	"github.com/badrootd/nibiru-cometbft/p2p/pex"
    10  )
    11  
    12  var addrBook = pex.NewAddrBook("./testdata/addrbook.json", true)
    13  
    14  func Fuzz(data []byte) int {
    15  	addr := new(p2p.NetAddress)
    16  	if err := json.Unmarshal(data, addr); err != nil {
    17  		return -1
    18  	}
    19  
    20  	// Fuzz AddAddress.
    21  	err := addrBook.AddAddress(addr, addr)
    22  	if err != nil {
    23  		return 0
    24  	}
    25  
    26  	// Also, make sure PickAddress always returns a non-nil address.
    27  	bias := rand.Intn(100) //nolint:gosec
    28  	if p := addrBook.PickAddress(bias); p == nil {
    29  		panic(fmt.Sprintf("picked a nil address (bias: %d, addrBook size: %v)",
    30  			bias, addrBook.Size()))
    31  	}
    32  
    33  	return 1
    34  }