public void TestDbConnectionTraceForAfterCount()
{
// Prepare
var trace = new Mock <ITrace>();
var connection = new CustomDbConnection();
// Setup
trace.Setup(t => t.AfterCount(It.IsAny <TraceLog>()));
// Act
connection.Count <TraceEntity>(trace: trace.Object, statementBuilder: new SqlStatementBuilder());
// Assert
trace.Verify(t => t.AfterCount(It.IsAny <TraceLog>()), Times.Once);
}
public void TestDbConnectionStatementBuilderForCount()
{
// Prepare
var statementBuilder = new Mock <IStatementBuilder>();
var connection = new CustomDbConnection();
// Setup
statementBuilder.Setup(builder =>
builder.CreateCount <DataEntity>(
It.IsAny <QueryBuilder <DataEntity> >(),
It.IsAny <QueryGroup>(),
It.IsAny <string>()));
// Act
connection.Count <DataEntity>(statementBuilder: statementBuilder.Object);
// Assert
statementBuilder.Verify(builder =>
builder.CreateCount <DataEntity>(
It.IsAny <QueryBuilder <DataEntity> >(),
It.IsAny <QueryGroup>(),
It.IsAny <string>()), Times.Once);
}
