src/test/java/com/google/gwtorm/schema/QueryModelTest.java

// Copyright 2014 Google Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package com.google.gwtorm.schema;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import com.google.gwtorm.data.PhoneBookDb;
import com.google.gwtorm.schema.java.JavaSchemaModel;
import com.google.gwtorm.schema.sql.DialectH2;
import com.google.gwtorm.server.OrmException;

import org.junit.Before;
import org.junit.Test;

import java.util.List;

public class QueryModelTest {

  private RelationModel people;

  @Before
  public void setUp() throws OrmException {
    JavaSchemaModel schema = new JavaSchemaModel(PhoneBookDb.class);
    people = schema.getRelation("people");
  }

  @Test
  public void testLimitWithConstant() throws OrmException {
    QueryModel qm = new QueryModel(people, null, "LIMIT 5");
    List<ColumnModel> params = qm.getParameters();
    assertNotNull(params);
    assertTrue(params.size() == 0);
    assertTrue(qm.hasLimit());

    String sql = qm.getSelectSql(new DialectH2(), "T");
    assertEquals("SELECT T.age,T.registered,T.name FROM people T LIMIT 5", sql);
  }

  @Test
  public void testLimitWithPlaceholder() throws OrmException {
    QueryModel qm = new QueryModel(people, null, "LIMIT ?");
    List<ColumnModel> params = qm.getParameters();
    assertNotNull(params);
    assertTrue(params.size() == 0);
    assertTrue(qm.hasLimit());

    String sql = qm.getSelectSql(new DialectH2(), "T");
    assertEquals("SELECT T.age,T.registered,T.name FROM people T LIMIT ?", sql);
  }
}