github.com/ccccaoqing/test@v0.0.0-20220510085219-3985d23445c0/src/encoding/hex/hex_test.go (about)

     1  // Copyright 2009 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package hex
     6  
     7  import (
     8  	"bytes"
     9  	"testing"
    10  )
    11  
    12  type encDecTest struct {
    13  	enc string
    14  	dec []byte
    15  }
    16  
    17  var encDecTests = []encDecTest{
    18  	{"", []byte{}},
    19  	{"0001020304050607", []byte{0, 1, 2, 3, 4, 5, 6, 7}},
    20  	{"08090a0b0c0d0e0f", []byte{8, 9, 10, 11, 12, 13, 14, 15}},
    21  	{"f0f1f2f3f4f5f6f7", []byte{0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7}},
    22  	{"f8f9fafbfcfdfeff", []byte{0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff}},
    23  	{"67", []byte{'g'}},
    24  	{"e3a1", []byte{0xe3, 0xa1}},
    25  }
    26  
    27  func TestEncode(t *testing.T) {
    28  	for i, test := range encDecTests {
    29  		dst := make([]byte, EncodedLen(len(test.dec)))
    30  		n := Encode(dst, test.dec)
    31  		if n != len(dst) {
    32  			t.Errorf("#%d: bad return value: got: %d want: %d", i, n, len(dst))
    33  		}
    34  		if string(dst) != test.enc {
    35  			t.Errorf("#%d: got: %#v want: %#v", i, dst, test.enc)
    36  		}
    37  	}
    38  }
    39  
    40  func TestDecode(t *testing.T) {
    41  	// Case for decoding uppercase hex characters, since
    42  	// Encode always uses lowercase.
    43  	decTests := append(encDecTests, encDecTest{"F8F9FAFBFCFDFEFF", []byte{0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff}})
    44  	for i, test := range decTests {
    45  		dst := make([]byte, DecodedLen(len(test.enc)))
    46  		n, err := Decode(dst, []byte(test.enc))
    47  		if err != nil {
    48  			t.Errorf("#%d: bad return value: got:%d want:%d", i, n, len(dst))
    49  		} else if !bytes.Equal(dst, test.dec) {
    50  			t.Errorf("#%d: got: %#v want: %#v", i, dst, test.dec)
    51  		}
    52  	}
    53  }
    54  
    55  func TestEncodeToString(t *testing.T) {
    56  	for i, test := range encDecTests {
    57  		s := EncodeToString(test.dec)
    58  		if s != test.enc {
    59  			t.Errorf("#%d got:%s want:%s", i, s, test.enc)
    60  		}
    61  	}
    62  }
    63  
    64  func TestDecodeString(t *testing.T) {
    65  	for i, test := range encDecTests {
    66  		dst, err := DecodeString(test.enc)
    67  		if err != nil {
    68  			t.Errorf("#%d: unexpected err value: %s", i, err)
    69  			continue
    70  		}
    71  		if !bytes.Equal(dst, test.dec) {
    72  			t.Errorf("#%d: got: %#v want: #%v", i, dst, test.dec)
    73  		}
    74  	}
    75  }
    76  
    77  type errTest struct {
    78  	in  string
    79  	err string
    80  }
    81  
    82  var errTests = []errTest{
    83  	{"0", "encoding/hex: odd length hex string"},
    84  	{"0g", "encoding/hex: invalid byte: U+0067 'g'"},
    85  	{"00gg", "encoding/hex: invalid byte: U+0067 'g'"},
    86  	{"0\x01", "encoding/hex: invalid byte: U+0001"},
    87  }
    88  
    89  func TestInvalidErr(t *testing.T) {
    90  	for i, test := range errTests {
    91  		dst := make([]byte, DecodedLen(len(test.in)))
    92  		_, err := Decode(dst, []byte(test.in))
    93  		if err == nil {
    94  			t.Errorf("#%d: expected error; got none", i)
    95  		} else if err.Error() != test.err {
    96  			t.Errorf("#%d: got: %v want: %v", i, err, test.err)
    97  		}
    98  	}
    99  }
   100  
   101  func TestInvalidStringErr(t *testing.T) {
   102  	for i, test := range errTests {
   103  		_, err := DecodeString(test.in)
   104  		if err == nil {
   105  			t.Errorf("#%d: expected error; got none", i)
   106  		} else if err.Error() != test.err {
   107  			t.Errorf("#%d: got: %v want: %v", i, err, test.err)
   108  		}
   109  	}
   110  }
   111  
   112  func TestDumper(t *testing.T) {
   113  	var in [40]byte
   114  	for i := range in {
   115  		in[i] = byte(i + 30)
   116  	}
   117  
   118  	for stride := 1; stride < len(in); stride++ {
   119  		var out bytes.Buffer
   120  		dumper := Dumper(&out)
   121  		done := 0
   122  		for done < len(in) {
   123  			todo := done + stride
   124  			if todo > len(in) {
   125  				todo = len(in)
   126  			}
   127  			dumper.Write(in[done:todo])
   128  			done = todo
   129  		}
   130  
   131  		dumper.Close()
   132  		if !bytes.Equal(out.Bytes(), expectedHexDump) {
   133  			t.Errorf("stride: %d failed. got:\n%s\nwant:\n%s", stride, out.Bytes(), expectedHexDump)
   134  		}
   135  	}
   136  }
   137  
   138  func TestDump(t *testing.T) {
   139  	var in [40]byte
   140  	for i := range in {
   141  		in[i] = byte(i + 30)
   142  	}
   143  
   144  	out := []byte(Dump(in[:]))
   145  	if !bytes.Equal(out, expectedHexDump) {
   146  		t.Errorf("got:\n%s\nwant:\n%s", out, expectedHexDump)
   147  	}
   148  }
   149  
   150  var expectedHexDump = []byte(`00000000  1e 1f 20 21 22 23 24 25  26 27 28 29 2a 2b 2c 2d  |.. !"#$%&'()*+,-|
   151  00000010  2e 2f 30 31 32 33 34 35  36 37 38 39 3a 3b 3c 3d  |./0123456789:;<=|
   152  00000020  3e 3f 40 41 42 43 44 45                           |>?@ABCDE|
   153  `)