public void TestDbConnectionStatementBuilderForBatchQuery() { // Prepare var statementBuilder = new Mock <IStatementBuilder>(); var connection = new CustomDbConnection(); // Setup statementBuilder.Setup(builder => builder.CreateBatchQuery <DataEntity>( It.IsAny <QueryBuilder <DataEntity> >(), It.IsAny <QueryGroup>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <IEnumerable <OrderField> >(), It.IsAny <string>())); // Act connection.BatchQuery <DataEntity>(0, 10, null, null, statementBuilder: statementBuilder.Object); // Assert statementBuilder.Verify(builder => builder.CreateBatchQuery <DataEntity>( It.IsAny <QueryBuilder <DataEntity> >(), It.IsAny <QueryGroup>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <IEnumerable <OrderField> >(), It.IsAny <string>()), Times.Once); }
public void TestDbConnectionTraceForAfterBatchQuery() { // Prepare var trace = new Mock <ITrace>(); var connection = new CustomDbConnection(); // Setup trace.Setup(t => t.AfterBatchQuery(It.IsAny <TraceLog>())); // Act connection.BatchQuery <TraceEntity>(0, 10, null, null, trace: trace.Object, statementBuilder: new SqlStatementBuilder()); // Assert trace.Verify(t => t.AfterBatchQuery(It.IsAny <TraceLog>()), Times.Once); }