Package org.dbunit.database.statement

Class MockStatementFactory

java.lang.Object

org.dbunit.database.statement.MockStatementFactory

All Implemented Interfaces:

IStatementFactory

public class MockStatementFactory
extends Object
implements IStatementFactory

Mock implementation of IStatementFactory for use in unit tests. Tracks the number of statement creation calls and delegates to a configured IBatchStatement.

Use forSingleBatch(String...) to create a fully-wired factory expecting exactly one batch execution of the given SQL statements, reducing per-test setup boilerplate.

Since:

Mar 16, 2002

Version:

$Revision$

Author:

Manuel Laflamme

Constructor Summary

Constructors

Constructor	Description
MockStatementFactory()

Method Summary

Modifier and Type	Method	Description
IBatchStatement	createBatchStatement(IDatabaseConnection connection)
IPreparedBatchStatement	createPreparedBatchStatement(String sql, IDatabaseConnection connection)
static MockStatementFactory	forSingleBatch(String... expectedSql)	Creates a MockStatementFactory pre-wired with a MockBatchStatement expecting exactly the given SQL statements in a single batch execution.
MockBatchStatement	getBatchStatement()	Returns the MockBatchStatement configured via setupStatement(IBatchStatement).
void	setExpectedCreatePreparedStatementCalls(int callsCount)
void	setExpectedCreateStatementCalls(int callsCount)
void	setupStatement(IBatchStatement batchStatement)
void	verify()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

MockStatementFactory

public MockStatementFactory()

Method Details

forSingleBatch

public static MockStatementFactory forSingleBatch(String... expectedSql)

Creates a MockStatementFactory pre-wired with a MockBatchStatement expecting exactly the given SQL statements in a single batch execution. Equivalent to manually creating and configuring both a MockBatchStatement and a MockStatementFactory, but in one call.

Call getBatchStatement() on the returned factory to access the underlying MockBatchStatement for additional assertions such as MockBatchStatement.getCapturedSql().

Parameters:

expectedSql - the SQL strings expected to be batched, in order

Returns:

a fully configured factory ready for use in a test

getBatchStatement

public MockBatchStatement getBatchStatement()

Returns the MockBatchStatement configured via setupStatement(IBatchStatement). Useful after using forSingleBatch(String...) to access the underlying statement for pattern-based assertions via MockBatchStatement.getCapturedSql().

Returns:

the configured batch statement, or null if none was set

setupStatement

public void setupStatement(IBatchStatement batchStatement)

setExpectedCreateStatementCalls

public void setExpectedCreateStatementCalls(int callsCount)

setExpectedCreatePreparedStatementCalls

public void setExpectedCreatePreparedStatementCalls(int callsCount)

verify

public void verify()

createBatchStatement

public IBatchStatement createBatchStatement(IDatabaseConnection connection)
                                     throws SQLException

Specified by:

createBatchStatement in interface IStatementFactory

Throws:

SQLException

createPreparedBatchStatement

public IPreparedBatchStatement createPreparedBatchStatement(String sql,
 IDatabaseConnection connection)
                                                     throws SQLException

Specified by:

createPreparedBatchStatement in interface IStatementFactory

Throws:

SQLException