mybatis项目测试是否连接成功java代码

在MyBatis项目中，通常会编写一些简单的单元测试来验证数据库连接是否成功。以下是一个基本的Java代码示例，使用`Junit`和`Mockito`库来模拟数据库连接：

import org.junit.jupiter.api.*;
import org.mockito.Mock;
import org.mybatis.spring.SqlSessionFactory;
import org.springframework.jdbc.core.JdbcTemplate;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.Mockito.when;

public class DatabaseConnectionTest {

    @Mock
    private DataSource dataSource;

    @Mock
    private SqlSessionFactory sqlSessionFactory;

    @Mock
    private Connection connection;

    @BeforeEach
    public void setUp() {
        // 模拟DataSource有getConnection方法并返回connection
        when(dataSource.getConnection()).thenReturn(connection);
    }

    @Test
    public void testDatabaseConnectivity() throws SQLException {
        JdbcTemplate jdbcTemplate = new JdbcTemplate(sqlSessionFactory); // 这里假设SqlSessionFactory有JdbcTemplate方法
        try (Connection conn = jdbcTemplate.getDataSource().getConnection()) { // 使用JdbcTemplate尝试获取连接
            assertTrue(conn != null, "Database connection should not be null");
            System.out.println("Connected to database successfully!");
        }
    }

    @AfterEach
    public void tearDown() {
        // 在这里可以添加清理断言或资源释放等操作
    }