filter_test.go 590 B

12345678910111213141516171819202122232425
  1. package core
  2. import (
  3. "testing"
  4. "github.com/stretchr/testify/assert"
  5. )
  6. type quoterOnly struct {
  7. Dialect
  8. }
  9. func (q *quoterOnly) Quote(item string) string {
  10. return "[" + item + "]"
  11. }
  12. func TestQuoteFilter_Do(t *testing.T) {
  13. f := QuoteFilter{}
  14. sql := "SELECT `COLUMN_NAME` FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE `TABLE_SCHEMA` = ? AND `TABLE_NAME` = ? AND `COLUMN_NAME` = ?"
  15. res := f.Do(sql, new(quoterOnly), nil)
  16. assert.EqualValues(t,
  17. "SELECT [COLUMN_NAME] FROM [INFORMATION_SCHEMA].[COLUMNS] WHERE [TABLE_SCHEMA] = ? AND [TABLE_NAME] = ? AND [COLUMN_NAME] = ?",
  18. res,
  19. )
  20. }