making the mock object local

Signed-off-by: Santhosh Gandhe <[email protected]>
kkondaka · Nov 20, 2024 · 0bbd3d2 · 0bbd3d2
1 parent 19b1701
commit 0bbd3d2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...rc/test/java/org/opensearch/dataprepper/plugins/lambda/processor/LambdaProcessorTest.java b/...rc/test/java/org/opensearch/dataprepper/plugins/lambda/processor/LambdaProcessorTest.java
@@ -290,6 +290,8 @@ public void testDoExecute_UnableParseResponse(String configFileName) throws Exce
         // Arrange
         int recordCount = (int) (Math.random() * 100);
         List<Record<Event>> records = getSampleEventRecords(recordCount);
+        InvokeResponse invokeResponse = mock(InvokeResponse.class);
+
 
         // Mock Buffer to return empty payload
         when(invokeResponse.payload()).thenReturn(SdkBytes.fromUtf8String("[{\"key\": \"value\"}]"));
@@ -534,7 +536,7 @@ public void testConvertLambdaResponseToEvent_ExpectException_when_request_respon
         LambdaProcessorConfig lambdaProcessorConfig = createLambdaConfigurationFromYaml(configFile);
         LambdaProcessor localLambdaProcessor = new LambdaProcessor(pluginFactory, pluginSetting,
                 lambdaProcessorConfig, awsCredentialsSupplier, expressionEvaluator);
-
+        InvokeResponse invokeResponse = mock(InvokeResponse.class);
         // Mock LambdaResponse with a valid payload containing three events
         when(invokeResponse.payload()).thenReturn(lambdaReponse);
         when(invokeResponse.statusCode()).thenReturn(200); // Success status code