// Copyright 2015 The Prometheus Authors // 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 agreed to 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 promql import ( "fmt" "math" "reflect" "strings" "testing" "time" "github.com/prometheus/common/model" "github.com/prometheus/prometheus/storage/metric" ) var testExpr = []struct { input string // The input to be parsed. expected Expr // The expected expression AST. fail bool // Whether parsing is supposed to fail. errMsg string // If not empty the parsing error has to contain this string. }{ // Scalars and scalar-to-scalar operations. { input: "1", expected: &NumberLiteral{1}, }, { input: "+Inf", expected: &NumberLiteral{model.SampleValue(math.Inf(1))}, }, { input: "-Inf", expected: &NumberLiteral{model.SampleValue(math.Inf(-1))}, }, { input: ".5", expected: &NumberLiteral{0.5}, }, { input: "5.", expected: &NumberLiteral{5}, }, { input: "123.4567", expected: &NumberLiteral{123.4567}, }, { input: "5e-3", expected: &NumberLiteral{0.005}, }, { input: "5e3", expected: &NumberLiteral{5000}, }, { input: "0xc", expected: &NumberLiteral{12}, }, { input: "0755", expected: &NumberLiteral{493}, }, { input: "+5.5e-3", expected: &NumberLiteral{0.0055}, }, { input: "-0755", expected: &NumberLiteral{-493}, }, { input: "1 + 1", expected: &BinaryExpr{itemADD, &NumberLiteral{1}, &NumberLiteral{1}, nil, false}, }, { input: "1 - 1", expected: &BinaryExpr{itemSUB, &NumberLiteral{1}, &NumberLiteral{1}, nil, false}, }, { input: "1 * 1", expected: &BinaryExpr{itemMUL, &NumberLiteral{1}, &NumberLiteral{1}, nil, false}, }, { input: "1 % 1", expected: &BinaryExpr{itemMOD, &NumberLiteral{1}, &NumberLiteral{1}, nil, false}, }, { input: "1 / 1", expected: &BinaryExpr{itemDIV, &NumberLiteral{1}, &NumberLiteral{1}, nil, false}, }, { input: "1 == bool 1", expected: &BinaryExpr{itemEQL, &NumberLiteral{1}, &NumberLiteral{1}, nil, true}, }, { input: "1 != bool 1", expected: &BinaryExpr{itemNEQ, &NumberLiteral{1}, &NumberLiteral{1}, nil, true}, }, { input: "1 > bool 1", expected: &BinaryExpr{itemGTR, &NumberLiteral{1}, &NumberLiteral{1}, nil, true}, }, { input: "1 >= bool 1", expected: &BinaryExpr{itemGTE, &NumberLiteral{1}, &NumberLiteral{1}, nil, true}, }, { input: "1 < bool 1", expected: &BinaryExpr{itemLSS, &NumberLiteral{1}, &NumberLiteral{1}, nil, true}, }, { input: "1 <= bool 1", expected: &BinaryExpr{itemLTE, &NumberLiteral{1}, &NumberLiteral{1}, nil, true}, }, { input: "+1 + -2 * 1", expected: &BinaryExpr{ Op: itemADD, LHS: &NumberLiteral{1}, RHS: &BinaryExpr{ Op: itemMUL, LHS: &NumberLiteral{-2}, RHS: &NumberLiteral{1}, }, }, }, { input: "1 + 2/(3*1)", expected: &BinaryExpr{ Op: itemADD, LHS: &NumberLiteral{1}, RHS: &BinaryExpr{ Op: itemDIV, LHS: &NumberLiteral{2}, RHS: &ParenExpr{&BinaryExpr{ Op: itemMUL, LHS: &NumberLiteral{3}, RHS: &NumberLiteral{1}, }}, }, }, }, { input: "-some_metric", expected: &UnaryExpr{ Op: itemSUB, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, }, }, { input: "+some_metric", expected: &UnaryExpr{ Op: itemADD, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, }, }, { input: "", fail: true, errMsg: "no expression found in input", }, { input: "# just a comment\n\n", fail: true, errMsg: "no expression found in input", }, { input: "1+", fail: true, errMsg: "no valid expression found", }, { input: ".", fail: true, errMsg: "unexpected character: '.'", }, { input: "2.5.", fail: true, errMsg: "could not parse remaining input \".\"...", }, { input: "100..4", fail: true, errMsg: "could not parse remaining input \".4\"...", }, { input: "0deadbeef", fail: true, errMsg: "bad number or duration syntax: \"0de\"", }, { input: "1 /", fail: true, errMsg: "no valid expression found", }, { input: "*1", fail: true, errMsg: "no valid expression found", }, { input: "(1))", fail: true, errMsg: "could not parse remaining input \")\"...", }, { input: "((1)", fail: true, errMsg: "unclosed left parenthesis", }, { input: "999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999", fail: true, errMsg: "out of range", }, { input: "(", fail: true, errMsg: "unclosed left parenthesis", }, { input: "1 and 1", fail: true, errMsg: "AND and OR not allowed in binary scalar expression", }, { input: "1 == 1", fail: true, errMsg: "parse error at char 7: comparisons between scalars must use BOOL modifier", }, { input: "1 or 1", fail: true, errMsg: "AND and OR not allowed in binary scalar expression", }, { input: "1 !~ 1", fail: true, errMsg: "could not parse remaining input \"!~ 1\"...", }, { input: "1 =~ 1", fail: true, errMsg: "could not parse remaining input \"=~ 1\"...", }, { input: `-"string"`, fail: true, errMsg: `unary expression only allowed on expressions of type scalar or vector, got "string"`, }, { input: `-test[5m]`, fail: true, errMsg: `unary expression only allowed on expressions of type scalar or vector, got "matrix"`, }, { input: `*test`, fail: true, errMsg: "no valid expression found", }, { input: "1 offset 1d", fail: true, errMsg: "offset modifier must be preceded by an instant or range selector", }, // Vector binary operations. { input: "foo * bar", expected: &BinaryExpr{ Op: itemMUL, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, VectorMatching: &VectorMatching{Card: CardOneToOne}, }, }, { input: "foo == 1", expected: &BinaryExpr{ Op: itemEQL, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &NumberLiteral{1}, }, }, { input: "foo == bool 1", expected: &BinaryExpr{ Op: itemEQL, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &NumberLiteral{1}, ReturnBool: true, }, }, { input: "2.5 / bar", expected: &BinaryExpr{ Op: itemDIV, LHS: &NumberLiteral{2.5}, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, }, }, { input: "foo and bar", expected: &BinaryExpr{ Op: itemLAND, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, VectorMatching: &VectorMatching{Card: CardManyToMany}, }, }, { input: "foo or bar", expected: &BinaryExpr{ Op: itemLOR, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, VectorMatching: &VectorMatching{Card: CardManyToMany}, }, }, { // Test and/or precedence and reassigning of operands. input: "foo + bar or bla and blub", expected: &BinaryExpr{ Op: itemLOR, LHS: &BinaryExpr{ Op: itemADD, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, VectorMatching: &VectorMatching{Card: CardOneToOne}, }, RHS: &BinaryExpr{ Op: itemLAND, LHS: &VectorSelector{ Name: "bla", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bla"}, }, }, RHS: &VectorSelector{ Name: "blub", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "blub"}, }, }, VectorMatching: &VectorMatching{Card: CardManyToMany}, }, VectorMatching: &VectorMatching{Card: CardManyToMany}, }, }, { // Test precedence and reassigning of operands. input: "bar + on(foo) bla / on(baz, buz) group_right(test) blub", expected: &BinaryExpr{ Op: itemADD, LHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, RHS: &BinaryExpr{ Op: itemDIV, LHS: &VectorSelector{ Name: "bla", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bla"}, }, }, RHS: &VectorSelector{ Name: "blub", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "blub"}, }, }, VectorMatching: &VectorMatching{ Card: CardOneToMany, On: model.LabelNames{"baz", "buz"}, Include: model.LabelNames{"test"}, }, }, VectorMatching: &VectorMatching{ Card: CardOneToOne, On: model.LabelNames{"foo"}, }, }, }, { input: "foo * on(test,blub) bar", expected: &BinaryExpr{ Op: itemMUL, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, VectorMatching: &VectorMatching{ Card: CardOneToOne, On: model.LabelNames{"test", "blub"}, }, }, }, { input: "foo and on(test,blub) bar", expected: &BinaryExpr{ Op: itemLAND, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, VectorMatching: &VectorMatching{ Card: CardManyToMany, On: model.LabelNames{"test", "blub"}, }, }, }, { input: "foo / on(test,blub) group_left(bar) bar", expected: &BinaryExpr{ Op: itemDIV, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, VectorMatching: &VectorMatching{ Card: CardManyToOne, On: model.LabelNames{"test", "blub"}, Include: model.LabelNames{"bar"}, }, }, }, { input: "foo - on(test,blub) group_right(bar,foo) bar", expected: &BinaryExpr{ Op: itemSUB, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, VectorMatching: &VectorMatching{ Card: CardOneToMany, On: model.LabelNames{"test", "blub"}, Include: model.LabelNames{"bar", "foo"}, }, }, }, { input: "foo and 1", fail: true, errMsg: "AND and OR not allowed in binary scalar expression", }, { input: "1 and foo", fail: true, errMsg: "AND and OR not allowed in binary scalar expression", }, { input: "foo or 1", fail: true, errMsg: "AND and OR not allowed in binary scalar expression", }, { input: "1 or foo", fail: true, errMsg: "AND and OR not allowed in binary scalar expression", }, { input: "1 or on(bar) foo", fail: true, errMsg: "vector matching only allowed between vectors", }, { input: "foo == on(bar) 10", fail: true, errMsg: "vector matching only allowed between vectors", }, { input: "foo and on(bar) group_left(baz) bar", fail: true, errMsg: "no grouping allowed for AND and OR operations", }, { input: "foo and on(bar) group_right(baz) bar", fail: true, errMsg: "no grouping allowed for AND and OR operations", }, { input: "foo or on(bar) group_left(baz) bar", fail: true, errMsg: "no grouping allowed for AND and OR operations", }, { input: "foo or on(bar) group_right(baz) bar", fail: true, errMsg: "no grouping allowed for AND and OR operations", }, { input: `http_requests{group="production"} / on(instance) group_left cpu_count{type="smp"}`, fail: true, errMsg: "unexpected identifier \"cpu_count\" in grouping opts, expected \"(\"", }, { input: `http_requests{group="production"} + on(instance) group_left(job,instance) cpu_count{type="smp"}`, fail: true, errMsg: "label \"instance\" must not occur in ON and INCLUDE clause at once", }, { input: "foo + bool bar", fail: true, errMsg: "bool modifier can only be used on comparison operators", }, { input: "foo + bool 10", fail: true, errMsg: "bool modifier can only be used on comparison operators", }, { input: "foo and bool 10", fail: true, errMsg: "bool modifier can only be used on comparison operators", }, // Test vector selector. { input: "foo", expected: &VectorSelector{ Name: "foo", Offset: 0, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, }, { input: "foo offset 5m", expected: &VectorSelector{ Name: "foo", Offset: 5 * time.Minute, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, }, { input: `foo:bar{a="bc"}`, expected: &VectorSelector{ Name: "foo:bar", Offset: 0, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: "a", Value: "bc"}, {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo:bar"}, }, }, }, { input: `foo{NaN='bc'}`, expected: &VectorSelector{ Name: "foo", Offset: 0, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: "NaN", Value: "bc"}, {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, }, { input: `foo{a="b", foo!="bar", test=~"test", bar!~"baz"}`, expected: &VectorSelector{ Name: "foo", Offset: 0, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: "a", Value: "b"}, {Type: metric.NotEqual, Name: "foo", Value: "bar"}, mustLabelMatcher(metric.RegexMatch, "test", "test"), mustLabelMatcher(metric.RegexNoMatch, "bar", "baz"), {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, }, { input: `{`, fail: true, errMsg: "unexpected end of input inside braces", }, { input: `}`, fail: true, errMsg: "unexpected character: '}'", }, { input: `some{`, fail: true, errMsg: "unexpected end of input inside braces", }, { input: `some}`, fail: true, errMsg: "could not parse remaining input \"}\"...", }, { input: `some_metric{a=b}`, fail: true, errMsg: "unexpected identifier \"b\" in label matching, expected string", }, { input: `some_metric{a:b="b"}`, fail: true, errMsg: "unexpected character inside braces: ':'", }, { input: `foo{a*"b"}`, fail: true, errMsg: "unexpected character inside braces: '*'", }, { input: `foo{a>="b"}`, fail: true, // TODO(fabxc): willingly lexing wrong tokens allows for more precrise error // messages from the parser - consider if this is an option. errMsg: "unexpected character inside braces: '>'", }, { input: `foo{gibberish}`, fail: true, errMsg: "expected label matching operator but got }", }, { input: `foo{1}`, fail: true, errMsg: "unexpected character inside braces: '1'", }, { input: `{}`, fail: true, errMsg: "vector selector must contain label matchers or metric name", }, { input: `{x=""}`, fail: true, errMsg: "vector selector must contain at least one non-empty matcher", }, { input: `{x=~".*"}`, fail: true, errMsg: "vector selector must contain at least one non-empty matcher", }, { input: `{x!~".+"}`, fail: true, errMsg: "vector selector must contain at least one non-empty matcher", }, { input: `{x!="a"}`, fail: true, errMsg: "vector selector must contain at least one non-empty matcher", }, { input: `foo{__name__="bar"}`, fail: true, errMsg: "metric name must not be set twice: \"foo\" or \"bar\"", // }, { // input: `:foo`, // fail: true, // errMsg: "bla", }, // Test matrix selector. { input: "test[5s]", expected: &MatrixSelector{ Name: "test", Offset: 0, Range: 5 * time.Second, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "test"}, }, }, }, { input: "test[5m]", expected: &MatrixSelector{ Name: "test", Offset: 0, Range: 5 * time.Minute, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "test"}, }, }, }, { input: "test[5h] OFFSET 5m", expected: &MatrixSelector{ Name: "test", Offset: 5 * time.Minute, Range: 5 * time.Hour, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "test"}, }, }, }, { input: "test[5d] OFFSET 10s", expected: &MatrixSelector{ Name: "test", Offset: 10 * time.Second, Range: 5 * 24 * time.Hour, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "test"}, }, }, }, { input: "test[5w] offset 2w", expected: &MatrixSelector{ Name: "test", Offset: 14 * 24 * time.Hour, Range: 5 * 7 * 24 * time.Hour, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "test"}, }, }, }, { input: `test{a="b"}[5y] OFFSET 3d`, expected: &MatrixSelector{ Name: "test", Offset: 3 * 24 * time.Hour, Range: 5 * 365 * 24 * time.Hour, LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: "a", Value: "b"}, {Type: metric.Equal, Name: model.MetricNameLabel, Value: "test"}, }, }, }, { input: `foo[5mm]`, fail: true, errMsg: "bad duration syntax: \"5mm\"", }, { input: `foo[0m]`, fail: true, errMsg: "duration must be greater than 0", }, { input: `foo[5m30s]`, fail: true, errMsg: "bad duration syntax: \"5m3\"", }, { input: `foo[5m] OFFSET 1h30m`, fail: true, errMsg: "bad number or duration syntax: \"1h3\"", }, { input: `foo["5m"]`, fail: true, }, { input: `foo[]`, fail: true, errMsg: "missing unit character in duration", }, { input: `foo[1]`, fail: true, errMsg: "missing unit character in duration", }, { input: `some_metric[5m] OFFSET 1`, fail: true, errMsg: "unexpected number \"1\" in offset, expected duration", }, { input: `some_metric[5m] OFFSET 1mm`, fail: true, errMsg: "bad number or duration syntax: \"1mm\"", }, { input: `some_metric[5m] OFFSET`, fail: true, errMsg: "unexpected end of input in offset, expected duration", }, { input: `some_metric OFFSET 1m[5m]`, fail: true, errMsg: "could not parse remaining input \"[5m]\"...", }, { input: `(foo + bar)[5m]`, fail: true, errMsg: "could not parse remaining input \"[5m]\"...", }, // Test aggregation. { input: "sum by (foo)(some_metric)", expected: &AggregateExpr{ Op: itemSum, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, Grouping: model.LabelNames{"foo"}, }, }, { input: "sum by (foo) keep_common (some_metric)", expected: &AggregateExpr{ Op: itemSum, KeepExtraLabels: true, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, Grouping: model.LabelNames{"foo"}, }, }, { input: "sum (some_metric) by (foo,bar) keep_common", expected: &AggregateExpr{ Op: itemSum, KeepExtraLabels: true, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, Grouping: model.LabelNames{"foo", "bar"}, }, }, { input: "avg by (foo)(some_metric)", expected: &AggregateExpr{ Op: itemAvg, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, Grouping: model.LabelNames{"foo"}, }, }, { input: "COUNT by (foo) keep_common (some_metric)", expected: &AggregateExpr{ Op: itemCount, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, Grouping: model.LabelNames{"foo"}, KeepExtraLabels: true, }, }, { input: "MIN (some_metric) by (foo) keep_common", expected: &AggregateExpr{ Op: itemMin, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, Grouping: model.LabelNames{"foo"}, KeepExtraLabels: true, }, }, { input: "max by (foo)(some_metric)", expected: &AggregateExpr{ Op: itemMax, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, Grouping: model.LabelNames{"foo"}, }, }, { input: "stddev(some_metric)", expected: &AggregateExpr{ Op: itemStddev, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, }, }, { input: "stdvar by (foo)(some_metric)", expected: &AggregateExpr{ Op: itemStdvar, Expr: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, Grouping: model.LabelNames{"foo"}, }, }, { input: `sum some_metric by (test)`, fail: true, errMsg: "unexpected identifier \"some_metric\" in aggregation, expected \"(\"", }, { input: `sum (some_metric) by test`, fail: true, errMsg: "unexpected identifier \"test\" in grouping opts, expected \"(\"", }, { input: `sum (some_metric) by ()`, fail: true, errMsg: "unexpected \")\" in grouping opts, expected identifier", }, { input: `sum (some_metric) by test`, fail: true, errMsg: "unexpected identifier \"test\" in grouping opts, expected \"(\"", }, { input: `sum () by (test)`, fail: true, errMsg: "no valid expression found", }, { input: "MIN keep_common (some_metric) by (foo)", fail: true, errMsg: "could not parse remaining input \"by (foo)\"...", }, { input: "MIN by(test) (some_metric) keep_common", fail: true, errMsg: "could not parse remaining input \"keep_common\"...", }, // Test function calls. { input: "time()", expected: &Call{ Func: mustGetFunction("time"), }, }, { input: `floor(some_metric{foo!="bar"})`, expected: &Call{ Func: mustGetFunction("floor"), Args: Expressions{ &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.NotEqual, Name: "foo", Value: "bar"}, {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, }, }, }, { input: "rate(some_metric[5m])", expected: &Call{ Func: mustGetFunction("rate"), Args: Expressions{ &MatrixSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, Range: 5 * time.Minute, }, }, }, }, { input: "round(some_metric)", expected: &Call{ Func: mustGetFunction("round"), Args: Expressions{ &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, }, }, }, { input: "round(some_metric, 5)", expected: &Call{ Func: mustGetFunction("round"), Args: Expressions{ &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, &NumberLiteral{5}, }, }, }, { input: "floor()", fail: true, errMsg: "expected at least 1 argument(s) in call to \"floor\", got 0", }, { input: "floor(some_metric, other_metric)", fail: true, errMsg: "expected at most 1 argument(s) in call to \"floor\", got 2", }, { input: "floor(1)", fail: true, errMsg: "expected type vector in call to function \"floor\", got scalar", }, { input: "non_existant_function_far_bar()", fail: true, errMsg: "unknown function with name \"non_existant_function_far_bar\"", }, { input: "rate(some_metric)", fail: true, errMsg: "expected type matrix in call to function \"rate\", got vector", }, // Fuzzing regression tests. { input: "-=", fail: true, errMsg: `no valid expression found`, }, { input: "++-++-+-+-<", fail: true, errMsg: `no valid expression found`, }, { input: "e-+=/(0)", fail: true, errMsg: `no valid expression found`, }, { input: "-If", fail: true, errMsg: `no valid expression found`, }, // String quoting and escape sequence interpretation tests. { input: `"double-quoted string \" with escaped quote"`, expected: &StringLiteral{ Val: "double-quoted string \" with escaped quote", }, }, { input: `'single-quoted string \' with escaped quote'`, expected: &StringLiteral{ Val: "single-quoted string ' with escaped quote", }, }, { input: "`backtick-quoted string`", expected: &StringLiteral{ Val: "backtick-quoted string", }, }, { input: `"\a\b\f\n\r\t\v\\\" - \xFF\377\u1234\U00010111\U0001011111☺"`, expected: &StringLiteral{ Val: "\a\b\f\n\r\t\v\\\" - \xFF\377\u1234\U00010111\U0001011111☺", }, }, { input: `'\a\b\f\n\r\t\v\\\' - \xFF\377\u1234\U00010111\U0001011111☺'`, expected: &StringLiteral{ Val: "\a\b\f\n\r\t\v\\' - \xFF\377\u1234\U00010111\U0001011111☺", }, }, { input: "`" + `\a\b\f\n\r\t\v\\\"\' - \xFF\377\u1234\U00010111\U0001011111☺` + "`", expected: &StringLiteral{ Val: `\a\b\f\n\r\t\v\\\"\' - \xFF\377\u1234\U00010111\U0001011111☺`, }, }, { input: "`\\``", fail: true, errMsg: "could not parse remaining input", }, { input: `"\`, fail: true, errMsg: "escape sequence not terminated", }, { input: `"\c"`, fail: true, errMsg: "unknown escape sequence U+0063 'c'", }, { input: `"\x."`, fail: true, errMsg: "illegal character U+002E '.' in escape sequence", }, } func TestParseExpressions(t *testing.T) { for _, test := range testExpr { parser := newParser(test.input) expr, err := parser.parseExpr() // Unexpected errors are always caused by a bug. if err == errUnexpected { t.Fatalf("unexpected error occurred") } if !test.fail && err != nil { t.Errorf("error in input '%s'", test.input) t.Fatalf("could not parse: %s", err) } if test.fail && err != nil { if !strings.Contains(err.Error(), test.errMsg) { t.Errorf("unexpected error on input '%s'", test.input) t.Fatalf("expected error to contain %q but got %q", test.errMsg, err) } continue } err = parser.typecheck(expr) if !test.fail && err != nil { t.Errorf("error on input '%s'", test.input) t.Fatalf("typecheck failed: %s", err) } if test.fail { if err != nil { if !strings.Contains(err.Error(), test.errMsg) { t.Errorf("unexpected error on input '%s'", test.input) t.Fatalf("expected error to contain %q but got %q", test.errMsg, err) } continue } t.Errorf("error on input '%s'", test.input) t.Fatalf("failure expected, but passed with result: %q", expr) } if !reflect.DeepEqual(expr, test.expected) { t.Errorf("error on input '%s'", test.input) t.Fatalf("no match\n\nexpected:\n%s\ngot: \n%s\n", Tree(test.expected), Tree(expr)) } } } // NaN has no equality. Thus, we need a separate test for it. func TestNaNExpression(t *testing.T) { parser := newParser("NaN") expr, err := parser.parseExpr() if err != nil { t.Errorf("error on input 'NaN'") t.Fatalf("coud not parse: %s", err) } nl, ok := expr.(*NumberLiteral) if !ok { t.Errorf("error on input 'NaN'") t.Fatalf("expected number literal but got %T", expr) } if !math.IsNaN(float64(nl.Val)) { t.Errorf("error on input 'NaN'") t.Fatalf("expected 'NaN' in number literal but got %v", nl.Val) } } var testStatement = []struct { input string expected Statements fail bool }{ { // Test a file-like input. input: ` # A simple test recording rule. dc:http_request:rate5m = sum(rate(http_request_count[5m])) by (dc) # A simple test alerting rule. ALERT GlobalRequestRateLow IF(dc:http_request:rate5m < 10000) FOR 5m LABELS { service = "testservice" # ... more fields here ... } ANNOTATIONS { summary = "Global request rate low", description = "The global request rate is low" } foo = bar{label1="value1"} ALERT BazAlert IF foo > 10 ANNOTATIONS { description = "BazAlert", runbook = "http://my.url", summary = "Baz", } `, expected: Statements{ &RecordStmt{ Name: "dc:http_request:rate5m", Expr: &AggregateExpr{ Op: itemSum, Grouping: model.LabelNames{"dc"}, Expr: &Call{ Func: mustGetFunction("rate"), Args: Expressions{ &MatrixSelector{ Name: "http_request_count", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "http_request_count"}, }, Range: 5 * time.Minute, }, }, }, }, Labels: nil, }, &AlertStmt{ Name: "GlobalRequestRateLow", Expr: &ParenExpr{&BinaryExpr{ Op: itemLSS, LHS: &VectorSelector{ Name: "dc:http_request:rate5m", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "dc:http_request:rate5m"}, }, }, RHS: &NumberLiteral{10000}, }}, Labels: model.LabelSet{"service": "testservice"}, Duration: 5 * time.Minute, Annotations: model.LabelSet{ "summary": "Global request rate low", "description": "The global request rate is low", }, }, &RecordStmt{ Name: "foo", Expr: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: "label1", Value: "value1"}, {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, Labels: nil, }, &AlertStmt{ Name: "BazAlert", Expr: &BinaryExpr{ Op: itemGTR, LHS: &VectorSelector{ Name: "foo", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "foo"}, }, }, RHS: &NumberLiteral{10}, }, Labels: model.LabelSet{}, Annotations: model.LabelSet{ "summary": "Baz", "description": "BazAlert", "runbook": "http://my.url", }, }, }, }, { input: `foo{x="", a="z"} = bar{a="b", x=~"y"}`, expected: Statements{ &RecordStmt{ Name: "foo", Expr: &VectorSelector{ Name: "bar", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: "a", Value: "b"}, mustLabelMatcher(metric.RegexMatch, "x", "y"), {Type: metric.Equal, Name: model.MetricNameLabel, Value: "bar"}, }, }, Labels: model.LabelSet{"x": "", "a": "z"}, }, }, }, { input: `ALERT SomeName IF some_metric > 1 WITH {} SUMMARY "Global request rate low" DESCRIPTION "The global request rate is low" `, expected: Statements{ &AlertStmt{ Name: "SomeName", Expr: &BinaryExpr{ Op: itemGTR, LHS: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, RHS: &NumberLiteral{1}, }, Labels: model.LabelSet{}, Annotations: model.LabelSet{ "summary": "Global request rate low", "description": "The global request rate is low", }, }, }, }, { input: `ALERT SomeName IF some_metric > 1 LABELS {} ANNOTATIONS { summary = "Global request rate low", description = "The global request rate is low", } `, expected: Statements{ &AlertStmt{ Name: "SomeName", Expr: &BinaryExpr{ Op: itemGTR, LHS: &VectorSelector{ Name: "some_metric", LabelMatchers: metric.LabelMatchers{ {Type: metric.Equal, Name: model.MetricNameLabel, Value: "some_metric"}, }, }, RHS: &NumberLiteral{1}, }, Labels: model.LabelSet{}, Annotations: model.LabelSet{ "summary": "Global request rate low", "description": "The global request rate is low", }, }, }, }, { input: ` # A simple test alerting rule. ALERT GlobalRequestRateLow IF(dc:http_request:rate5m < 10000) FOR 5 LABELS { service = "testservice" # ... more fields here ... } ANNOTATIONS { summary = "Global request rate low" description = "The global request rate is low" } `, fail: true, }, { input: "", expected: Statements{}, }, { input: "foo = time()", expected: Statements{ &RecordStmt{ Name: "foo", Expr: &Call{Func: mustGetFunction("time")}, Labels: nil, }}, }, { input: "foo = 1", expected: Statements{ &RecordStmt{ Name: "foo", Expr: &NumberLiteral{1}, Labels: nil, }}, }, { input: "foo = bar[5m]", fail: true, }, { input: `foo = "test"`, fail: true, }, { input: `foo = `, fail: true, }, { input: `foo{a!="b"} = bar`, fail: true, }, { input: `foo{a=~"b"} = bar`, fail: true, }, { input: `foo{a!~"b"} = bar`, fail: true, }, // Fuzzing regression tests. { input: `I=-/`, fail: true, }, { input: `I=3E8/-=`, fail: true, }, { input: `M=-=-0-0`, fail: true, }, } func TestParseStatements(t *testing.T) { for _, test := range testStatement { parser := newParser(test.input) stmts, err := parser.parseStmts() // Unexpected errors are always caused by a bug. if err == errUnexpected { t.Fatalf("unexpected error occurred") } if !test.fail && err != nil { t.Errorf("error in input: \n\n%s\n", test.input) t.Fatalf("could not parse: %s", err) } if test.fail && err != nil { continue } err = parser.typecheck(stmts) if !test.fail && err != nil { t.Errorf("error in input: \n\n%s\n", test.input) t.Fatalf("typecheck failed: %s", err) } if test.fail { if err != nil { continue } t.Errorf("error in input: \n\n%s\n", test.input) t.Fatalf("failure expected, but passed") } if !reflect.DeepEqual(stmts, test.expected) { t.Errorf("error in input: \n\n%s\n", test.input) t.Fatalf("no match\n\nexpected:\n%s\ngot: \n%s\n", Tree(test.expected), Tree(stmts)) } } } func mustLabelMatcher(mt metric.MatchType, name model.LabelName, val model.LabelValue) *metric.LabelMatcher { m, err := metric.NewLabelMatcher(mt, name, val) if err != nil { panic(err) } return m } func mustGetFunction(name string) *Function { f, ok := getFunction(name) if !ok { panic(fmt.Errorf("function %q does not exist", name)) } return f } var testSeries = []struct { input string expectedMetric model.Metric expectedValues []sequenceValue fail bool }{ { input: `{} 1 2 3`, expectedMetric: model.Metric{}, expectedValues: newSeq(1, 2, 3), }, { input: `{a="b"} -1 2 3`, expectedMetric: model.Metric{ "a": "b", }, expectedValues: newSeq(-1, 2, 3), }, { input: `my_metric 1 2 3`, expectedMetric: model.Metric{ model.MetricNameLabel: "my_metric", }, expectedValues: newSeq(1, 2, 3), }, { input: `my_metric{} 1 2 3`, expectedMetric: model.Metric{ model.MetricNameLabel: "my_metric", }, expectedValues: newSeq(1, 2, 3), }, { input: `my_metric{a="b"} 1 2 3`, expectedMetric: model.Metric{ model.MetricNameLabel: "my_metric", "a": "b", }, expectedValues: newSeq(1, 2, 3), }, { input: `my_metric{a="b"} 1 2 3-10x4`, expectedMetric: model.Metric{ model.MetricNameLabel: "my_metric", "a": "b", }, expectedValues: newSeq(1, 2, 3, -7, -17, -27, -37), }, { input: `my_metric{a="b"} 1 2 3-0x4`, expectedMetric: model.Metric{ model.MetricNameLabel: "my_metric", "a": "b", }, expectedValues: newSeq(1, 2, 3, 3, 3, 3, 3), }, { input: `my_metric{a="b"} 1 3 _ 5 _x4`, expectedMetric: model.Metric{ model.MetricNameLabel: "my_metric", "a": "b", }, expectedValues: newSeq(1, 3, none, 5, none, none, none, none), }, { input: `my_metric{a="b"} 1 3 _ 5 _a4`, fail: true, }, } // For these tests only, we use the smallest float64 to signal an omitted value. const none = math.SmallestNonzeroFloat64 func newSeq(vals ...float64) (res []sequenceValue) { for _, v := range vals { if v == none { res = append(res, sequenceValue{omitted: true}) } else { res = append(res, sequenceValue{value: model.SampleValue(v)}) } } return res } func TestParseSeries(t *testing.T) { for _, test := range testSeries { parser := newParser(test.input) parser.lex.seriesDesc = true metric, vals, err := parser.parseSeriesDesc() // Unexpected errors are always caused by a bug. if err == errUnexpected { t.Fatalf("unexpected error occurred") } if !test.fail && err != nil { t.Errorf("error in input: \n\n%s\n", test.input) t.Fatalf("could not parse: %s", err) } if test.fail && err != nil { continue } if test.fail { if err != nil { continue } t.Errorf("error in input: \n\n%s\n", test.input) t.Fatalf("failure expected, but passed") } if !reflect.DeepEqual(vals, test.expectedValues) || !reflect.DeepEqual(metric, test.expectedMetric) { t.Errorf("error in input: \n\n%s\n", test.input) t.Fatalf("no match\n\nexpected:\n%s %s\ngot: \n%s %s\n", test.expectedMetric, test.expectedValues, metric, vals) } } } func TestRecoverParserRuntime(t *testing.T) { var p *parser var err error defer p.recover(&err) // Cause a runtime panic. var a []int a[123] = 1 if err != errUnexpected { t.Fatalf("wrong error message: %q, expected %q", err, errUnexpected) } } func TestRecoverParserError(t *testing.T) { var p *parser var err error e := fmt.Errorf("custom error") defer func() { if err.Error() != e.Error() { t.Fatalf("wrong error message: %q, expected %q", err, e) } }() defer p.recover(&err) panic(e) }