// Copyright 2019 PingCAP, Inc. // // 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, // See the License for the specific language governing permissions and // limitations under the License. package main import "testing" func TestIsQuery(t *testing.T) { tbl := []struct { sql string ok bool }{ {"/*comment*/ select 1;", true}, {"/*comment*/ /*comment*/ select 1;", true}, {"select /*comment*/ 1 /*comment*/;", true}, {"(select /*comment*/ 1 /*comment*/);", true}, } for _, tb := range tbl { if isQuery(tb.sql) != tb.ok { t.Fatalf("%s", tb.sql) } } } func TestTrimSQL(t *testing.T) { tbl := []struct { sql string target string }{ {"/*comment*/ select 1; ", "select 1;"}, {"/*comment*/ /*comment*/ select 1;", "select 1;"}, {"select /*comment*/ 1 /*comment*/;", "select /*comment*/ 1 /*comment*/;"}, {"/*comment select 1; ", "/*comment select 1;"}, } for _, tb := range tbl { if trimSQL(tb.sql) != tb.target { t.Fatalf("%s", tb.sql) } } }