是否可以模拟数据库事务参数?

时间:2009-07-29 18:34:02

标签: c# unit-testing mocking moq transactions

我正在尝试对接口的实现进行单元测试,并且我在成功模拟其中一个接口方法的SqlTransaction参数时遇到了一些困难。

以下是我感兴趣的界面和测试方法......

public class MyInterface 
{ 
  void MyMethod(SqlTransaction SharedTransaction, DateTime EventTime);
} 

public class MyImplementation : MyInterface
{
  public void MyMethod(SqlTransaction SharedTransaction, DateTime EventTime)
  {
    DateTime dtLastEventTime = DateTime.MinValue;
    using(SqlCommand comm = SharedTransaction.Connection.CreateCommand())
    {
      comm.CommandText = SQL_GET_LAST_EVENTTIME_DEFINED_ELSEWHERE;
      comm.Parameters.AddWithValue("ParamName", 123);
      object oResult = comm.ExecuteScalar();
      dtLastEventTime = DateTime.Parse(oResult.ToString());
    }
    //Do something with dtLastEventTime
  }
}

我一直在使用Moq和各种语法方法来模拟数据库对象而没有太多运气..(我不得不做一些转换到System.Data.Common对象以便能够获得一点点进一步.. DbTransaction,DbConnection,DbCommand等)。

我想知道的主要是能否以这种方式模拟交易,或者我是否在这里咆哮错误的树。幸运的是,我可能能够将接口转换为使用通用的DbTransaction参数而不是特定于提供者的SqlTransaction,但我不相信这就是为什么我在模拟时遇到困难。

这是(这可能是完全错误的,所以请纠正我或评论,如果我正在接近这一点)我到目前为止嘲笑代码...

   var mockParams = new Mock<DbParameterCollection>();
    mockParams.Setup(p => p.Add(new SqlParameter("ParamName", 123)));
    var mockCommand = new Mock<DbCommand>();
    mockCommand.Setup(p => p.Parameters).Returns(mockParams.Object);
    var mockConnection = new Mock<DbConnection>();
    mockConnection.Setup(con => con.CreateCommand()).Returns(mockCommand.Object);
    var mockTrans = new Mock<DbTransaction>();
    mockTrans.Setup(t => t.Connection).Returns(mockConnection.Object);   

但是,这似乎在mockCommand.Setup行上抛出了ArgumentException。 (不可覆盖的成员上的设置无效)

有没有人对如何使用模拟的SqlTransaction参数正确地对此方法进行单元测试有任何想法或建议?

1 个答案:

答案 0 :(得分:3)

彼得,我将假设SqlServer后端。

首先,我会通过在MyMethod()上将入站参数类型更改为IDbTransaction来重构MyMethod(),以便您可以传入模拟。然后我将抽象出ExecuteScalar()调用,这样我就可以在我的测试中轻松检查SqlCommand参数。

由于Moq支持递归模拟,在测试中我会像这样模拟对sharedTransaction的调用:

var mockTrx = new Mock<IDbTransaction>();
mockTrx.Setup(txn => txn.Connection.CreateCommand()).Returns(new SqlCommand());

再创建一个模拟器来检查方法中的命令参数,并在ExecuteScalar()调用上返回一个值,然后完成!

重构了MyMethod():

public void MyMethod(IDbTransaction sharedTransaction, DateTime eventTime)
{
    DateTime lastEventTime;
    using (var cmd = (SqlCommand)sharedTransaction.Connection.CreateCommand())
    {
        cmd.CommandText = "somecmdtext";
        cmd.Parameters.AddWithValue("ParamName", 123);

        object oResult = dbUtility.ExecuteScalar(cmd);

        lastEventTime = DateTime.Parse(oResult.ToString());
    }

    //Do something with dtLastEventTime
    lastEventTime += TimeSpan.FromMinutes(1);
}

测试:

[Test]
public void ShouldCallCorrectProcWithParams()
{
    var mockTrx = new Mock<IDbTransaction>();
    mockTrx.Setup(txn => txn.Connection.CreateCommand()).Returns(new SqlCommand());

    var dbUtil = new Mock<IDbUtility>();
    dbUtil.Setup(exec => exec.ExecuteScalar(
                             It.Is<SqlCommand>(
                             cmd => cmd.CommandText == "somecmdtext"
                             && cmd.Parameters.Count == 1
                             && cmd.Parameters[0].ParameterName == "ParamName")))
                 .Returns(DateTime.Now);

     var session = new Session {dbUtility = dbUtil.Object};

     session.MyMethod(mockTrx.Object, DateTime.Now);

     mockTrx.VerifyAll();
     dbUtil.VerifyAll();

}

就个人而言,我喜欢在我的SqlCommand调用上测试参数,所以我在我的例子中从这个角度攻击了你的问题。