我是一名有用的助手,可以翻译文本。
我正在尝试模拟数据库操作。我在模拟SqlParameterCollection时遇到了问题。我尝试创建一个虚拟方法来返回DbParameterCollection,但是这样我就失去了SqlParameterCollection提供的所有功能,例如AddWithValue等。有没有办法模拟SqlParameterCollection?是否有其他方法来单元测试DAL?我正在使用Moq。
代码如下:
在DAL中:
我正在尝试模拟数据库操作。我在模拟SqlParameterCollection时遇到了问题。我尝试创建一个虚拟方法来返回DbParameterCollection,但是这样我就失去了SqlParameterCollection提供的所有功能,例如AddWithValue等。有没有办法模拟SqlParameterCollection?是否有其他方法来单元测试DAL?我正在使用Moq。
代码如下:
在DAL中:
protected virtual IDbConnection GetConnection(string connectionString)
{
return new SqlConnection(connectionString);
}
protected virtual IDbCommand GetCommand(IDbConnection cn)
{
return cn.CreateCommand();
}
protected virtual IDbTransaction GetTransaction(IDbConnection cn)
{
return cn.BeginTransaction(IsolationLevel.Serializable);
}
Public Bool InsertInDatabase(DataTable dt)
{
using (IDbConnection cn = GetConnection(cnstr))
{
cn.Open();
using (IDbTransaction tran = GetTransaction(cn))
{
IDbCommand cmd = GetCommand(cn);
cmd.Transaction = tran;
cmd.Connection = cn;
cmd.CommandType = CommandType.StoredProcedure;
cmd.CommandText = "sp_InsertInDatabase";
SqlParameterCollection cmdParams = cmd.Parameters as SqlParameterCollection;
cmdParams.AddWithValue("@param1", dt);
cmd.ExecuteNonQuery();
}
}
}
在单元测试项目中:
protected override IDbConnection GetConnection(string connectionString)
{
return Mock.Of<IDbConnection>();
}
protected override IDbCommand GetCommand(IDbConnection cn)
{
return Mock.Of<IDbCommand>();
}
protected override IDbTransaction GetTransaction(IDbConnection cn)
{
return Mock.Of<IDbTransaction>();
}
public void TestInsertInDatabase()
{
base.InsertInDatabase(new DataTable());
}
--解决方法--
创建了一个扩展方法来添加带有值的参数。感谢 Marc Gravell 指引我朝这个方向。
public static IDbDataParameter AddParameterWithValue(this IDbCommand cmd, string paramName, object paramValue)
{
var dbParam = cmd.CreateParameter();
if (dbParam != null)
{
dbParam.ParameterName = paramName;
dbParam.Value = paramValue;
}
return dbParam;
}