com.technophobia.substeps.database.impl.SQLSubStepImplementations Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of database-substeps Show documentation
Show all versions of database-substeps Show documentation
Database step implementations
The newest version!
/*
* Copyright Technophobia Ltd & Alan Raison 2013
*
* This file is part of Substeps.
*
* Substeps is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Substeps is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Substeps. If not, see > results = Database.query(sql);
getExecutionContext().stashResultSet(results);
}
/**
* Executes the supplied sql query and store the results under the specified
* name
*
* @example ExecuteQuery {select * from wherever where
* something="something-else"} and stash results under key fred
* @section Database
* @param sql
* the sql query to run
* @param key
* the key under which the results should be stored
*/
@Step("ExecuteQuery \\{([^\\}]*)\\} and stash results under key ([^\"]*)")
public void executeQueryAndStashResultsUnderKey(final String sql, final String key) {
final List> results = Database.query(sql);
LOG.debug("query executed returning " + results.size() + " rows");
getExecutionContext().stashResultSet(key, results);
}
/**
* Checks the default stash for a row that has all of the supplied values
*
* @example AssertResults contains a row with values [COLOUR="red",
* name="bob", age=26]
* @section Database
* @param rowDetails
* a comma separated list of column names and values. String
* values should be quoted, column names should not be quoted
*/
@Step("AssertResults contains a row with values \\[([^\\]]*)\\]")
public void checkDefaultResultsContainsARow(final String rowDetails) {
checkResultsContainsARow(DatabaseExecutionContext.DEFAULT_RESULT_NAMESPACE, rowDetails);
}
/**
* Checks the named stash for a row that has all of the supplied values
*
* @param key
* the key under which the results to be checked can be found
* @example AssertResults under my-stash contains a row with values
* [COLOUR="red", name="bob", age=26]
* @section Database
* @param rowDetails
* a comma separated list of column names and values. String
* values should be quoted, column names should not be quoted
*/
@Step("AssertResults under ([^\"]*) contains a row with values \\[([^\\]]*)\\]")
public void checkResultsContainsARow(final String key, final String rowDetails) {
final Map expectedRow = convertToMap(rowDetails);
final List> resultSet = getExecutionContext().getResultSet(key);
// does the resultSet contain the values we're after ?
boolean found = false;
for (final Map resultRow : resultSet) {
// compare resultRow with expectedRow
// type conversion...?
boolean foundInThisIteration = false;
for (final Entry entry : expectedRow.entrySet()) {
final Object actualVal = resultRow.get(entry.getKey());
if (actualVal != null) {
final String stringVal = actualVal.toString();
if (entry.getValue().equalsIgnoreCase(stringVal)) {
foundInThisIteration = true;
}
} else if (entry.getValue().equalsIgnoreCase("null")) {
foundInThisIteration = true;
}
if (!foundInThisIteration) {
break;
}
}
if (foundInThisIteration) {
found = true;
break;
}
}
Assert.assertTrue("didn't find a row like: " + rowDetails, found);
}
/**
* Convert a string of the form [type="submit",value="Search"] to a map,
* minus [].
*
* TODO this came from AbstractWebDriverSubStepImplementations - refactor
* into somewhere core.
*
* @param attributes
* @return
*/
private Map convertToMap(final String attributes) {
Map attributeMap = null;
// split the attributes up, will be received as a comma separated list
// of name value pairs
final String[] nvps = attributes.split(",");
if (nvps != null) {
for (final String nvp : nvps) {
final String[] split = nvp.split("=");
if (split != null && split.length == 2) {
if (attributeMap == null) {
attributeMap = new HashMap();
}
attributeMap.put(split[0], split[1].replaceAll("\"", ""));
}
}
}
return attributeMap;
}
/**
* Checks the size of results under the default stash
*
* @example AssertResults are of size 53
* @section Database
* @param size
* the expected size of the results
*/
@Step("AssertResults are of size ([\\d])")
public void assertResultsAreOfSize(@StepParameter(converter = IntegerConverter.class) final Integer size) {
assertResultsAreOfSize(DatabaseExecutionContext.DEFAULT_RESULT_NAMESPACE, size);
}
/**
* Checks the size of results under a named stash
*
* @param key
* the key under which the results to be checked can be found
* @example AssertResults under my-stash are of size 53
* @section Database
* @param size
* the expected size of the results
*/
@Step("AssertResults under ([^\"]*) are of size ([\\d])")
public void assertResultsAreOfSize(final String key,
@StepParameter(converter = IntegerConverter.class) final Integer size) {
final List> resultSet = getExecutionContext().getResultSet(key);
if (size > 0) {
Assert.assertNotNull("mismatch of expected result size, found None, expected: " + size, resultSet);
}
Assert.assertThat("mismatch of expected result size", size, equalTo(Integer.valueOf(resultSet.size())));
}
}