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); }