82 lines
2.0 KiB
Go
82 lines
2.0 KiB
Go
package influxql_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/influxdata/influxdb/influxql"
|
|
)
|
|
|
|
func TestRewriteStatement(t *testing.T) {
|
|
tests := []struct {
|
|
stmt string
|
|
s string
|
|
}{
|
|
{
|
|
stmt: `SHOW FIELD KEYS`,
|
|
s: `SELECT fieldKey FROM _fieldKeys`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS FROM cpu`,
|
|
s: `SELECT fieldKey FROM _fieldKeys WHERE "name" = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS FROM /c.*/`,
|
|
s: `SELECT fieldKey FROM _fieldKeys WHERE "name" =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW MEASUREMENTS`,
|
|
s: `SELECT "name" FROM _measurements`,
|
|
},
|
|
{
|
|
stmt: `SHOW MEASUREMENTS WITH MEASUREMENT = cpu`,
|
|
s: `SELECT "name" FROM _measurements WHERE "name" = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW MEASUREMENTS WITH MEASUREMENT =~ /c.*/`,
|
|
s: `SELECT "name" FROM _measurements WHERE "name" =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW MEASUREMENTS WHERE region = 'uswest'`,
|
|
s: `SELECT "name" FROM _measurements WHERE region = 'uswest'`,
|
|
},
|
|
{
|
|
stmt: `SHOW MEASUREMENTS WITH MEASUREMENT = cpu WHERE region = 'uswest'`,
|
|
s: `SELECT "name" FROM _measurements WHERE "name" = 'cpu' AND region = 'uswest'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS`,
|
|
s: `SELECT tagKey FROM _tagKeys`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM cpu`,
|
|
s: `SELECT tagKey FROM _tagKeys WHERE "name" = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM /c.*/`,
|
|
s: `SELECT tagKey FROM _tagKeys WHERE "name" =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM cpu WHERE region = 'uswest'`,
|
|
s: `SELECT tagKey FROM _tagKeys WHERE "name" = 'cpu' AND region = 'uswest'`,
|
|
},
|
|
{
|
|
stmt: `SELECT value FROM cpu`,
|
|
s: `SELECT value FROM cpu`,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
stmt, err := influxql.ParseStatement(test.stmt)
|
|
if err != nil {
|
|
t.Errorf("error parsing statement: %s", err)
|
|
} else {
|
|
stmt, err = influxql.RewriteStatement(stmt)
|
|
if err != nil {
|
|
t.Errorf("error rewriting statement: %s", err)
|
|
} else if s := stmt.String(); s != test.s {
|
|
t.Errorf("error rendering string. expected %s, actual: %s", test.s, s)
|
|
}
|
|
}
|
|
}
|
|
}
|