parser_test.go

package opentick

import (
	"github.com/alecthomas/repr"
	"github.com/stretchr/testify/assert"
	"strings"
	"testing"
)

var sqlSelectStmt = "select a, adj(b) from test where a > 1.2 and b < 2 limit -2"
var sqlInsertStmt = "INSERT into x(x, y) values(1., ?)"
var sqlInsertAst = `
&opentick.Ast{
  Insert: &opentick.AstInsert{
    Table: &opentick.AstTableName{
      A: &"x",
    },
    Cols: []string{
      "x",
      "y",
    },
    Values: []opentick.AstValue{
      opentick.AstValue{
        Number: &opentick.AstNumber{
          Float: &1,
        },
      },
      opentick.AstValue{
        Placeholder: &"?",
      },
    },
  },
}
`
var sqlSelectAst = `
&opentick.Ast{
  Select: &opentick.AstSelect{
    Selected: &opentick.AstSelectExpression{
      Cols: []opentick.AstSelectCol{
        opentick.AstSelectCol{
          Name: &"a",
        },
        opentick.AstSelectCol{
          Func: &opentick.AstSelectFunc{
            Name: &"ADJ",
            Col: &"b",
          },
        },
      },
    },
    Table: &opentick.AstTableName{
      A: &"test",
    },
    Where: &opentick.AstExpression{
      And: []opentick.AstCondition{
        opentick.AstCondition{
          LHS: &"a",
          Operator: &">",
          RHS: &opentick.AstValue{
            Number: &opentick.AstNumber{
              Float: &1.2,
            },
          },
        },
        opentick.AstCondition{
          LHS: &"b",
          Operator: &"<",
          RHS: &opentick.AstValue{
            Number: &opentick.AstNumber{
              Int: &2,
            },
          },
        },
      },
    },
    Limit: &-2,
  },
}
`

func Test_Parse(t *testing.T) {
	stmt, err := Parse(sqlSelectStmt)
	assert.Equal(t, nil, err)
	assert.Equal(t, strings.TrimSpace(sqlSelectAst), repr.String(stmt, repr.Indent("  "), repr.OmitEmpty(true)))
	stmt, err = Parse(sqlInsertStmt)
	assert.Equal(t, nil, err)
	assert.Equal(t, strings.TrimSpace(sqlInsertAst), repr.String(stmt, repr.Indent("  "), repr.OmitEmpty(true)))
}

func Benchmark_Parse(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, err := Parse(sqlSelectStmt)
		if err != nil {
			b.Fatal(err)
		}
	}
}

func Test_CreateTableSql(t *testing.T) {
	sqlCreateTable1 := `
	create table test.test(
		symbol_id bigint,
		interval int, 
  	tm timestamp,
		open double,
		high double,
		low double,
		close double,
		volume double,
		primary key (symbol_id, interval, tm)
	)
  `
	_, err := Parse(sqlCreateTable1)
	assert.Equal(t, nil, err)

	_, err = Parse("create table test.test(x x)")
	assert.NotEqual(t, nil, err)
}