Function TestInlineIncludes

expr/include_test.go:59–156

(t *testing.T)

}

func TestInlineIncludes(t *testing.T) {

	t1 := time.Now()

	readers := []expr.ContextReader{
		datasource.NewContextMap(map[string]interface{}{
			"name":       "bob",
			"city":       "Peoria, IL",
			"zip":        5,
			"signedup":   t1,
			"lastevent":  map[string]time.Time{"signedup": t1},
			"last.event": map[string]time.Time{"has.period": t1},
		}, true),
	}

	nc := datasource.NewNestedContextReader(readers, time.Now())
	includerCtx := newIncluderCtx(nc, `
		FILTER name == "Yoda" ALIAS is_yoda_true;
		FILTER AND (
			planet == "Dagobah"
			INCLUDE is_yoda_true
		) ALIAS nested_includes_yoda;
	`)

	tests := []incTest{
		{
			in:  `lastvisit_ts < "now-1d"`,
			out: `lastvisit_ts < "now-1d"`,
		},
		{
			in:  `AND ( lastvisit_ts < "now-1d", INCLUDE is_yoda_true )`,
			out: `AND ( lastvisit_ts < "now-1d", name == "Yoda" )`,
		},
		{
			in:  `AND ( lastvisit_ts < "now-2d", NOT INCLUDE is_yoda_true )`,
			out: `AND ( lastvisit_ts < "now-2d", NOT (name == "Yoda") )`,
		},
		{
			in:  `AND ( lastvisit_ts < "now-3d", NOT INCLUDE nested_includes_yoda )`,
			out: `AND ( lastvisit_ts < "now-3d", NOT AND ( planet == "Dagobah", (name == "Yoda") ) )`,
		},
	}
	tests = []incTest{
		{
			in:  `AND ( lastvisit_ts < "now-3d", NOT INCLUDE nested_includes_yoda )`,
			out: `AND ( lastvisit_ts < "now-3d", NOT AND ( planet == "Dagobah", name == "Yoda" ) )`,
		},
	}
	for _, tc := range tests {
		n := expr.MustParse(tc.in)
		out, err := expr.InlineIncludes(includerCtx, n)
		assert.Equal(t, nil, err)
		assert.NotEqual(t, nil, out)
		if out != nil {
			assert.Equal(t, tc.out, out.String())
		}
	}