add test export attribute

moosetechnology · Jun 4, 2024 · e10fcbd · e10fcbd
1 parent 25488cb
commit e10fcbd
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 4 deletions.
diff --git a/src/Famix2Java-Tests/FAMIX2JavaVisitorTest.class.st b/src/Famix2Java-Tests/FAMIX2JavaVisitorTest.class.st
@@ -918,6 +918,32 @@ FAMIX2JavaVisitorTest >> testVisitParametricClassWithTwoParameters [
 }'
 ]
 
+{ #category : #test }
+FAMIX2JavaVisitorTest >> testVisitPrivateAttribute [
+
+	| attribute attributeClass attributeDeclaredType sourceAnchor model resultString |
+	model := FamixJavaModel new.
+	attributeDeclaredType := FamixJavaClass new.
+	attributeDeclaredType name: 'AttributeClass'.
+	attribute := FamixJavaAttribute new.
+	attribute name: 'anAttribute'.
+	attribute isPrivate: true.
+	attribute declaredType: attributeDeclaredType. 
+	sourceAnchor := FamixJavaSourceTextAnchor new.
+	sourceAnchor source: ''.
+	attribute sourceAnchor: sourceAnchor.
+	attributeClass := FamixJavaClass new.
+	attributeClass addAttribute: attribute.
+	model addAll: { 
+			attribute.
+			attributeClass.
+			sourceAnchor }.
+	resultString := String streamContents: [ :stream | 
+		                visitor currentStream: stream.
+		                attribute accept: visitor ].
+	self assert: resultString equals: 'private AttributeClass anAttribute;'
+]
+
 { #category : #test }
 FAMIX2JavaVisitorTest >> testVisitPrivateMethod [
 

diff --git a/src/Famix2Java/FAMIX2JavaVisitor.class.st b/src/Famix2Java/FAMIX2JavaVisitor.class.st
@@ -313,10 +313,10 @@ FAMIX2JavaVisitor >> visitAttribute: aAttribute [
 
 	"Printing modifiers"
 	self indent.
-	"modifiers := aAttribute cacheAt: #modifiers ifAbsent: [ aAttribute modifiers ].
-	modifiers isString
-		ifTrue: [ self <<< modifiers ]
-		ifFalse: [ modifiers do: [ :modifier | self <<< modifier ] separatedBy: [ self space ] ]."
+	aAttribute isPrivate ifTrue: [ self << 'private ' ].
+	aAttribute isProtected ifTrue: [ self << 'protected ' ].
+	aAttribute isPublic ifTrue: [ self << 'public ' ].
+	aAttribute isClassSide ifTrue: [ self << 'static ' ].
 	"Declared Type"
 	aAttribute declaredType
 		ifNotNil: [ self printDeclaredType: aAttribute declaredType ]