github.com/arieschain/arieschain@v0.0.0-20191023063405-37c074544356/rlp/raw_test.go (about)

     1  package rlp
     2  
     3  import (
     4  	"bytes"
     5  	"io"
     6  	"reflect"
     7  	"testing"
     8  )
     9  
    10  func TestCountValues(t *testing.T) {
    11  	tests := []struct {
    12  		input string // note: spaces in input are stripped by unhex
    13  		count int
    14  		err   error
    15  	}{
    16  		// simple cases
    17  		{"", 0, nil},
    18  		{"00", 1, nil},
    19  		{"80", 1, nil},
    20  		{"C0", 1, nil},
    21  		{"01 02 03", 3, nil},
    22  		{"01 C406070809 02", 3, nil},
    23  		{"820101 820202 8403030303 04", 4, nil},
    24  
    25  		// size errors
    26  		{"8142", 0, ErrCanonSize},
    27  		{"01 01 8142", 0, ErrCanonSize},
    28  		{"02 84020202", 0, ErrValueTooLarge},
    29  
    30  		{
    31  			input: "A12000BF49F440A1CD0527E4D06E2765654C0F56452257516D793A9B8D604DCFDF2AB853F851808D10000000000000000000000000A056E81F171BCC55A6FF8345E692C0F86E5B48E01B996CADC001622FB5E363B421A0C5D2460186F7233C927E7DB2DCC703C0E500B653CA82273B7BFAD8045D85A470",
    32  			count: 2,
    33  		},
    34  	}
    35  	for i, test := range tests {
    36  		count, err := CountValues(unhex(test.input))
    37  		if count != test.count {
    38  			t.Errorf("test %d: count mismatch, got %d want %d\ninput: %s", i, count, test.count, test.input)
    39  		}
    40  		if !reflect.DeepEqual(err, test.err) {
    41  			t.Errorf("test %d: err mismatch, got %q want %q\ninput: %s", i, err, test.err, test.input)
    42  		}
    43  	}
    44  }
    45  
    46  func TestSplitTypes(t *testing.T) {
    47  	if _, _, err := SplitString(unhex("C100")); err != ErrExpectedString {
    48  		t.Errorf("SplitString returned %q, want %q", err, ErrExpectedString)
    49  	}
    50  	if _, _, err := SplitList(unhex("01")); err != ErrExpectedList {
    51  		t.Errorf("SplitString returned %q, want %q", err, ErrExpectedList)
    52  	}
    53  	if _, _, err := SplitList(unhex("81FF")); err != ErrExpectedList {
    54  		t.Errorf("SplitString returned %q, want %q", err, ErrExpectedList)
    55  	}
    56  }
    57  
    58  func TestSplit(t *testing.T) {
    59  	tests := []struct {
    60  		input     string
    61  		kind      Kind
    62  		val, rest string
    63  		err       error
    64  	}{
    65  		{input: "01FFFF", kind: Byte, val: "01", rest: "FFFF"},
    66  		{input: "80FFFF", kind: String, val: "", rest: "FFFF"},
    67  		{input: "C3010203", kind: List, val: "010203"},
    68  
    69  		// errors
    70  		{input: "", err: io.ErrUnexpectedEOF},
    71  
    72  		{input: "8141", err: ErrCanonSize, rest: "8141"},
    73  		{input: "B800", err: ErrCanonSize, rest: "B800"},
    74  		{input: "B802FFFF", err: ErrCanonSize, rest: "B802FFFF"},
    75  		{input: "B90000", err: ErrCanonSize, rest: "B90000"},
    76  		{input: "B90055", err: ErrCanonSize, rest: "B90055"},
    77  		{input: "BA0002FFFF", err: ErrCanonSize, rest: "BA0002FFFF"},
    78  		{input: "F800", err: ErrCanonSize, rest: "F800"},
    79  		{input: "F90000", err: ErrCanonSize, rest: "F90000"},
    80  		{input: "F90055", err: ErrCanonSize, rest: "F90055"},
    81  		{input: "FA0002FFFF", err: ErrCanonSize, rest: "FA0002FFFF"},
    82  
    83  		{input: "81", err: ErrValueTooLarge, rest: "81"},
    84  		{input: "8501010101", err: ErrValueTooLarge, rest: "8501010101"},
    85  		{input: "C60607080902", err: ErrValueTooLarge, rest: "C60607080902"},
    86  
    87  		// size check overflow
    88  		{input: "BFFFFFFFFFFFFFFFFF", err: ErrValueTooLarge, rest: "BFFFFFFFFFFFFFFFFF"},
    89  		{input: "FFFFFFFFFFFFFFFFFF", err: ErrValueTooLarge, rest: "FFFFFFFFFFFFFFFFFF"},
    90  
    91  		{
    92  			input: "B838FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF",
    93  			err:   ErrValueTooLarge,
    94  			rest:  "B838FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF",
    95  		},
    96  		{
    97  			input: "F838FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF",
    98  			err:   ErrValueTooLarge,
    99  			rest:  "F838FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF",
   100  		},
   101  
   102  		// a few bigger values, just for kicks
   103  		{
   104  			input: "F839FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF",
   105  			kind:  List,
   106  			val:   "FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF",
   107  			rest:  "",
   108  		},
   109  		{
   110  			input: "F90211A060EF29F20CC1007AE6E9530AEE16F4B31F8F1769A2D1264EC995C6D1241868D6A07C62AB8AC9838F5F5877B20BB37B387BC2106E97A3D52172CBEDB5EE17C36008A00EAB6B7324AADC0F6047C6AFC8229F09F7CF451B51D67C8DFB08D49BA8C3C626A04453343B2F3A6E42FCF87948F88AF7C8FC16D0C2735CBA7F026836239AB2C15FA024635C7291C882CE4C0763760C1A362DFC3FFCD802A55722236DE058D74202ACA0A220C808DE10F55E40AB25255201CFF009EA181D3906638E944EE2BF34049984A08D325AB26796F1CCB470F69C0F842501DC35D368A0C2575B2D243CFD1E8AB0FDA0B5298FF60DA5069463D610513C9F04F24051348391A143AFFAB7197DFACDEA72A02D2A7058A4463F8FB69378369E11EF33AE3252E2DB86CB545B36D3C26DDECE5AA0888F97BCA8E0BD83DC5B3B91CFF5FAF2F66F9501010682D67EF4A3B4E66115FBA0E8175A60C93BE9ED02921958F0EA55DA0FB5E4802AF5846147BAD92BC2D8AF26A08B3376FF433F3A4250FA64B7F804004CAC5807877D91C4427BD1CD05CF912ED8A09B32EF0F03BD13C37FF950C0CCCEFCCDD6669F2E7F2AA5CB859928E84E29763EA09BBA5E46610C8C8B1F8E921E5691BF8C7E40D75825D5EA3217AA9C3A8A355F39A0EEB95BC78251CCCEC54A97F19755C4A59A293544EEE6119AFA50531211E53C4FA00B6E86FE150BF4A9E0FEEE9C90F5465E617A861BB5E357F942881EE762212E2580",
   111  			kind:  List,
   112  			val:   "A060EF29F20CC1007AE6E9530AEE16F4B31F8F1769A2D1264EC995C6D1241868D6A07C62AB8AC9838F5F5877B20BB37B387BC2106E97A3D52172CBEDB5EE17C36008A00EAB6B7324AADC0F6047C6AFC8229F09F7CF451B51D67C8DFB08D49BA8C3C626A04453343B2F3A6E42FCF87948F88AF7C8FC16D0C2735CBA7F026836239AB2C15FA024635C7291C882CE4C0763760C1A362DFC3FFCD802A55722236DE058D74202ACA0A220C808DE10F55E40AB25255201CFF009EA181D3906638E944EE2BF34049984A08D325AB26796F1CCB470F69C0F842501DC35D368A0C2575B2D243CFD1E8AB0FDA0B5298FF60DA5069463D610513C9F04F24051348391A143AFFAB7197DFACDEA72A02D2A7058A4463F8FB69378369E11EF33AE3252E2DB86CB545B36D3C26DDECE5AA0888F97BCA8E0BD83DC5B3B91CFF5FAF2F66F9501010682D67EF4A3B4E66115FBA0E8175A60C93BE9ED02921958F0EA55DA0FB5E4802AF5846147BAD92BC2D8AF26A08B3376FF433F3A4250FA64B7F804004CAC5807877D91C4427BD1CD05CF912ED8A09B32EF0F03BD13C37FF950C0CCCEFCCDD6669F2E7F2AA5CB859928E84E29763EA09BBA5E46610C8C8B1F8E921E5691BF8C7E40D75825D5EA3217AA9C3A8A355F39A0EEB95BC78251CCCEC54A97F19755C4A59A293544EEE6119AFA50531211E53C4FA00B6E86FE150BF4A9E0FEEE9C90F5465E617A861BB5E357F942881EE762212E2580",
   113  			rest:  "",
   114  		},
   115  		{
   116  			input: "F877A12000BF49F440A1CD0527E4D06E2765654C0F56452257516D793A9B8D604DCFDF2AB853F851808D10000000000000000000000000A056E81F171BCC55A6FF8345E692C0F86E5B48E01B996CADC001622FB5E363B421A0C5D2460186F7233C927E7DB2DCC703C0E500B653CA82273B7BFAD8045D85A470",
   117  			kind:  List,
   118  			val:   "A12000BF49F440A1CD0527E4D06E2765654C0F56452257516D793A9B8D604DCFDF2AB853F851808D10000000000000000000000000A056E81F171BCC55A6FF8345E692C0F86E5B48E01B996CADC001622FB5E363B421A0C5D2460186F7233C927E7DB2DCC703C0E500B653CA82273B7BFAD8045D85A470",
   119  			rest:  "",
   120  		},
   121  	}
   122  
   123  	for i, test := range tests {
   124  		kind, val, rest, err := Split(unhex(test.input))
   125  		if kind != test.kind {
   126  			t.Errorf("test %d: kind mismatch: got %v, want %v", i, kind, test.kind)
   127  		}
   128  		if !bytes.Equal(val, unhex(test.val)) {
   129  			t.Errorf("test %d: val mismatch: got %x, want %s", i, val, test.val)
   130  		}
   131  		if !bytes.Equal(rest, unhex(test.rest)) {
   132  			t.Errorf("test %d: rest mismatch: got %x, want %s", i, rest, test.rest)
   133  		}
   134  		if err != test.err {
   135  			t.Errorf("test %d: error mismatch: got %q, want %q", i, err, test.err)
   136  		}
   137  	}
   138  }
   139  
   140  func TestReadSize(t *testing.T) {
   141  	tests := []struct {
   142  		input string
   143  		slen  byte
   144  		size  uint64
   145  		err   error
   146  	}{
   147  		{input: "", slen: 1, err: io.ErrUnexpectedEOF},
   148  		{input: "FF", slen: 2, err: io.ErrUnexpectedEOF},
   149  		{input: "00", slen: 1, err: ErrCanonSize},
   150  		{input: "36", slen: 1, err: ErrCanonSize},
   151  		{input: "37", slen: 1, err: ErrCanonSize},
   152  		{input: "38", slen: 1, size: 0x38},
   153  		{input: "FF", slen: 1, size: 0xFF},
   154  		{input: "FFFF", slen: 2, size: 0xFFFF},
   155  		{input: "FFFFFF", slen: 3, size: 0xFFFFFF},
   156  		{input: "FFFFFFFF", slen: 4, size: 0xFFFFFFFF},
   157  		{input: "FFFFFFFFFF", slen: 5, size: 0xFFFFFFFFFF},
   158  		{input: "FFFFFFFFFFFF", slen: 6, size: 0xFFFFFFFFFFFF},
   159  		{input: "FFFFFFFFFFFFFF", slen: 7, size: 0xFFFFFFFFFFFFFF},
   160  		{input: "FFFFFFFFFFFFFFFF", slen: 8, size: 0xFFFFFFFFFFFFFFFF},
   161  		{input: "0102", slen: 2, size: 0x0102},
   162  		{input: "010203", slen: 3, size: 0x010203},
   163  		{input: "01020304", slen: 4, size: 0x01020304},
   164  		{input: "0102030405", slen: 5, size: 0x0102030405},
   165  		{input: "010203040506", slen: 6, size: 0x010203040506},
   166  		{input: "01020304050607", slen: 7, size: 0x01020304050607},
   167  		{input: "0102030405060708", slen: 8, size: 0x0102030405060708},
   168  	}
   169  
   170  	for _, test := range tests {
   171  		size, err := readSize(unhex(test.input), test.slen)
   172  		if err != test.err {
   173  			t.Errorf("readSize(%s, %d): error mismatch: got %q, want %q", test.input, test.slen, err, test.err)
   174  			continue
   175  		}
   176  		if size != test.size {
   177  			t.Errorf("readSize(%s, %d): size mismatch: got %#x, want %#x", test.input, test.slen, size, test.size)
   178  		}
   179  	}
   180  }