Function TestGBKEncoding

pkg/parser/parser_test.go:7432–7478

(t *testing.T)

}

func TestGBKEncoding(t *testing.T) {
	p := parser.New()
	gbkEncoding, _ := charset.Lookup("gbk")
	encoder := gbkEncoding.NewEncoder()
	sql, err := encoder.String("create table 测试表 (测试列 varchar(255) default 'GBK测试用例');")
	require.NoError(t, err)

	stmt, _, err := p.ParseSQL(sql)
	require.NoError(t, err)
	checker := &gbkEncodingChecker{}
	_, _ = stmt[0].Accept(checker)
	require.NotEqual(t, "测试表", checker.tblName)
	require.NotEqual(t, "测试列", checker.colName)

	gbkOpt := parser.CharsetClient("gbk")
	stmt, _, err = p.ParseSQL(sql, gbkOpt)
	require.NoError(t, err)
	_, _ = stmt[0].Accept(checker)
	require.Equal(t, "测试表", checker.tblName)
	require.Equal(t, "测试列", checker.colName)
	require.Equal(t, "GBK测试用例", checker.expr)

	_, _, err = p.ParseSQL("select _gbk '\xc6\x5c' from dual;")
	require.Error(t, err)

	for _, test := range []struct {
		sql string
		err bool
	}{
		{"select '\xc6\x5c' from `\xab\x60`;", false},
		{`prepare p1 from "insert into t values ('中文');";`, false},
		{"select '啊';", false},
		{"create table t1(s set('a一','b二','c三'));", false},
		{"insert into t3 values('一a');", false},
		{"select '\xa5\x5c'", false},
		{"select '''\xa5\x5c'", false},
		{"select ```\xa5\x5c`", false},
		{"select '\x65\x5c'", true},
	} {
		_, _, err = p.ParseSQL(test.sql, gbkOpt)
		if test.err {
			require.Error(t, err, test.sql)
		} else {
			require.NoError(t, err, test.sql)
		}
	}
}

type gbkEncodingChecker struct {
	tblName string