optimize assert in ut

Signed-off-by: zhichao-aws <zhichaog@amazon.com>
opensearch-project · Feb 27, 2024 · 571d4f4 · 571d4f4
1 parent 1ae0fe8
commit 571d4f4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/test/java/org/opensearch/neuralsearch/processor/NeuralQueryEnricherProcessorTests.java b/src/test/java/org/opensearch/neuralsearch/processor/NeuralQueryEnricherProcessorTests.java
@@ -50,9 +50,11 @@ public void testProcessRequest_whenVisitingEmptyQueryBody_thenSuccess() throws E
         NeuralQueryEnricherProcessor.Factory factory = new NeuralQueryEnricherProcessor.Factory();
         SearchRequest searchRequest = new SearchRequest();
         searchRequest.source(new SearchSourceBuilder());
+        assertNull(searchRequest.source().query());
         NeuralQueryEnricherProcessor processor = createTestProcessor(factory);
         SearchRequest processSearchRequest = processor.processRequest(searchRequest);
-        assert (processSearchRequest == searchRequest);
+        // should do nothing
+        assertNull(processSearchRequest.source().query());
     }
 
     public void testType() throws Exception {