1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
// Copyright 2018 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package parser
import (
"errors"
"strings"
"testing"
)
func TestUnescape(t *testing.T) {
tests := []struct {
in string
out interface{}
isBytes bool
}{
// Simple string unescaping tests.
{in: `'hello'`, out: `hello`},
{in: `r'hello'`, out: `hello`},
{in: `""`, out: ``},
{in: `"\\\""`, out: `\"`},
{in: `"\\"`, out: `\`},
{in: `'''x''x'''`, out: `x''x`},
{in: `"""x""x"""`, out: `x""x`},
{in: `"\303\277"`, out: `ÿ`},
{in: `"\377"`, out: `ÿ`},
{in: `"\u263A\u263A"`, out: `☺☺`},
{in: `"\a\b\f\n\r\t\v\'\"\\\? Legal escapes"`, out: "\a\b\f\n\r\t\v'\"\\? Legal escapes"},
// Byte unescaping tests.
{in: `"abc"`, out: "\x61\x62\x63", isBytes: true},
{in: `"ÿ"`, out: "\xc3\xbf", isBytes: true},
{in: `"\303\277"`, out: "\xc3\xbf", isBytes: true},
{in: `"\377"`, out: "\xff", isBytes: true},
{in: `"\xff"`, out: "\xff", isBytes: true},
{in: `"\xc3\xbf"`, out: "\xc3\xbf", isBytes: true},
{in: `'''"Kim\t"'''`, out: "\x22\x4b\x69\x6d\x09\x22", isBytes: true},
// Escaping errors.
{in: `"\a\b\f\n\r\t\v\'\"\\\? Illegal escape \>"`, out: errors.New("unable to unescape string")},
{in: `"\u00f"`, out: errors.New("unable to unescape string")},
{in: `"\u00fÿ"`, out: errors.New("unable to unescape string")},
{in: `"\u00ff"`, out: errors.New("unable to unescape string"), isBytes: true},
{in: `"\U00ff"`, out: errors.New("unable to unescape string"), isBytes: true},
{in: `"\26"`, out: errors.New("unable to unescape octal sequence")},
{in: `"\268"`, out: errors.New("unable to unescape octal sequence")},
{in: `"\267\"`, out: errors.New(`found '\' as last character`)},
{in: `'`, out: errors.New("unable to unescape string")},
{in: `*hello*`, out: errors.New("unable to unescape string")},
{in: `r'''hello'`, out: errors.New("unable to unescape string")},
{in: `r"""hello"`, out: errors.New("unable to unescape string")},
{in: `r"""hello"`, out: errors.New("unable to unescape string")},
}
for _, tst := range tests {
tc := tst
t.Run(tc.in, func(t *testing.T) {
got, err := unescape(tc.in, tc.isBytes)
if err != nil {
expect, isErr := tc.out.(error)
if isErr {
if !strings.Contains(err.Error(), expect.Error()) {
t.Errorf("unescape(%s, %v) errored with %v, wanted %v", tc.in, tc.isBytes, err, expect)
}
} else {
t.Fatalf("unescape(%s, %v) failed: %v", tc.in, tc.isBytes, err)
}
} else if got != tc.out {
t.Errorf("unescape(%s, %v) got %v, wanted %v", tc.in, tc.isBytes, got, tc.out)
}
})
}
}
|
