code changes per code review comments

Author: NathanQingyangXu

src/integrationTest/java/com/mongodb/hibernate/query/select/SortingSelectQueryIntegrationTests.java

@@ -36,6 +36,7 @@
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.EnumSource;
+import org.junit.jupiter.params.provider.ValueSource;
 
 @DomainModel(annotatedClasses = Book.class)
 class SortingSelectQueryIntegrationTests extends AbstractSelectionQueryIntegrationTests {
@@ -188,7 +189,8 @@ void testCaseInsensitiveSortSpecNotSupported() {
                 criteria.select(root);
                 criteria.orderBy(cb.sort(root.get("title"), SortDirection.ASCENDING, NullPrecedence.NONE, true));
                 assertThatThrownBy(() -> session.createSelectionQuery(criteria).getResultList())
-                        .isInstanceOf(FeatureNotSupportedException.class);
+                        .isInstanceOf(FeatureNotSupportedException.class)
+                        .hasMessage("Case-insensitive sorting not supported");
             });
         }
     }

src/main/java/com/mongodb/hibernate/internal/translate/AbstractMqlTranslator.java

@@ -371,11 +371,8 @@ public void visitQuerySpec(QuerySpec querySpec) {
         var stages = new ArrayList<AstStage>(3);
 
         createMatchStage(querySpec).ifPresent(stages::add);
-
         createSortStage(querySpec).ifPresent(stages::add);
-
-        var projectStageSpecifications = acceptAndYield(querySpec.getSelectClause(), PROJECT_STAGE_SPECIFICATIONS);
-        stages.add(new AstProjectStage(projectStageSpecifications));
+        stages.add(createProjectStage(querySpec.getSelectClause()));
 
         astVisitorValueHolder.yield(COLLECTION_AGGREGATE, new AstAggregateCommand(collection, stages));
     }
@@ -402,6 +399,11 @@ private Optional<AstSortStage> createSortStage(QuerySpec querySpec) {
         return Optional.empty();
     }
 
+    private AstProjectStage createProjectStage(SelectClause selectClause) {
+        var projectStageSpecifications = acceptAndYield(selectClause, PROJECT_STAGE_SPECIFICATIONS);
+        return new AstProjectStage(projectStageSpecifications);
+    }
+
     @Override
     public void visitFromClause(FromClause fromClause) {
         if (fromClause.getRoots().size() != 1) {