com.smartbear.readyapi.client.teststeps.jdbcrequest.JdbcRequestTestStepBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ready-api-testserver-client Show documentation
Java client library for creating and executing test recipes against Ready!API TestServer
The newest version!
package com.smartbear.readyapi.client.teststeps.jdbcrequest;

import com.smartbear.readyapi.client.assertions.AbstractAssertionBuilder;
import com.smartbear.readyapi.client.assertions.AssertionBuilder;
import com.smartbear.readyapi.client.model.Assertion;
import com.smartbear.readyapi.client.model.JdbcRequestTestStep;
import com.smartbear.readyapi.client.teststeps.TestStepBuilder;
import com.smartbear.readyapi.client.teststeps.TestStepTypes;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;

import static com.smartbear.readyapi.client.assertions.Assertions.jdbcRequestStatusOk;
import static com.smartbear.readyapi.client.assertions.Assertions.jdbcRequestTimeout;

/**
 * Builder for JdbcRequestTestStep objects.
 */
public class JdbcRequestTestStepBuilder implements TestStepBuilder {

    private final String driver;
    private final String connectionString;
    private final boolean storedProcedure;
    private String sqlQuery;
    private List assertionBuilders = new ArrayList<>();
    private Map properties = new HashMap<>();
    private String name;

    public JdbcRequestTestStepBuilder(String driver, String connectionString, boolean storedProcedure) {
        this.driver = driver;
        this.connectionString = connectionString;
        this.storedProcedure = storedProcedure;
    }

    public JdbcRequestTestStepBuilder withSql(String sql) {
        this.sqlQuery = sql;
        return this;
    }

    public JdbcRequestTestStepBuilder named(String name) {
        this.name = name;
        return this;
    }

    public JdbcRequestTestStepBuilder withProperties(Map newProperties) {
        Objects.requireNonNull(newProperties, "Properties can't be null");
        this.properties = newProperties;
        return this;
    }

    public JdbcRequestTestStepBuilder addProperty(String propertyName, Object value) {
        properties.put(propertyName, value);
        return this;
    }

    public JdbcRequestTestStepBuilder addAssertion(AssertionBuilder assertion) {
        assertionBuilders.add(assertion);
        return this;
    }

    @Override
    public JdbcRequestTestStep build() {
        JdbcRequestTestStep testStep = new JdbcRequestTestStep();
        testStep.setType(TestStepTypes.JDBC_REQUEST.getName());
        testStep.setDriver(driver);
        testStep.setConnectionString(connectionString);
        testStep.setStoredProcedure(storedProcedure);
        testStep.setSqlQuery(sqlQuery);
        testStep.setProperties(properties);
        testStep.setName(name);
        setAssertions(testStep);
        return testStep;
    }

    private void setAssertions(JdbcRequestTestStep testStep) {
        List assertions = new ArrayList<>();
        for (AssertionBuilder assertionBuilder : assertionBuilders) {
            assertions.add(((AbstractAssertionBuilder) assertionBuilder).build());
        }
        testStep.setAssertions(assertions);
    }

    /**
     * Assertion shortcuts
     */

    public JdbcRequestTestStepBuilder assertTimeout(long timeout){
        return addAssertion( jdbcRequestTimeout( timeout ));
    }

    public JdbcRequestTestStepBuilder assertStatus(){
        return addAssertion( jdbcRequestStatusOk());
    }
}