使用 Moq 模拟插入查询到 MySQL 数据库

Question

我目前正在尝试使用 Moq 学习 Mocking，我想在我现有的数据库上尝试一下，但是我不确定如何才是正确的方法。

在我的数据层中，我有一个 class 处理与数据库的连接，并具有用于插入、选择等的各种方法。我想测试一个演员是否被正确插入到数据库中。

我的 Insert 方法目前看起来像这样：

public void Insert(string firstname, string lastname)
{
    string query = $"INSERT INTO `sakila`.`actor`(`first_name`,`last_name`) VALUES('" + firstname + "','" + lastname + "')";
    Console.WriteLine(query);
    //open connection
    if (this.OpenConnection() == true)
    {
        Console.WriteLine("Established connection");
        //create command and assign the query and connection from the constructor
        MySqlCommand cmd = new MySqlCommand(query, connection);

        //Execute command
        cmd.ExecuteNonQuery();
        Console.WriteLine("Insert query succesfully executed.");

        //close connection
        this.CloseConnection();
    }
}

我将如何使用 Mocks 来做到这一点？我是否为演员实体创建了 class？我应该为我的 DbConnection class 创建一个界面吗？很抱歉提出所有问题，但我真的不知道如何解决这个问题。

Answer 1

目前正在测试的方法与实现问题紧密耦合，使其易于单独进行单元测试。尝试抽象出这些实现问题，以便可以轻松地模拟它们以进行孤立测试。

public interface IDbConnectionFactory {
    IDbConnection CreateConnection();
}

以上连接工厂抽象可用于访问 MySql 数据存储的其他必要 System.Data 抽象。

public class MyDataAccessClass {
    private IDbConnectionFactory connectionFactory;

    public MyDataAccessClass(IDbConnectionFactory connectionFactory) {
        this.connectionFactory = connectionFactory;
    }

    public void Insert(string firstname, string lastname) {
        var query = $"INSERT INTO `sakila`.`actor`(`first_name`,`last_name`) VALUES('" + firstname + "','" + lastname + "')";
        Console.WriteLine(query);
        using(var connection = connectionFactory.CreateConnection() {
            //Creates and returns a MySqlCommand object associated with the MySqlConnection. 
            using(var command = connection.CreateCommand()) {
                command.CommandText = query;
                Console.WriteLine("Established connection");
                connection.Open();
                command.ExecuteNonQuery();
                Console.WriteLine("Insert query succesfully executed.");
                connection.Close();//is not actually necessary as the using statement will make sure to close the connection.
            }
        }
    }
}

工厂的生产实施将return实际MySqlConnection

public class MySqlConnectionFactory: IDbConnectionFactory {
    public IDbConnection CreateConnection() {
        return new MySqlConnection("connection string");
    }
}

可以通过依赖注入传递到数据层

为了测试，您可以使用您选择的模拟框架来模拟接口，或者创建您自己的模拟来注入和测试您的方法。

[TestClass]
public class DataAccessLayerUnitTest {
    [TestMethod]
    public void TestInsert() {
        //Arrange
        var commandMock = new Mock<IDbCommand>();
        commandMock
            .Setup(m => m.ExecuteNonQuery())
            .Verifiable();

        var connectionMock = new Mock<IDbConnection>();
        connectionMock
            .Setup(m => m.CreateCommand())
            .Returns(commandMock.Object);

        var connectionFactoryMock = new Mock<IDbConnectionFactory>();
        connectionFactoryMock
            .Setup(m => m.CreateConnection())
            .Returns(connectionMock.Object);

        var sut = new MyDataAccessClass(connectionFactoryMock.Object);
        var firstName = "John";
        var lastName = "Doe";

        //Act
        sut.Insert(firstName, lastName);

        //Assert
        commandMock.Verify();
    }
}

最后，建议您在命令文本中使用命令参数，因为手动构造查询字符串会打开代码，导致 SQL 注入攻击。

为了更好地了解如何使用 Moq，请查看他们的 Quickstart

使用 Moq 模拟插入查询到 MySQL 数据库

Mocking insert query to a MySQL Database using Moq

c#

mysql

unit-testing

moq

mocking