Fix report lineage for graphFetchChecked (#3305)

finos · Dec 16, 2024 · 59e691b · 59e691b
1 parent 5a43b2e
commit 59e691b
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/...gend-engine-pure-code-compiled-core/src/main/resources/core/pure/lineage/scanProject.pure b/...gend-engine-pure-code-compiled-core/src/main/resources/core/pure/lineage/scanProject.pure
@@ -72,6 +72,11 @@ function <<access.private>> meta::pure::lineage::scanProject::scanProjectRecursi
                                                         | 
                                                           let tree = $fe.parametersValues->at(1)->match([s:InstanceValue[1]|$s.values, a:Any[*]|^Unknown()]);
                                                           ^Project(projectfuncEntryPoint = $fe, columns = scanRootGraphFetchTree($tree->cast(@RootGraphFetchTree<Any>)->toOne()));
+                                                      ),
+                                                      pair('meta::pure::graphFetch::execution::graphFetchChecked_T_MANY__RootGraphFetchTree_1__Checked_MANY_',
+                                                        |
+                                                          let tree = $fe.parametersValues->at(1)->match([s:InstanceValue[1]|$s.values, a:Any[*]|^Unknown()]);
+                                                          ^Project(projectfuncEntryPoint = $fe, columns = scanRootGraphFetchTree($tree->cast(@RootGraphFetchTree<Any>)->toOne()));
                                                       )
                                                     ];
                                      let funcFullPath = $fe.func->toOne()->elementToPath();

diff --git a/...-analytics-lineage-pure/src/main/resources/core_analytics_lineage/tests/lineageTests.pure b/...-analytics-lineage-pure/src/main/resources/core_analytics_lineage/tests/lineageTests.pure
@@ -562,4 +562,30 @@ function <<meta::pure::profiles::test.Test>> meta::analytics::lineage::tests::re
                                          ['Lambda', 'meta::pure::tds::TDSRow', 'meta::relational::tests::milestoning::Product', 'meta::relational::tests::milestoning::ProductClassification', 'pack_meta::pure::tds', 'pack_meta::relational::tests::milestoning'],  
                                         '[type: [ProductClassificationTable.type <JoinTreeNode>, ProductClassificationTable.type <TableAliasColumn>, ProductTable.type <JoinTreeNode>]]',
                                           $lineage);
+}
+
+
+###Pure
+import meta::pure::graphFetch::execution::*;
+import meta::relational::tests::model::simple::*;
+function <<test.Test>> meta::analytics::lineage::tests::relational::graphFetch::testForSimpleRelationalGraphFetchChecked():Boolean[1]
+{
+  let tree = #{
+    Person {
+        firstName,
+        lastName
+    }
+  }#;
+
+  let query = {|Person.all()->graphFetchChecked($tree)->serialize($tree)};
+  let mapping = meta::relational::tests::simpleRelationalMapping;
+  let runtime = meta::external::store::relational::tests::testRuntime();
+
+  let lineage = meta::analytics::lineage::computeLineage($query, $mapping, $runtime, meta::relational::extension::relationalExtensions());
+
+  meta::analytics::lineage::assertLineage(['Lambda', 'db_dbInc', 'tb_dbIncdefaultpersonTable'],
+                                          ['Lambda', 'meta::relational::tests::model::simple::Person', 'pack_meta::relational::tests::model::simple'],
+                                          '[Person.firstName: [personTable.FIRSTNAME <TableAliasColumn>], Person.lastName: [personTable.LASTNAME <TableAliasColumn>]]',
+                                           $lineage);
+
 }