我已经开始学习单元测试了。我正在使用 JUnit 5,我想测试将一些数据插入数据库的方法(使用 JDBC)。这是我的代码:
数据源.java
import java.sql.*;
public class Datasource {
public static final String CONNECTION = "jdbc:mysql://127.0.0.1:3306/java";
private Connection connection;
public boolean open() {
try {
connection = DriverManager.getConnection(CONNECTION, "root", "");
return true;
} catch (SQLException e) {
System.out.println(e.getMessage());
return false;
}
}
public boolean insertTable() {
try {
String query = "INSERT INTO artists(name) VALUES(?)";
PreparedStatement stmt = connection.prepareStatement(query);
stmt.setString(1, "Test");
int result = stmt.executeUpdate();
if (result == 1) return true;
return false;
} catch (SQLException e) {
System.out.println(e.getMessage());
return false;
}
}
}
数据源测试.java
import static org.junit.jupiter.api.Assertions.*;
class DatasourceTest {
private Datasource datasource;
@org.junit.jupiter.api.BeforeEach
void setUp() {
datasource = new Datasource();
if (!datasource.open()) {
System.exit(-1);
}
}
@org.junit.jupiter.api.Test
void insertTable() {
assertTrue(datasource.insertTable());
}
}
它工作正常,但它实际上在我的数据库中插入了一条记录,但我想要做的是模拟它。是否可以仅使用 JUnit 来实现?如果没有,我需要什么?一个简单的实现将不胜感激。
编辑
我发现了一个名为Mockito的工具,这是我需要的吗?如果是这样,谁能告诉我如何部署我的方法 insertTable() 的简单测试?