test: refactor usage of the editor content in the chat test

Author: nickchecan

com.developer.nefarious.zjoule.test/src/com/developer/nefarious/zjoule/test/chat/ChatOrchestratorTest.java

@@ -68,8 +68,9 @@ public void shouldNotStopIfThereIsNoMessageHistory() throws IOException, Interru
 
 		List<IChatMessage> mockMessageHistory = new ArrayList<>();
 		when(mockAIClient.getMessageHistory()).thenReturn(mockMessageHistory);
-
-		mockedStaticEditorContentReader.when(EditorContentReader::readActiveEditorContent).thenReturn(null);
+		
+		String mockEditorContent = null;
+		mockedStaticEditorContentReader.when(EditorContentReader::readActiveEditorContent).thenReturn(mockEditorContent);
 
 		IChatMessage mockSystemMessage = mock(IChatMessage.class);
 		when(mockAIClient.createMessage(Role.SYSTEM, mockBaseInstructions)).thenReturn(mockSystemMessage);
@@ -84,7 +85,7 @@ public void shouldNotStopIfThereIsNoMessageHistory() throws IOException, Interru
 		when(mockAnswer.getMessage()).thenReturn(expectedValue);
 
 		// Act
-		String returnValue = cut.getAnswer(mockUserPrompt);
+		String returnValue = cut.getAnswer(mockUserPrompt, mockEditorContent);
 
 		// Assert
 		verify(mockAIClient).setMessageHistory(mockAllMessages);
@@ -125,7 +126,7 @@ public void shouldPlumbAnswer() throws IOException, InterruptedException {
 		when(mockAnswer.getMessage()).thenReturn(expectedValue);
 
 		// Act
-		String returnValue = cut.getAnswer(mockUserPrompt);
+		String returnValue = cut.getAnswer(mockUserPrompt, mockEditorContent);
 
 		// Assert
 		verify(mockAIClient).setMessageHistory(mockAllMessages);
@@ -145,7 +146,8 @@ public void shouldPlumbErrorMessageIfChatCompletionDoesntWork() throws IOExcepti
 		List<IChatMessage> mockMessageHistory = Arrays.asList(mockOldMessage);
 		when(mockAIClient.getMessageHistory()).thenReturn(mockMessageHistory);
 
-		mockedStaticEditorContentReader.when(EditorContentReader::readActiveEditorContent).thenReturn(null);
+		String mockEditorContent = null;
+		mockedStaticEditorContentReader.when(EditorContentReader::readActiveEditorContent).thenReturn(mockEditorContent);
 
 		IChatMessage mockSystemMessage = mock(IChatMessage.class);
 		when(mockAIClient.createMessage(Role.SYSTEM, mockBaseInstructions)).thenReturn(mockSystemMessage);
@@ -156,7 +158,7 @@ public void shouldPlumbErrorMessageIfChatCompletionDoesntWork() throws IOExcepti
 		String expectedValue = "Error during the AI request execution";
 
 		// Act
-		String returnValue = cut.getAnswer(mockUserPrompt);
+		String returnValue = cut.getAnswer(mockUserPrompt, mockEditorContent);
 
 		// Assert
 		verify(mockAIClient, never()).setMessageHistory(any());
@@ -170,11 +172,14 @@ public void shouldReturnErrorIfModelIsIncompatible() {
 		String expectedValue = "The model you have selected is incompatible with the current "
 				+ "operation. Please verify the model's configuration or choose a "
 				+ "compatible alternative.";
+		
+		String mockEditorContent = null;
+		mockedStaticEditorContentReader.when(EditorContentReader::readActiveEditorContent).thenReturn(mockEditorContent);
 
 		mockedStaticAIClientFactory.when(AIClientFactory::getClient).thenReturn(null);
 
 		// Act
-		String returnValue = cut.getAnswer(mockUserPrompt);
+		String returnValue = cut.getAnswer(mockUserPrompt, mockEditorContent);
 
 		// Assert
 		assertEquals(expectedValue, returnValue);