org.tomitribe.beryllium.db.DatabaseSteps.java Source code

Introduction

Here is the source code for org.tomitribe.beryllium.db.DatabaseSteps.java

Source

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.tomitribe.beryllium.db;

import com.google.common.io.Resources;

import com.ninja_squad.dbsetup.DbSetup;
import com.ninja_squad.dbsetup.destination.Destination;
import com.ninja_squad.dbsetup.destination.DriverManagerDestination;
import com.ninja_squad.dbsetup.operation.Insert;
import com.ninja_squad.dbsetup.operation.Operation;

import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.jdbc.ScriptRunner;

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

import cucumber.api.DataTable;
import cucumber.api.java.en.Given;
import cucumber.api.java.en.Then;
import gherkin.formatter.model.DataTableRow;

import static com.google.common.truth.Truth.assertThat;
import static com.ninja_squad.dbsetup.Operations.deleteAllFrom;
import static com.ninja_squad.dbsetup.operation.CompositeOperation.sequenceOf;

public class DatabaseSteps {

    private static final Properties properties = new Properties();

    static {
        try (InputStream resource = Thread.currentThread().getContextClassLoader()
                .getResourceAsStream("test-db.properties")) {
            properties.load(resource);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private final Destination destination = new DriverManagerDestination(properties.getProperty("database.url"),
            properties.getProperty("database.user"), properties.getProperty("database.password"));

    @Given("^I have the following rows in the \"(.*?)\" table:$")
    public void iHaveTheFollowingRowsInTheTable(final String tableName, final DataTable data) throws Throwable {
        this.insert(tableName, data);
    }

    @Given("^I have only the following rows in the \"([^\"]*)\" table:$")
    public void iHaveOnlyTheFollowingRowsInTheTable(final String tableName, final DataTable data) throws Throwable {
        this.deleteAll(tableName);
        this.insert(tableName, data);
    }

    @Given("^I have no rows in the \"([^\"]*)\" table$")
    public void iHaveNoRowsInTheTable(final String tableName) throws Throwable {
        this.deleteAll(tableName);
    }

    void insert(final String tableName, final DataTable data) {
        final List<DataTableRow> rows = data.getGherkinRows();
        final List<String> columns = rows.get(0).getCells();

        final List<Operation> operations = new ArrayList<>();

        for (DataTableRow row : rows.subList(1, rows.size())) {
            final Insert.Builder builder = Insert.into(tableName);
            builder.columns(columns.toArray(new String[columns.size()]));
            builder.values(row.getCells().toArray(new String[row.getCells().size()]));
            operations.add(builder.build());
        }

        this.apply(sequenceOf(operations));
    }

    void deleteAll(final String tableName) {
        this.apply(deleteAllFrom(tableName));
    }

    void apply(final Operation operation) {
        new DbSetup(destination, operation).launch();
    }

    @Given("^I have the following sql script \"([^\"]*)\"$")
    public void iHaveTheFollowingSQLScript(final String script) throws Throwable {
        new ScriptRunner(this.destination.getConnection())
                .runScript(new BufferedReader(new FileReader(Resources.getResource(script).getPath())));
    }

    @Then("^I should have the following rows in the \"([^\"]*)\" table:$")
    public void iShouldHaveTheFollowingRowsInTheTable(final String tableName, final DataTable data)
            throws SQLException, ClassNotFoundException {
        exists(tableName, data);
    }

    void exists(final String tableName, final DataTable data) throws SQLException, ClassNotFoundException {
        final List<DataTableRow> rows = data.getGherkinRows();
        final List<String> columns = rows.get(0).getCells();
        final StringBuilder queryBuilder = new StringBuilder();
        queryBuilder.append(String.format("SELECT %s FROM %s WHERE ", StringUtils.join(columns, ","), tableName));
        queryBuilder.append(StringUtils.join(columns, " = ? AND "));
        queryBuilder.append(" = ?;");
        try (Connection conn = getConnection()) {
            PreparedStatement stmt = conn.prepareStatement(queryBuilder.toString());
            for (DataTableRow row : rows.subList(1, rows.size())) {
                List<String> rowValues = row.getCells();
                for (int i = 0; i < columns.size(); i++) {
                    stmt.setString(i + 1, rowValues.get(i));
                }
                final ResultSet rs = stmt.executeQuery();
                assertThat(rs.next()).isTrue();
            }
        }
    }

    private Connection getConnection() throws ClassNotFoundException, SQLException {
        Class.forName(properties.getProperty("database.driver"));
        return DriverManager.getConnection(properties.getProperty("database.url"),
                properties.getProperty("database.user"), properties.getProperty("database.password"));
    }
}