From 8bb8e87f373453aa054ea2e5f273084a97438fed Mon Sep 17 00:00:00 2001
From: Paul Yang <github@ardorful.com>
Date: Tue, 28 Nov 2023 17:11:40 -0800
Subject: [PATCH] Add snapshots for resolution DAG cases.

---
 ...ccumulate_last_2_months_metric__gbir_6.xml | 21 +++++++++++
 ...h_different_parent_time_grains__gbir_5.xml | 36 +++++++++++++++++++
 ...c_with_same_parent_time_grains__gbir_4.xml | 36 +++++++++++++++++++
 ...ics_with_different_time_grains__gbir_3.xml | 31 ++++++++++++++++
 ..._metrics_with_same_time_grains__gbir_2.xml | 31 ++++++++++++++++
 .../test_snapshot__no_metrics__gbir_0.xml     | 12 +++++++
 .../test_snapshot__simple_metric__gbir_1.xml  | 18 ++++++++++
 7 files changed, 185 insertions(+)
 create mode 100644 metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__accumulate_last_2_months_metric__gbir_6.xml
 create mode 100644 metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__derived_metric_with_different_parent_time_grains__gbir_5.xml
 create mode 100644 metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__derived_metric_with_same_parent_time_grains__gbir_4.xml
 create mode 100644 metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__metrics_with_different_time_grains__gbir_3.xml
 create mode 100644 metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__metrics_with_same_time_grains__gbir_2.xml
 create mode 100644 metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__no_metrics__gbir_0.xml
 create mode 100644 metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__simple_metric__gbir_1.xml

diff --git a/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__accumulate_last_2_months_metric__gbir_6.xml b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__accumulate_last_2_months_metric__gbir_6.xml
new file mode 100644
index 0000000000..d680a2e7d1
--- /dev/null
+++ b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__accumulate_last_2_months_metric__gbir_6.xml
@@ -0,0 +1,21 @@
+<GroupByItemResolutionDag>
+    <QueryGroupByItemResolutionNode>
+        <!-- description = Output the group-by items for query. -->
+        <!-- node_id = qr_6 -->
+        <!-- metrics_in_query =                                                     -->
+        <!--   ["MetricReference(element_name='accumulate_last_2_months_metric')"]  -->
+        <MetricGroupByItemResolutionNode>
+            <!-- description = Output group-by-items available for this metric. -->
+            <!-- node_id = mtr_11 -->
+            <!-- metric_reference =                                                 -->
+            <!--   MetricReference(element_name='accumulate_last_2_months_metric')  -->
+            <MeasureGroupByItemSourceNode>
+                <!-- description = Output group-by-items available for this measure. -->
+                <!-- node_id = msr_9 -->
+                <!-- measure_reference = MeasureReference(element_name='monthly_measure_0') -->
+                <!-- child_metric_reference =                                           -->
+                <!--   MetricReference(element_name='accumulate_last_2_months_metric')  -->
+            </MeasureGroupByItemSourceNode>
+        </MetricGroupByItemResolutionNode>
+    </QueryGroupByItemResolutionNode>
+</GroupByItemResolutionDag>
diff --git a/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__derived_metric_with_different_parent_time_grains__gbir_5.xml b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__derived_metric_with_different_parent_time_grains__gbir_5.xml
new file mode 100644
index 0000000000..5c91b5e978
--- /dev/null
+++ b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__derived_metric_with_different_parent_time_grains__gbir_5.xml
@@ -0,0 +1,36 @@
+<GroupByItemResolutionDag>
+    <QueryGroupByItemResolutionNode>
+        <!-- description = Output the group-by items for query. -->
+        <!-- node_id = qr_5 -->
+        <!-- metrics_in_query =                                                                      -->
+        <!--   ["MetricReference(element_name='derived_metric_with_different_parent_time_grains')"]  -->
+        <MetricGroupByItemResolutionNode>
+            <!-- description = Output group-by-items available for this metric. -->
+            <!-- node_id = mtr_10 -->
+            <!-- metric_reference =                                                                  -->
+            <!--   MetricReference(element_name='derived_metric_with_different_parent_time_grains')  -->
+            <MetricGroupByItemResolutionNode>
+                <!-- description = Output group-by-items available for this metric. -->
+                <!-- node_id = mtr_8 -->
+                <!-- metric_reference = MetricReference(element_name='monthly_metric_0') -->
+                <MeasureGroupByItemSourceNode>
+                    <!-- description = Output group-by-items available for this measure. -->
+                    <!-- node_id = msr_7 -->
+                    <!-- measure_reference = MeasureReference(element_name='monthly_measure_0') -->
+                    <!-- child_metric_reference = MetricReference(element_name='monthly_metric_0') -->
+                </MeasureGroupByItemSourceNode>
+            </MetricGroupByItemResolutionNode>
+            <MetricGroupByItemResolutionNode>
+                <!-- description = Output group-by-items available for this metric. -->
+                <!-- node_id = mtr_9 -->
+                <!-- metric_reference = MetricReference(element_name='yearly_metric_0') -->
+                <MeasureGroupByItemSourceNode>
+                    <!-- description = Output group-by-items available for this measure. -->
+                    <!-- node_id = msr_8 -->
+                    <!-- measure_reference = MeasureReference(element_name='yearly_measure_0') -->
+                    <!-- child_metric_reference = MetricReference(element_name='yearly_metric_0') -->
+                </MeasureGroupByItemSourceNode>
+            </MetricGroupByItemResolutionNode>
+        </MetricGroupByItemResolutionNode>
+    </QueryGroupByItemResolutionNode>
+</GroupByItemResolutionDag>
diff --git a/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__derived_metric_with_same_parent_time_grains__gbir_4.xml b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__derived_metric_with_same_parent_time_grains__gbir_4.xml
new file mode 100644
index 0000000000..18a1adc07e
--- /dev/null
+++ b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__derived_metric_with_same_parent_time_grains__gbir_4.xml
@@ -0,0 +1,36 @@
+<GroupByItemResolutionDag>
+    <QueryGroupByItemResolutionNode>
+        <!-- description = Output the group-by items for query. -->
+        <!-- node_id = qr_4 -->
+        <!-- metrics_in_query =                                                                 -->
+        <!--   ["MetricReference(element_name='derived_metric_with_same_parent_time_grains')"]  -->
+        <MetricGroupByItemResolutionNode>
+            <!-- description = Output group-by-items available for this metric. -->
+            <!-- node_id = mtr_7 -->
+            <!-- metric_reference =                                                             -->
+            <!--   MetricReference(element_name='derived_metric_with_same_parent_time_grains')  -->
+            <MetricGroupByItemResolutionNode>
+                <!-- description = Output group-by-items available for this metric. -->
+                <!-- node_id = mtr_5 -->
+                <!-- metric_reference = MetricReference(element_name='monthly_metric_0') -->
+                <MeasureGroupByItemSourceNode>
+                    <!-- description = Output group-by-items available for this measure. -->
+                    <!-- node_id = msr_5 -->
+                    <!-- measure_reference = MeasureReference(element_name='monthly_measure_0') -->
+                    <!-- child_metric_reference = MetricReference(element_name='monthly_metric_0') -->
+                </MeasureGroupByItemSourceNode>
+            </MetricGroupByItemResolutionNode>
+            <MetricGroupByItemResolutionNode>
+                <!-- description = Output group-by-items available for this metric. -->
+                <!-- node_id = mtr_6 -->
+                <!-- metric_reference = MetricReference(element_name='monthly_metric_1') -->
+                <MeasureGroupByItemSourceNode>
+                    <!-- description = Output group-by-items available for this measure. -->
+                    <!-- node_id = msr_6 -->
+                    <!-- measure_reference = MeasureReference(element_name='monthly_measure_1') -->
+                    <!-- child_metric_reference = MetricReference(element_name='monthly_metric_1') -->
+                </MeasureGroupByItemSourceNode>
+            </MetricGroupByItemResolutionNode>
+        </MetricGroupByItemResolutionNode>
+    </QueryGroupByItemResolutionNode>
+</GroupByItemResolutionDag>
diff --git a/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__metrics_with_different_time_grains__gbir_3.xml b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__metrics_with_different_time_grains__gbir_3.xml
new file mode 100644
index 0000000000..4df563dc72
--- /dev/null
+++ b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__metrics_with_different_time_grains__gbir_3.xml
@@ -0,0 +1,31 @@
+<GroupByItemResolutionDag>
+    <QueryGroupByItemResolutionNode>
+        <!-- description = Output the group-by items for query. -->
+        <!-- node_id = qr_3 -->
+        <!-- metrics_in_query =                                      -->
+        <!--   ["MetricReference(element_name='monthly_metric_0')",  -->
+        <!--    "MetricReference(element_name='yearly_metric_0')"]   -->
+        <MetricGroupByItemResolutionNode>
+            <!-- description = Output group-by-items available for this metric. -->
+            <!-- node_id = mtr_3 -->
+            <!-- metric_reference = MetricReference(element_name='monthly_metric_0') -->
+            <MeasureGroupByItemSourceNode>
+                <!-- description = Output group-by-items available for this measure. -->
+                <!-- node_id = msr_3 -->
+                <!-- measure_reference = MeasureReference(element_name='monthly_measure_0') -->
+                <!-- child_metric_reference = MetricReference(element_name='monthly_metric_0') -->
+            </MeasureGroupByItemSourceNode>
+        </MetricGroupByItemResolutionNode>
+        <MetricGroupByItemResolutionNode>
+            <!-- description = Output group-by-items available for this metric. -->
+            <!-- node_id = mtr_4 -->
+            <!-- metric_reference = MetricReference(element_name='yearly_metric_0') -->
+            <MeasureGroupByItemSourceNode>
+                <!-- description = Output group-by-items available for this measure. -->
+                <!-- node_id = msr_4 -->
+                <!-- measure_reference = MeasureReference(element_name='yearly_measure_0') -->
+                <!-- child_metric_reference = MetricReference(element_name='yearly_metric_0') -->
+            </MeasureGroupByItemSourceNode>
+        </MetricGroupByItemResolutionNode>
+    </QueryGroupByItemResolutionNode>
+</GroupByItemResolutionDag>
diff --git a/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__metrics_with_same_time_grains__gbir_2.xml b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__metrics_with_same_time_grains__gbir_2.xml
new file mode 100644
index 0000000000..8284796125
--- /dev/null
+++ b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__metrics_with_same_time_grains__gbir_2.xml
@@ -0,0 +1,31 @@
+<GroupByItemResolutionDag>
+    <QueryGroupByItemResolutionNode>
+        <!-- description = Output the group-by items for query. -->
+        <!-- node_id = qr_2 -->
+        <!-- metrics_in_query =                                      -->
+        <!--   ["MetricReference(element_name='monthly_metric_0')",  -->
+        <!--    "MetricReference(element_name='monthly_metric_1')"]  -->
+        <MetricGroupByItemResolutionNode>
+            <!-- description = Output group-by-items available for this metric. -->
+            <!-- node_id = mtr_1 -->
+            <!-- metric_reference = MetricReference(element_name='monthly_metric_0') -->
+            <MeasureGroupByItemSourceNode>
+                <!-- description = Output group-by-items available for this measure. -->
+                <!-- node_id = msr_1 -->
+                <!-- measure_reference = MeasureReference(element_name='monthly_measure_0') -->
+                <!-- child_metric_reference = MetricReference(element_name='monthly_metric_0') -->
+            </MeasureGroupByItemSourceNode>
+        </MetricGroupByItemResolutionNode>
+        <MetricGroupByItemResolutionNode>
+            <!-- description = Output group-by-items available for this metric. -->
+            <!-- node_id = mtr_2 -->
+            <!-- metric_reference = MetricReference(element_name='monthly_metric_1') -->
+            <MeasureGroupByItemSourceNode>
+                <!-- description = Output group-by-items available for this measure. -->
+                <!-- node_id = msr_2 -->
+                <!-- measure_reference = MeasureReference(element_name='monthly_measure_1') -->
+                <!-- child_metric_reference = MetricReference(element_name='monthly_metric_1') -->
+            </MeasureGroupByItemSourceNode>
+        </MetricGroupByItemResolutionNode>
+    </QueryGroupByItemResolutionNode>
+</GroupByItemResolutionDag>
diff --git a/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__no_metrics__gbir_0.xml b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__no_metrics__gbir_0.xml
new file mode 100644
index 0000000000..e5bb45aae8
--- /dev/null
+++ b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__no_metrics__gbir_0.xml
@@ -0,0 +1,12 @@
+<GroupByItemResolutionDag>
+    <QueryGroupByItemResolutionNode>
+        <!-- description = Output the group-by items for query. -->
+        <!-- node_id = qr_0 -->
+        <!-- metrics_in_query = [] -->
+        <NoMetricsGroupByItemSourceNode>
+            <!-- description =                                                           -->
+            <!--   Output the available group-by-items for a query without any metrics.  -->
+            <!-- node_id = vr_0 -->
+        </NoMetricsGroupByItemSourceNode>
+    </QueryGroupByItemResolutionNode>
+</GroupByItemResolutionDag>
diff --git a/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__simple_metric__gbir_1.xml b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__simple_metric__gbir_1.xml
new file mode 100644
index 0000000000..4ee4653ee8
--- /dev/null
+++ b/metricflow/test/snapshots/test_resolution_dags.py/GroupByItemResolutionDag/test_snapshot__simple_metric__gbir_1.xml
@@ -0,0 +1,18 @@
+<GroupByItemResolutionDag>
+    <QueryGroupByItemResolutionNode>
+        <!-- description = Output the group-by items for query. -->
+        <!-- node_id = qr_1 -->
+        <!-- metrics_in_query = ["MetricReference(element_name='monthly_metric_0')"] -->
+        <MetricGroupByItemResolutionNode>
+            <!-- description = Output group-by-items available for this metric. -->
+            <!-- node_id = mtr_0 -->
+            <!-- metric_reference = MetricReference(element_name='monthly_metric_0') -->
+            <MeasureGroupByItemSourceNode>
+                <!-- description = Output group-by-items available for this measure. -->
+                <!-- node_id = msr_0 -->
+                <!-- measure_reference = MeasureReference(element_name='monthly_measure_0') -->
+                <!-- child_metric_reference = MetricReference(element_name='monthly_metric_0') -->
+            </MeasureGroupByItemSourceNode>
+        </MetricGroupByItemResolutionNode>
+    </QueryGroupByItemResolutionNode>
+</GroupByItemResolutionDag>