Merge pull request #746 from moosetechnology/745-add-withoutStubs-to-…

…MooseModel Fix #745: Add #withoutStubs to MooseAbstractGroup
moosetechnology · Mar 29, 2024 · eec9d16 · eec9d16
2 parents f4794eb + 8e5a254
commit eec9d16
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 0 deletions.
diff --git a/src/Moose-Core-Tests/MooseAbstractGroupTest.class.st b/src/Moose-Core-Tests/MooseAbstractGroupTest.class.st
@@ -494,6 +494,29 @@ MooseAbstractGroupTest >> testWithIndexDo [
 	self assertCollection: t hasSameElements: {{el1 . 1} . {el2 . 2} . {el3 . 3}}
 ]
 
+{ #category : #tests }
+MooseAbstractGroupTest >> testWithoutStubs [
+
+	| stubEntity1 stubEntity2 |
+
+	stubEntity1 := MooseEntity new
+		               isStub: true;
+		               yourself.
+	stubEntity2 := MooseEntity new
+		               isStub: true;
+		               yourself.
+	group
+		add: MooseEntity new;
+		add: MooseEntity new;
+		add: stubEntity1;
+		add: stubEntity2.
+
+	self assert: group size equals: 4.
+	self assert: group withoutStubs isMooseObject.
+	self assert: group withoutStubs size equals: 2.
+	group withoutStubs do: [ :entity | self deny: entity isStub ]
+]
+
 { #category : #utilities }
 MooseAbstractGroupTest >> twoClasses [
 	| classA classB |

diff --git a/src/Moose-Core/MooseAbstractGroup.class.st b/src/Moose-Core/MooseAbstractGroup.class.st
@@ -786,6 +786,12 @@ MooseAbstractGroup >> withIndexDo: aBlockClosure [
 	self entities withIndexDo: aBlockClosure 
 ]
 
+{ #category : #accessing }
+MooseAbstractGroup >> withoutStubs [
+
+	^ (self reject: #isStub) asMooseGroup
+]
+
 { #category : #'set operations' }
 MooseAbstractGroup >> | aCollection [
 	^ self union: aCollection

diff --git a/src/Moose-Core/MooseObject.class.st b/src/Moose-Core/MooseObject.class.st
@@ -358,6 +358,11 @@ MooseObject >> isMooseModel [
 	^ false
 ]
 
+{ #category : #testing }
+MooseObject >> isMooseObject [
+	^ true
+]
+
 { #category : #testing }
 MooseObject >> isOfType: aClassFAMIX [
 	^ self class isOfType: aClassFAMIX

diff --git a/src/Moose-Core/Object.extension.st b/src/Moose-Core/Object.extension.st
@@ -29,6 +29,11 @@ Object >> isMooseEntity [
 	^ false 
 ]
 
+{ #category : #'*Moose-Core' }
+Object >> isMooseObject [
+	^ false
+]
+
 { #category : #'*Moose-Core' }
 Object class >> isOfType: aClassFAMIX [
 	^ false