使用 H2 DB 来做单元测试

- 日理万妓 2022-04-15 04:49 169阅读 0赞

# 使用 H2 DB 来做单元测试 #

H2 是一款嵌入式数据库。当然它也可以C/S方式运行，但这种方式并不常用，因为Oracle、SQL Server、MySQL等产品在这方面比它强大的多。我们一般喜欢用它来做单元测试，更喜欢把它的数据库放在内存中，因为这样访问速度快，而且每次启动程序得到的数据库都是一致的。

### 创建内存数据库并使用 ###

@Test
    public void useH2() throws ClassNotFoundException, SQLException {
    	Class.forName("org.h2.Driver");
    	// 所有数据都存储在内存中，数据库没有名字
    	String url = "jdbc:h2:mem:";
    	// 默认的用户名，没有密码
    	String user = "sa";
    	String password = "";
    	Connection conn = DriverManager.getConnection(url, user, password);
    
    	Statement statement = conn.createStatement();
    
    	// H2的语法和MySQL的语法还是有些差别的，比如KEY `w`，CONSTRAINT `w`
    	statement.execute("create table test (id char(10) not null)");
    	ResultSet rs = statement.executeQuery("show tables");
    	assert rs.next();
    
    	assertEquals("TEST", rs.getString(1));
    	rs = statement.executeQuery("select * from test");
    	assert !rs.next();
    
    	statement.execute("insert into test values ('123')");
    	rs = statement.executeQuery("select * from test");
    	assert rs.next();
    	assertEquals("123", rs.getString(1));
    
    	statement.close();
    	conn.close();
    }

可以写个脚本文件，用于创建数据库和填充数据。

### 在Spring中使用H2 DB（基于注解的配置） ###

import javax.sql.DataSource;
    
    import org.springframework.beans.factory.annotation.Configurable;
    import org.springframework.context.annotation.Bean;
    import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
    import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
    
    @Configurable
    public class DbConfig {
    	@Bean
    	public DataSource dataSource() {
    		return new EmbeddedDatabaseBuilder()
    				.setType(EmbeddedDatabaseType.H2)
    				// 用于创建表的sql脚本
    				.addScript("classpath:conf/sql/schema.sql")
    				// 用于填充数据的sql脚本
    				.addScript("classpath:conf/sql/data.sql")
    				.build();
    	}
    }

### JUnit4 测试类 ###

import static org.junit.Assert.assertEquals;
    
    import java.sql.Connection;
    import java.sql.DriverManager;
    import java.sql.ResultSet;
    import java.sql.SQLException;
    import java.sql.Statement;
    
    import javax.sql.DataSource;
    
    import org.junit.Test;
    import org.junit.runner.RunWith;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.test.context.ContextConfiguration;
    import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
    
    import com.mxh.h2.config.DbConfig;
    
    @RunWith(SpringJUnit4ClassRunner.class)
    @ContextConfiguration(classes = DbConfig.class)
    public class MenTest {
    	@Autowired
    	private DataSource dataSource;
    
    	@Test
    	public void getConnection() throws SQLException {
    		dataSource.getConnection();
    	}
    
    	@Test
    	public void useH2() throws ClassNotFoundException, SQLException {
    		Class.forName("org.h2.Driver");
    		String url = "jdbc:h2:mem:";
    		String user = "sa";
    		String password = "";
    		Connection conn = DriverManager.getConnection(url, user, password);
    
    		Statement statement = conn.createStatement();
    
    		statement.execute("create table test (id char(10) not null)");
    		ResultSet rs = statement.executeQuery("show tables");
    		assert rs.next();
    
    		assertEquals("TEST", rs.getString(1));
    		rs = statement.executeQuery("select * from test");
    		assert !rs.next();
    
    		statement.execute("insert into test values ('123')");
    		rs = statement.executeQuery("select * from test");
    		assert rs.next();
    		assertEquals("123", rs.getString(1));
    
    		statement.close();
    		conn.close();
    	}
    
    	@Test
    	public void useH2WithSpring() throws SQLException {
    		Connection conn = dataSource.getConnection();
    		Statement stmt = conn.createStatement();
    
    		ResultSet rs = stmt.executeQuery("show tables");
    		assert rs.next();
    		assertEquals("TEST", rs.getString(1));
    
    		stmt.close();
    		conn.close();
    	}
    }

使用Spring来做集成更方便一些，毕竟有些东西人家已经给你写好了。

感觉Spring集成的H2就是专门用来做测试的，其他的功能好像并不支持，连url都没有设置的方法。

### 上述操作只需引入它的jar包即可。也可以下载它的程序包或是Windows上的安装程序，里面有一个Web控制台和其它的一些功能。 ###