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

Question

我目前正在尝试使用 Moq 学习 Mocking，并且我想在我拥有的现有数据库上尝试它，但是我不确定如何处理这个问题的正确方法。

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

我的 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 执行此操作？我是否为演员实体创建了一个类？我应该为我的DbConnection班级创建一个接口吗？很抱歉所有的问题，但我真的很难解决这个问题。

Answer 1

目前，被测试的方法与实现问题的耦合过于紧密，无法轻松地进行单元测试。尝试将这些实现问题抽象出来，以便在隔离测试中可以轻松模拟它们。

public interface IDbConnectionFactory {
    IDbConnection CreateConnection();
}

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

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.
            }
        }
    }
}

工厂的生产实施将返回一个实际 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 检查他们的快速入门