github_test.go

/*
Copyright 2017 Google 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 agreedto 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 sqlparser

// Additional tests to address the GitHub issues for this fork.

import (
	"io"
	"strings"
	"testing"
)

func TestParsing(t *testing.T) {
	tests := []struct {
		id   int // Github issue ID
		sql  string
		skip string
	}{
		{id: 9, sql: "select 1 as 测试 from dual", skip: "Broken due to ReadByte()"},
		{id: 12, sql: "SELECT * FROM AccessToken LIMIT 10 OFFSET 13"},
		{id: 14, sql: "SELECT DATE_SUB(NOW(), INTERVAL 1 MONTH)"},
		{id: 15, sql: "select STRAIGHT_JOIN t1.* FROM t1 INNER JOIN  t2 ON t1.CommonID = t2.CommonID WHERE t1.FilterID = 1"},
		{id: 16, sql: "SELECT a FROM t WHERE FUNC(a) = 1"}, // Doesn't seem broken, need better example
		{id: 21, sql: `CREATE TABLE t (UpdateDatetime TIMESTAMP NULL DEFAULT CURRENT_TIMESTAMP)`},
		{id: 21, sql: `CREATE TABLE t (UpdateDatetime TIMESTAMP NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间')`},
		{id: 24, sql: `select * from t1 join t2 using(id)`},
	}

	for _, test := range tests {
		if test.skip != "" {
			continue
		}

		if _, err := Parse(test.sql); err != nil {
			t.Errorf("https://github.com/deatheyes/sqlparser/issues/%d:\nParse(%q) err = %s, want nil", test.id, test.sql, err)
		}
	}
}

// ExampleParse is the first example from the README.md.
func ExampleParse() {
	sql := "SELECT * FROM table WHERE a = 'abc'"
	stmt, err := Parse(sql)
	if err != nil {
		// Do something with the err
	}

	// Otherwise do something with stmt
	switch stmt := stmt.(type) {
	case *Select:
		_ = stmt
	case *Insert:
	}
}

// ExampleParseNext is the second example from the README.md.
func ExampleParseNext() {
	r := strings.NewReader("INSERT INTO table1 VALUES (1, 'a'); INSERT INTO table2 VALUES (3, 4);")

	tokens := NewTokenizer(r)
	for {
		stmt, err := ParseNext(tokens)
		if err == io.EOF {
			break
		}
		// Do something with stmt or err.
		_ = stmt
	}
}