Model should have a root folder

moosetechnology · Sep 12, 2024 · 6eb121f · 6eb121f
1 parent 3de52e8
commit 6eb121f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/Famix-Python-Importer-Tests/FamixPythonProject1Test.class.st b/src/Famix-Python-Importer-Tests/FamixPythonProject1Test.class.st
@@ -17,6 +17,14 @@ FamixPythonProject1Test >> model [
 	^ FamixPythonProject1TestResource current model
 ]
 
+{ #category : 'tests' }
+FamixPythonProject1Test >> testModel [
+
+	self assert: self model isNotEmpty.
+	self assert: self model name equals: 'project1'.
+	self assert: self model rootFolder equals: FamixPythonBridge parsingExamples / 'project1'
+]
+
 { #category : 'tests - modules' }
 FamixPythonProject1Test >> testModuleInRootPackage [
 

diff --git a/src/Famix-Python-Importer/FamixPythonImporter.class.st b/src/Famix-Python-Importer/FamixPythonImporter.class.st
@@ -34,11 +34,14 @@ FamixPythonImporter >> import: aFileName [
 	visitor rootFilePath: aFileName.
 
 	fileReference := aFileName asFileReference.
-	visitor model name: fileReference basename.
+	visitor model
+		name: fileReference basename;
+		rootFolder: aFileName asFileReference.
+
 
 	self importFileReference: fileReference.
 
-	^visitor model
+	^ visitor model
 ]
 
 { #category : 'api' }