@@ -69,8 +69,8 @@ public class UnionQueryTest extends AbstractQueryTest {
6969 protected ContentRepository createRepository () {
7070 store = new MemoryNodeStore ();
7171 qeSettings = new QueryEngineSettings ();
72- Feature sortFeature = createFeature (true );
73- qeSettings .setSortUnionQueryByScoreFeature (sortFeature );
72+ Feature sortFeature = createFeature (false );
73+ qeSettings .setSortUnionQueryLegacyModeFeature (sortFeature );
7474
7575 return new Oak (store )
7676 .with (new OpenSecurityProvider ())
@@ -439,36 +439,37 @@ public void testUnionMergingEmptyIterators() throws Exception {
439439 }
440440
441441 @ Test
442- public void testSortUnionQueryScoreFlagDisabled () throws Exception {
443- QueryEngineSettings disabledSettings = new QueryEngineSettings ();
444- Feature sortFeature = createFeature (false );
445- disabledSettings .setSortUnionQueryByScoreFeature (sortFeature );
442+ public void testSortUnionQueryLegacyModeEnabled () throws Exception {
443+ // When legacy mode is enabled, query results should be concatenated
444+ QueryEngineSettings legacySettings = new QueryEngineSettings ();
445+ Feature legacyModeFeature = createFeature (true );
446+ legacySettings .setSortUnionQueryLegacyModeFeature (legacyModeFeature );
446447 MockQueryBuilder leftQuery = new MockQueryBuilder (true )
447448 .addResult ("/left/doc1" , 0.8 )
448449 .addResult ("/left/doc2" , 0.7 );
449450 MockQueryBuilder rightQuery = new MockQueryBuilder (true )
450451 .addResult ("/right/doc1" , 0.9 )
451452 .addResult ("/right/doc2" , 0.6 );
452453
453- UnionQueryImpl unionQuery = new UnionQueryImpl (true , leftQuery .build (), rightQuery .build (), disabledSettings );
454+ UnionQueryImpl unionQuery = new UnionQueryImpl (true , leftQuery .build (), rightQuery .build (), legacySettings );
454455 List <ScoredResult > results = executeUnionAndGetScoredResults (unionQuery );
455456 assertPathOrder (results , new String []{"/left/doc1" , "/left/doc2" , "/right/doc1" , "/right/doc2" });
456457 }
457458
458459 @ Test
459- public void testSortUnionQueryScoreFlagIsNull () throws Exception {
460- QueryEngineSettings disabledSettings = new QueryEngineSettings ();
461- disabledSettings . setSortUnionQueryByScoreFeature (null );
460+ public void testSortUnionQueryLegacyModeNotSet () throws Exception {
461+ QueryEngineSettings defaultSettings = new QueryEngineSettings ();
462+ defaultSettings . setSortUnionQueryLegacyModeFeature (null );
462463 MockQueryBuilder leftQuery = new MockQueryBuilder (true )
463464 .addResult ("/left/doc1" , 0.8 )
464465 .addResult ("/left/doc2" , 0.7 );
465466 MockQueryBuilder rightQuery = new MockQueryBuilder (true )
466467 .addResult ("/right/doc1" , 0.9 )
467468 .addResult ("/right/doc2" , 0.6 );
468469
469- UnionQueryImpl unionQuery = new UnionQueryImpl (true , leftQuery .build (), rightQuery .build (), disabledSettings );
470+ UnionQueryImpl unionQuery = new UnionQueryImpl (true , leftQuery .build (), rightQuery .build (), defaultSettings );
470471 List <ScoredResult > results = executeUnionAndGetScoredResults (unionQuery );
471- assertPathOrder (results , new String []{"/left /doc1" , "/left/doc2 " , "/right/doc1 " , "/right/doc2" });
472+ assertPathOrder (results , new String []{"/right /doc1" , "/left/doc1 " , "/left/doc2 " , "/right/doc2" });
472473 }
473474
474475 @ Test
0 commit comments