corrected one test

moosetechnology · Sep 26, 2023 · 21a3dd4 · 21a3dd4
1 parent 10677e2
commit 21a3dd4
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/Famix-UMLDocumentor-Tests/FamixUMLDocumentorTest.class.st b/src/Famix-UMLDocumentor-Tests/FamixUMLDocumentorTest.class.st
@@ -30,14 +30,12 @@ FamixUMLDocumentorTest >> testBeWithStub [
 	self assert: documentor isWithStub.
 	self assert: documentor umlEntities size equals: 2.
 
-	entity := documentor umlEntities first.
-	self assert: entity class equals: FamixUMLClass.
-	self assert: entity name equals: 'Trait3'.
+	entity := documentor umlEntities detect: [:e | (e class = FamixUMLClass) and: [e name = 'Trait3']] ifNone: [nil].
+	self assert: entity isNotNil.
 	self deny: entity isStub.
 
-	entity := documentor umlEntities second.
-	self assert: entity class equals: FamixUMLClass.
-	self assert: entity name equals: 'Trait2'.
+	entity := documentor umlEntities detect: [:e | (e class = FamixUMLClass) and: [e name = 'Trait2']] ifNone: [nil].
+	self assert: entity isNotNil.
 	self assert: entity isStub.
 
 	self deny: documentor umlEntities first color equals: documentor umlEntities second color.