Optimize mapr parsing and stabilize aggregate shutdown

1 files changed, 32 insertions, 0 deletions

diff --git a/internal/mapr/parserfieldplan_test.go b/internal/mapr/parserfieldplan_test.go
new file mode 100644
index 0000000..f6d664f
--- /dev/null
+++ b/internal/mapr/parserfieldplan_test.go
@@ -0,0 +1,32 @@
+package mapr
+
+import "testing"
+
+func TestParserFieldPlan(t *testing.T) {
+	q, err := NewQuery(
+		"select count($derived) from STATS where $goroutines > 10 " +
+			"set $derived = md5sum(foo), $other = $derived group by $derived",
+	)
+	if err != nil {
+		t.Fatalf("Unable to create query: %s", err.Error())
+	}
+
+	plan := q.ParserFieldPlan()
+	if plan.AllFields {
+		t.Fatalf("Expected query-specific field plan")
+	}
+
+	requiredFields := []string{"foo", "$goroutines"}
+	for _, field := range requiredFields {
+		if !plan.Needs(field) {
+			t.Errorf("Expected field '%s' to be required", field)
+		}
+	}
+
+	notRequiredFields := []string{"$derived", "$other", "$time"}
+	for _, field := range notRequiredFields {
+		if plan.Needs(field) {
+			t.Errorf("Expected field '%s' to not be required", field)
+		}
+	}
+}