Fix reference usages for Graph DSL

finos · Nov 19, 2024 · 85cf2d5 · 85cf2d5
1 parent 1be8c1e
commit 85cf2d5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/.../java/org/finos/legend/pure/m2/inlinedsl/graph/processor/RootGraphFetchTreeProcessor.java b/.../java/org/finos/legend/pure/m2/inlinedsl/graph/processor/RootGraphFetchTreeProcessor.java
@@ -216,12 +216,10 @@ public void populateReferenceUsages(RootGraphFetchTree<?> instance, ModelReposit
 
     private void populateReferenceUsagesForPropertyGraphFetchTrees(PropertyGraphFetchTree propertyGraphFetchTree, RootGraphFetchTree<?> mainTree, ModelRepository repository, ProcessorSupport processorSupport)
     {
-        // TODO fix this reference usage: the reference should be to propertyGraphFetchTree, not mainTree
-        this.addReferenceUsageForToOneProperty(mainTree, propertyGraphFetchTree._propertyCoreInstance(), M2GraphProperties.property, repository, processorSupport, propertyGraphFetchTree._propertyCoreInstance().getSourceInformation());
+        this.addReferenceUsageForToOneProperty(propertyGraphFetchTree, propertyGraphFetchTree._propertyCoreInstance(), M2GraphProperties.property, repository, processorSupport, propertyGraphFetchTree._propertyCoreInstance().getSourceInformation());
         if (propertyGraphFetchTree._subTypeCoreInstance() != null)
         {
-            // TODO fix this reference usage: the reference should be to propertyGraphFetchTree, not mainTree
-            this.addReferenceUsageForToOneProperty(mainTree, propertyGraphFetchTree._subTypeCoreInstance(), M2GraphProperties.subType, repository, processorSupport, propertyGraphFetchTree._subTypeCoreInstance().getSourceInformation());
+            this.addReferenceUsageForToOneProperty(propertyGraphFetchTree, propertyGraphFetchTree._subTypeCoreInstance(), M2GraphProperties.subType, repository, processorSupport, propertyGraphFetchTree._subTypeCoreInstance().getSourceInformation());
         }
         for (GraphFetchTree subTree : propertyGraphFetchTree._subTrees())
         {

diff --git a/...src/test/java/org/finos/legend/pure/m2/inlinedsl/graph/TestGraphDSLTestCodeIntegrity.java b/...src/test/java/org/finos/legend/pure/m2/inlinedsl/graph/TestGraphDSLTestCodeIntegrity.java
@@ -28,7 +28,6 @@ public static void initialize()
     }
 
     @Test
-    @Ignore
     @Override
     public void testReferenceUsages()
     {