#278 - SqlQuery setFirstRow fail - modify SQL using LIMIT/OFFSET etc to support firstRows/maxRows

src/test/java/com/avaje/tests/query/sqlquery/SqlQueryTests.java

@@ -1,5 +1,6 @@
 package com.avaje.tests.query.sqlquery;
 
+import com.avaje.ebean.BaseTestCase;
 import com.avaje.ebean.Ebean;
 import com.avaje.ebean.QueryEachConsumer;
 import com.avaje.ebean.QueryEachWhileConsumer;
@@ -7,13 +8,87 @@ import com.avaje.ebean.SqlQuery;
 import com.avaje.ebean.SqlRow;
 import com.avaje.tests.model.basic.Order;
 import com.avaje.tests.model.basic.ResetBasicData;
+import org.avaje.ebeantest.LoggedSqlCollector;
 import org.junit.Test;
 
+import java.util.List;
 import java.util.concurrent.atomic.AtomicInteger;
 
+import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.Assert.assertEquals;
 
-public class SqlQueryTests {
+public class SqlQueryTests extends BaseTestCase {
+
+  @Test
+  public void firstRowMaxRows() {
+
+    ResetBasicData.reset();
+
+    SqlQuery sqlQuery = Ebean.createSqlQuery("Select * from o_order");
+    sqlQuery.setFirstRow(3);
+    sqlQuery.setMaxRows(10);
+
+    LoggedSqlCollector.start();
+    List<SqlRow> list = sqlQuery.findList();
+
+    List<String> sql = LoggedSqlCollector.stop();
+
+    assertThat(sql.get(0)).contains("Select * from o_order limit 10 offset 3; --bind()");
+    assertThat(list).isNotEmpty();
+  }
+
+  @Test
+  public void firstRow() {
+
+    if (isPostgres()) {
+
+      ResetBasicData.reset();
+
+      SqlQuery sqlQuery = Ebean.createSqlQuery("Select * from o_order order by id");
+      sqlQuery.setFirstRow(3);
+
+      LoggedSqlCollector.start();
+      sqlQuery.findList();
+      List<String> sql = LoggedSqlCollector.stop();
+
+      assertThat(sql.get(0)).contains("Select * from o_order order by id offset 3");
+    }
+  }
+
+  @Test
+  public void maxRows() {
+
+    ResetBasicData.reset();
+
+    SqlQuery sqlQuery = Ebean.createSqlQuery("Select * from o_order order by id");
+    sqlQuery.setMaxRows(10);
+
+    LoggedSqlCollector.start();
+    sqlQuery.findList();
+    List<String> sql = LoggedSqlCollector.stop();
+
+    assertThat(sql.get(0)).contains("Select * from o_order order by id limit 10");
+  }
+
+  @Test
+  public void findEachMaxRows() {
+
+    ResetBasicData.reset();
+
+    SqlQuery sqlQuery = Ebean.createSqlQuery("Select * from o_order");
+    sqlQuery.setMaxRows(10);
+
+    LoggedSqlCollector.start();
+    sqlQuery.findEach(new QueryEachConsumer<SqlRow>() {
+      @Override
+      public void accept(SqlRow bean) {
+        bean.get("id");
+      }
+    });
+    List<String> sql = LoggedSqlCollector.stop();
+
+    assertThat(sql.get(0)).contains("limit 10");
+  }
 
   @Test
   public void findEach() {