Add source anchors to model and add test on source anchor

moosetechnology · Sep 12, 2024 · 0bdcb03 · 0bdcb03
1 parent 6eb121f
commit 0bdcb03
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/Famix-Python-Importer-Tests/FamixPythonProject1Test.class.st b/src/Famix-Python-Importer-Tests/FamixPythonProject1Test.class.st
@@ -53,6 +53,18 @@ FamixPythonProject1Test >> testModuleInSubPackage [
 	self assert: module parentPackage equals: (self packageNamed: 'subpackage1')
 ]
 
+{ #category : 'tests - packages' }
+FamixPythonProject1Test >> testPackageSourceAnchor [
+
+	| rootPackage |
+	self denyEmpty: self model allPackages.
+
+	rootPackage := self packageNamed: 'root'.
+
+	self assert: rootPackage sourceAnchor mooseModel equals: self model.
+	self assert: rootPackage sourceAnchor sourceText equals: (FamixPythonBridge parsingExamples / 'project1' / 'root' / '__init__.py') contents
+]
+
 { #category : 'tests - modules' }
 FamixPythonProject1Test >> testRootModule [
 

diff --git a/src/Famix-Python-Importer/FamixPythonImporterVisitor.class.st b/src/Famix-Python-Importer/FamixPythonImporterVisitor.class.st
@@ -610,6 +610,7 @@ FamixPythonImporterVisitor >> setSourceAnchor: aFamixEntity from: aSmaccNode [
 		fileName: (self relativeFilePath: aSmaccNode) ;
 		startPos: aSmaccNode startPosition;
 		endPos: aSmaccNode stopPosition;
+		mooseModel: self model;
 		yourself).
 
 	aFamixEntity stub: false.