@@ -57,11 +57,20 @@ public void testNeuralQueryEnricherProcessor_NeuralSparseSearch_E2EFlow() throws
57
57
getIndexNameForTest (),
58
58
Settings .builder ().put ("index.search.default_pipeline" , SPARSE_SEARCH_PIPELINE_NAME )
59
59
);
60
+ assertEquals (
61
+ search (getIndexNameForTest (), sparseEncodingQueryBuilderWithoutModelId , 1 ).get ("hits" ),
62
+ search (getIndexNameForTest (), sparseEncodingQueryBuilderWithModelId , 1 ).get ("hits" )
63
+ );
60
64
break ;
61
65
case MIXED :
62
66
sparseModelId = TestUtils .getModelId (getIngestionPipeline (SPARSE_INGEST_PIPELINE_NAME ), SPARSE_ENCODING_PROCESSOR );
63
67
loadModel (sparseModelId );
64
68
sparseEncodingQueryBuilderWithModelId .modelId (sparseModelId );
69
+
70
+ assertEquals (
71
+ search (getIndexNameForTest (), sparseEncodingQueryBuilderWithoutModelId , 1 ).get ("hits" ),
72
+ search (getIndexNameForTest (), sparseEncodingQueryBuilderWithModelId , 1 ).get ("hits" )
73
+ );
65
74
break ;
66
75
case UPGRADED :
67
76
try {
@@ -117,11 +126,6 @@ public void testNeuralQueryEnricherProcessor_NeuralSearch_E2EFlow() throws Excep
117
126
loadModel (denseModelId );
118
127
neuralQueryBuilderWithModelId .modelId (denseModelId );
119
128
120
- createSearchRequestProcessor (denseModelId , DENSE_SEARCH_PIPELINE_NAME );
121
- updateIndexSettings (
122
- getIndexNameForTest (),
123
- Settings .builder ().put ("index.search.default_pipeline" , DENSE_SEARCH_PIPELINE_NAME )
124
- );
125
129
assertEquals (
126
130
search (getIndexNameForTest (), neuralQueryBuilderWithoutModelId , 1 ).get ("hits" ),
127
131
search (getIndexNameForTest (), neuralQueryBuilderWithModelId , 1 ).get ("hits" )
0 commit comments