Request_PrimaryKeys_Name_Test.java

/**
 * 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.
 *
 * Copyright 2012-2016 the original author or authors.
 */
package org.assertj.db.type;

import org.assertj.db.common.AbstractTest;
import org.assertj.db.exception.AssertJDBException;
import org.junit.Test;

import static org.assertj.core.api.Assertions.assertThat;

/**
 * Tests on the primary keys name of {@code Request}.
 * <p>
 * These tests are on the name of the primary keys got from a {@code Request}.
 * </p>
 * 
 * @author Régis Pouiller
 * 
 */
public class Request_PrimaryKeys_Name_Test extends AbstractTest {

  /**
   * This method test the primary keys name got from a {@code Source}.
   */
  @Test
  public void test_pks_name_with_source_set_but_not_primary_keys() {
    Request request = new Request(source, 
        "SELECT actor.name, actor.firstname, movie.year, interpretation.character "
        + " FROM movie, actor, interpretation"
        + " WHERE movie.id = interpretation.id_movie"
        + " AND interpretation.id_actor = actor.id"
        + " ORDER BY actor.name, movie.year");

    assertThat(request.getPksNameList()).as("Primary Keys of the request")
        .hasSize(0);
  }

  /**
   * This method test the primary keys name got from a {@code Source}.
   */
  @Test
  public void test_pks_name_with_source_set() {
    Request request = new Request(source, 
        "SELECT actor.name, actor.firstname, movie.year, interpretation.character "
        + " FROM movie, actor, interpretation"
        + " WHERE movie.id = interpretation.id_movie"
        + " AND interpretation.id_actor = actor.id"
        + " ORDER BY actor.name, movie.year").setPksName("NAME");

    assertThat(request.getPksNameList()).as("Primary Keys of the request")
        .hasSize(1)
        .containsExactly("NAME");
  }

  /**
   * This method test the primary keys name got from a {@code DataSource}.
   */
  @Test
  public void test_pks_name_with_datasource_set() {
    Request request = new Request(dataSource, 
        "SELECT actor.name, actor.firstname, movie.year, interpretation.character "
        + " FROM movie, actor, interpretation"
        + " WHERE movie.id = interpretation.id_movie"
        + " AND interpretation.id_actor = actor.id"
        + " ORDER BY actor.name, movie.year").setPksName("NAME");

    assertThat(request.getPksNameList()).as("Primary Keys of the request")
        .hasSize(1)
        .containsExactly("NAME");
  }

  /**
   * This method test the primary keys name got from a {@code Source}.
   */
  @Test
  public void test_pks_name_with_source_and_parameters_set() {
    Request request = new Request(source, 
        "SELECT actor.name, actor.firstname, movie.year, interpretation.character "
            + " FROM movie, actor, interpretation"
            + " WHERE movie.id = interpretation.id_movie"
            + " AND interpretation.id_actor = actor.id"
            + " AND movie.year > ?"
            + " ORDER BY actor.name, movie.year", 2000).setPksName("NAME", "ID");

    assertThat(request.getPksNameList()).as("Primary Keys of the request")
        .hasSize(2)
        .containsExactly("NAME", "ID");
  }

  /**
   * This method test the primary keys name got from a {@code DataSource}.
   */
  @Test
  public void test_pks_name_with_datasource_and_parameters_set() {
    Request request = new Request(dataSource, 
        "SELECT actor.name, actor.firstname, movie.year, interpretation.character "
            + " FROM movie, actor, interpretation"
            + " WHERE movie.id = interpretation.id_movie"
            + " AND interpretation.id_actor = actor.id"
            + " AND movie.year > ?"
            + " ORDER BY actor.name, movie.year")
        .setParameters(2000).setPksName("NAME", "ID");

    assertThat(request.getPksNameList()).as("Primary Keys of the request")
        .hasSize(2)
        .containsExactly("NAME", "ID");
  }

  /**
   * This method should throw a {@code AssertJDBException} because of a {@code SQLException} caused by a table not
   * found.
   */
  @Test(expected = AssertJDBException.class)
  public void should_throw_AssertJDBException_because_SQLException_caused_by_table_not_found() {
    Table table = new Table(dataSource, "select * from interpret");
    table.getPksNameList();
  }

}