Как смоделировать SqlParameterCollection с помощью Moq - PullRequest
Новая
       22

Как смоделировать SqlParameterCollection с помощью Moq

6 голосов
/ 16 июня 2011

Я пытаюсь смоделировать операции с базой данных. У меня проблема с издевательством SqlParameterCollection. Я пытался создать virtual метод, который будет возвращать 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());
    }

- Решение -

Создан метод расширения для добавления параметра со значением. Спасибо, Марк Гравелл, за то, что указал мне в этом направлении. 

    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;

    }

Ответы [ 2 ]

8 голосов
/ 16 июня 2011

Лично я подхожу к этой проблеме, написав AddParameterWithValue метод расширения для DbCommand (или IDbCommand). Он должен быть по команде, чтобы иметь доступ к CreateParameter, а затем вызвать .Parameters.Add.

Это позволяет легко использовать любой стек ADO.NET, включая абстракции, такие как декораторы логирования.

1 голос
/ 02 февраля 2015

@ Asdfg HI Я в основном смоделировал коллекцию параметров, как показано ниже 

 string connectionString = "connectionstring";
        var sqlConnection = new SqlConnection(connectionString);
        var command = sqlConnection.CreateCommand();
        //****************Setup Mock************************//
        Castle.DynamicProxy.Generators.AttributesToAvoidReplicating.Add(typeof(System.Data.SqlClient.SqlClientPermissionAttribute));
        var mockDataReader1 = new Mock<IDataReader>();
        command.Parameters.Add(new SqlParameter("@po_tint_Result", 1));
        //setup read return value
        Queue<bool> responseQueue = new Queue<bool>();
        responseQueue.Enqueue(true);
        responseQueue.Enqueue(false);
        mockDataReader1.Setup(a => a.Read()).Returns(() => responseQueue.Dequeue());
        var mockDb = new Mock<SqlDatabase>(connectionString);
        mockDb.Setup(a => a.GetStoredProcCommand("SPNAME")).Returns(command);
        mockDb.Setup(a => a.ExecuteNonQuery(command));
        obj1.DbConn = mockDb.Object;
        //*************************************************//

Надеюсь, это поможет

...