Fix pharo-project#6411

tesonep · May 29, 2020 · e4860c6 · e4860c6
1 parent 9123795
commit e4860c6
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/src/HeuristicCompletion-Model/CoASTResultSetBuilder.class.st b/src/HeuristicCompletion-Model/CoASTResultSetBuilder.class.st
@@ -176,6 +176,12 @@ CoASTResultSetBuilder >> visitSuperNode: aRBSuperNode [
 	^ self visitNode: aRBSuperNode
 ]
 
+{ #category : #visiting }
+CoASTResultSetBuilder >> visitThisContextNode: aRBThisContextNode [
+
+	^ self visitNode: aRBThisContextNode
+]
+
 { #category : #visiting }
 CoASTResultSetBuilder >> visitVariableNode: aRBVariableNode [
 

diff --git a/src/HeuristicCompletion-Tests/CoASTResultSetBuilderTest.class.st b/src/HeuristicCompletion-Tests/CoASTResultSetBuilderTest.class.st
@@ -160,6 +160,19 @@ CoASTResultSetBuilderTest >> testBuildSequenceHeuristic [
 	self assert: builder heuristic equals: #sequence
 ]
 
+{ #category : #tests }
+CoASTResultSetBuilderTest >> testBuildThisContextHeuristic [
+
+	| builder |
+	builder := CoMockASTResultSetBuilder new.
+	builder
+		completionContext: CoCompletionContext new;
+		node: RBThisContextNode new;
+		buildCompletion.
+
+	self assert: builder heuristic equals: #thisContext
+]
+
 { #category : #tests }
 CoASTResultSetBuilderTest >> testBuildVariableHeuristic [
 

diff --git a/src/HeuristicCompletion-Tests/CoMockASTResultSetBuilder.class.st b/src/HeuristicCompletion-Tests/CoMockASTResultSetBuilder.class.st
@@ -97,6 +97,13 @@ CoMockASTResultSetBuilder >> visitSequenceNode: aNode [
 	^ super visitSequenceNode: aNode
 ]
 
+{ #category : #accessing }
+CoMockASTResultSetBuilder >> visitThisContextNode: aNode [
+
+	heuristic := #thisContext.
+	^ super visitThisContextNode: aNode
+]
+
 { #category : #accessing }
 CoMockASTResultSetBuilder >> visitVariableNode: aNode [