org.assertj.db.api.assertions.impl.AssertionsOnTableExistence Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of assertj-db Show documentation

AssertJ-DB - Rich and fluent assertions for testing with database

There is a newer version: 2.0.2

/**
 * 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.api.assertions.impl;

import org.assertj.core.api.WritableAssertionInfo;
import org.assertj.core.internal.Failures;
import org.assertj.db.api.AbstractDbAssert;
import org.assertj.db.exception.AssertJDBException;
import org.assertj.db.type.Source;

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;

import javax.sql.DataSource;

import static org.assertj.db.error.ShouldExist.shouldExist;
import static org.assertj.db.error.ShouldNotExist.shouldNotExist;

/**
 * Implements the assertion method on the existence of a table.
 *
 * @author Avinash Ananth Narayan
 *
 */
public class AssertionsOnTableExistence {

  /**
   * To notice failures in the assertion.
   */
  private static final Failures failures = Failures.instance();

  private AssertionsOnTableExistence() {
    // Empty
  }

  public static > A exists(A assertion, WritableAssertionInfo info,
                                                      String table, Source source, DataSource dataSource) {
    try (Connection connection = getConnection(source, dataSource)) {
      DatabaseMetaData metaData = connection.getMetaData();
      ResultSet result = metaData.getTables(null, null, table, null);
      if (!result.next()) {
        throw failures.failure(info, shouldExist());
      }
      result.close();
    } catch (SQLException e) {
      throw new AssertJDBException(e);
    }
    return assertion;
  }

  public static > A doesNotExists(A assertion, WritableAssertionInfo info,
                                                             String table, Source source, DataSource dataSource) {
    try (Connection connection = getConnection(source, dataSource)) {
      DatabaseMetaData metaData = connection.getMetaData();
      ResultSet result = metaData.getTables(null, null, table, null);
      if (result.next()) {
        throw failures.failure(info, shouldNotExist());
      }
      result.close();
    } catch (SQLException e) {
      throw new AssertJDBException(e);
    }
    return assertion;
  }

  private static Connection getConnection(Source source, DataSource dataSource) throws SQLException {
    if (source == null && dataSource == null) {
      throw new NullPointerException("connection or dataSource must be not null");
    }
    if (dataSource != null) {
      return dataSource.getConnection();
    }
    return DriverManager.getConnection(source.getUrl(), source.getUser(), source.getPassword());
  }
}

Related Artifacts