package mapr

import (
	"testing"
	"time"
)

func TestParseQuerySimple(t *testing.T) {
	errorQueries := []string{
		"select",
		"select foo from",
		"select foo from bar where baz",
		"select foo from bar where baz <",
		"select foo from bar where baz < 100 bay eq 12 group",
		"select foo from bar where baz < 100 bay eq 12 group by foo order by",
		"select foo from bar where baz < 100 bay eq 12 group by foo, bar, baz order by foo limit",
		"select foo from bar where baz < 100 bay eq 12 group by foo, bar, baz order by foo limit set foo = bar;",
	}
	okQueries := []string{"select foo from bar",
		"select foo from bar where",
		"select foo from bar where baz < 100 bay eq 12",
		"select foo from bar where baz < 100, bay eq 12",
		"select foo from bar where baz < 100 and bay eq 12",
		"select foo from bar where baz < 100 bay eq 12 group by foo, bar, baz order by foo",
		"select foo from bar where baz < 100 bay eq 12 group by foo, bar, baz order by foo limit 23",
		"select foo from bar where baz < 100 bay eq 12 group by foo, bar, baz order by foo limit 23 outfile \"result.csv\"",
		"select foo from bar where baz < 100 bay eq 12 group by foo, bar, baz order by foo limit 23 outfile \"result.csv\" set $foo = maskdigits(bar), $baz = 12, $bay = $foo;",
	}

	for _, queryStr := range errorQueries {
		q, err := NewQuery(queryStr)
		if err == nil {
			t.Errorf("Expected a parse error: %s\n%v", queryStr, q)
			continue
		}
	}

	for _, queryStr := range okQueries {
		_, err := NewQuery(queryStr)
		if err != nil {
			t.Errorf("%s: %s", err.Error(), queryStr)
			continue
		}
	}
}

func TestParseQueryDeep(t *testing.T) {
	dialects := []string{
		"select s1, `from`, count(s3) from table where w1 == 2 and w2 eq \"free beer\" group by g1, g2 order by count(s3) interval 10 limit 23 set $foo = maskdigits(bar), $baz = 12, $bay = $foo logformat generic",
		"SELECT s1, `from`, COUNT(s3) FROM table WHERE w1 == 2 AND w2 eq \"free beer\" GROUP g1, g2 ORDER count(s3) INTERVAL 10 LIMIT 23 SET $foo = maskdigits(bar), $baz = 12, $bay = $foo logformat generic",
	}

	for _, queryStr := range dialects {
		q, err := NewQuery(queryStr)
		if err != nil {
			t.Errorf("%s: %s", err.Error(), queryStr)
		}

		t.Log(q)

		// 'select' clause
		if len(q.Select) != 3 {
			t.Errorf("Expected three elements in 'select' clause but got '%v': %s\n%v", q.Select, queryStr, q)
		}

		if q.Select[0].Field != "s1" {
			t.Errorf("Expected 's1' as first element in 'select' clause but got '%v': %s\n%v", q.Select[0].Field, queryStr, q)
		}
		if q.Select[0].Operation != Last {
			t.Errorf("Expected 'last' as aggregation function of first element in 'select' clause but got '%v': %s\n%v", q.Select[0].Operation, queryStr, q)
		}

		if q.Select[1].Field != "from" {
			t.Errorf("Expected 'from' as second element in 'select' clause but got '%v': %s\n%v", q.Select[1].Field, queryStr, q)
		}
		if q.Select[1].Operation != Last {
			t.Errorf("Expected 'last' as aggregation function of second element in 'select' clause but got '%v': %s\n%v", q.Select[1].Operation, queryStr, q)
		}

		if q.Select[2].Field != "s3" {
			t.Errorf("Expected 's3' as third element in 'select' clause but got '%v': %s\n%v", q.Select[2].Field, queryStr, q)
		}
		if q.Select[2].Operation != Count {
			t.Errorf("Expected 'count' as aggregation function of third  element in 'select' clause but got '%v': %s\n%v", q.Select[2].Operation, queryStr, q)
		}
		if q.Select[2].FieldStorage != "count(s3)" {
			t.Errorf("Expected 'count(s3)' as third element's storage in 'select' clause but got '%v': %s\n%v", q.Select[2].FieldStorage, queryStr, q)
		}

		// 'from' clause
		if q.Table != "TABLE" {
			t.Errorf("Expected 'TABLE' in 'from' clause but got '%v': %s\n%v", q.Table, queryStr, q)
		}

		// 'where' clause
		if len(q.Where) != 2 {
			t.Errorf("Expected two elements in 'where' clause but got '%v': %s\n%v", q.Where, queryStr, q)
		}
		if q.Where[0].lString != "w1" {
			t.Errorf("Expected w1 as first element in 'where' clause but got '%v': %s\n%v", q.Where[0].lString, queryStr, q)
		}
		if q.Where[0].Operation != FloatEq {
			t.Errorf("Expected FloatEq operation in first 'where' condition but got '%v': %s\n%v", q.Where[0].Operation, queryStr, q)
		}
		if q.Where[0].rFloat != 2 {
			t.Errorf("Expected '2' as float argument in first 'where' condition but got '%v': %s\n%v", q.Where[0].rFloat, queryStr, q)
		}
		if q.Where[1].lString != "w2" {
			t.Errorf("Expected w2 as second element in 'where' clause but got '%v': %s\n%v", q.Where[1].lString, queryStr, q)
		}
		if q.Where[1].Operation != StringEq {
			t.Errorf("Expected StringEq operation in second 'where' condition but got '%v': %s\n%v", q.Where[0].Operation, queryStr, q)
		}
		if q.Where[1].rString != "free beer" {
			t.Errorf("Expected 'free beer' as string argument in second 'where' condition but got '%v': %s\n%v", q.Where[0].rString, queryStr, q)
		}

		// 'group by' clause
		if len(q.GroupBy) != 2 {
			t.Errorf("Expected two elements in 'group by' clause but got '%v': %s\n%v", q.GroupBy, queryStr, q)
		}
		if q.GroupBy[0] != "g1" {
			t.Errorf("Expected 'g1' as first element in 'group by' clause but got '%v': %s\n%v", q.GroupBy[0], queryStr, q)
		}
		if q.GroupBy[1] != "g2" {
			t.Errorf("Expected 'g2' as second element in 'group by' clause but got '%v': %s\n%v", q.GroupBy[1], queryStr, q)
		}
		if q.GroupKey != "g1,g2" {
			t.Errorf("Expected 'g1,g2' as group key in 'group by' clause but got '%v': %s\n%v", q.GroupKey, queryStr, q)
		}

		// 'order by' clause
		if q.OrderBy != "count(s3)" {
			t.Errorf("Expected 'count(s3)' as element in 'order by' clause but got '%v': %s\n%v", q.OrderBy, queryStr, q)
		}

		// 'interval' clause
		if q.Interval != time.Second*time.Duration(10) {
			t.Errorf("Expected '10s' as duration 'interval' clause but got '%v': %s\n%v", q.Interval, queryStr, q)
		}

		// 'limit' clause
		if q.Limit != 23 {
			t.Errorf("Expected '23' as limit in 'limit' clause but got '%v': %s\n%v", q.Limit, queryStr, q)
		}

		// 'set' clause
		if q.Set[0].lString != "$foo" {
			t.Errorf("Expected '$foo' lvalue in first 'set' condition clause but got '%v': %s\n%v", q.Set[0].lString, queryStr, q)
		}
		if q.Set[0].rString != "bar" {
			t.Errorf("Expected 'bar' rvalue in first 'set' condition clause but got '%v': %s\n%v", q.Set[0].rString, queryStr, q)
		}

		if q.Set[1].lString != "$baz" {
			t.Errorf("Expected '$baz' lvalue in second 'set' condition clause but got '%v': %s\n%v", q.Set[1].lString, queryStr, q)
		}
		if q.Set[1].rString != "12" {
			t.Errorf("Expected '12' rvalue in second 'set' condition clause but got '%v': %s\n%v", q.Set[1].rString, queryStr, q)
		}

		if q.Set[2].lString != "$bay" {
			t.Errorf("Expected '$bay' lvalue in third 'set' condition clause but got '%v': %s\n%v", q.Set[2].lString, queryStr, q)
		}
		if q.Set[2].rString != "$foo" {
			t.Errorf("Expected '$foo' rvalue in third 'set' condition clause but got '%v': %s\n%v", q.Set[2].rString, queryStr, q)
		}

		if q.LogFormat != "generic" {
			t.Errorf("Expected 'generic' logformat got '%v': %s\n%v", q.LogFormat, queryStr, q)
		}
	}
}