protocol_test.go raw

   1  package wire
   2  
   3  import "testing"
   4  
   5  // TestServiceFlagStringer tests the stringized output for service flag types.
   6  func TestServiceFlagStringer(t *testing.T) {
   7  	tests := []struct {
   8  		in   ServiceFlag
   9  		want string
  10  	}{
  11  		{0, "0x0"},
  12  		{SFNodeNetwork, "SFNodeNetwork"},
  13  		{SFNodeGetUTXO, "SFNodeGetUTXO"},
  14  		{SFNodeBloom, "SFNodeBloom"},
  15  		{SFNodeWitness, "SFNodeWitness"},
  16  		{SFNodeXthin, "SFNodeXthin"},
  17  		{SFNodeBit5, "SFNodeBit5"},
  18  		{SFNodeCF, "SFNodeCF"},
  19  		{SFNode2X, "SFNode2X"},
  20  		{0xffffffff,
  21  			"SFNodeNetwork|SFNodeGetUTXO|SFNodeBloom|SFNodeWitness|SFNodeXthin|SFNodeBit5|SFNodeCF|SFNode2X|0xffffff00",
  22  		},
  23  	}
  24  	t.Logf("Running %d tests", len(tests))
  25  	for i, test := range tests {
  26  		result := test.in.String()
  27  		if result != test.want {
  28  			t.Errorf("String #%d\n got: %s want: %s", i, result,
  29  				test.want,
  30  			)
  31  			continue
  32  		}
  33  	}
  34  }
  35  
  36  // TestBitcoinNetStringer tests the stringized output for bitcoin net types.
  37  func TestBitcoinNetStringer(t *testing.T) {
  38  	tests := []struct {
  39  		in   BitcoinNet
  40  		want string
  41  	}{
  42  		{MainNet, "MainNet"},
  43  		{TestNet, "TestNet"},
  44  		{TestNet3, "TestNet3"},
  45  		{SimNet, "SimNet"},
  46  		{0xffffffff, "Unknown BitcoinNet (4294967295)"},
  47  	}
  48  	t.Logf("Running %d tests", len(tests))
  49  	for i, test := range tests {
  50  		result := test.in.String()
  51  		if result != test.want {
  52  			t.Errorf("String #%d\n got: %s want: %s", i, result,
  53  				test.want,
  54  			)
  55  			continue
  56  		}
  57  	}
  58  }
  59