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 `) 154
