以下是一个使用embedded-postgres在Java中编写的单元测试示例，该测试会启动一个嵌入式PostgreSQL服务器，运行一个简单的查询，并断言结果是否如预期：
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import ru.yandex.qatools.embed.postgresql.EmbeddedPostgres;
import ru.yandex.qatools.embed.postgresql.util.SocketUtil;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.Statement;
import static org.junit.Assert.assertEquals;
public class EmbeddedPostgresTest {
private static EmbeddedPostgres postgres;
private static String dbUrl;
@BeforeClass
public static void startPostgres() throws Exception {
// 获取可用的网络端口
int port = SocketUtil.findAvailableTcpPort(5432);
// 启动嵌入式PostgreSQL服务器
postgres = new EmbeddedPostgres(port);
postgres.start();
// 获取数据库URL
dbUrl = postgres.getJdbcUrl("postgres", "postgres");
}
@AfterClass
public static void stopPostgres() throws Exception {
// 停止嵌入式服务器
postgres.stop();
}
@Test
public void testQuery() throws Exception {
// 建立数据库连接
try (Connection connection = DriverManager.getConnection(dbUrl, "postgres", "postgres")) {
// 创建Statement对象执行查询
try (Statement statement = connection.createStatement()) {
// 运行一个创建表和插入数据的SQL脚本
statement.execute("CREATE TABLE test_table (id SERIAL PRIMARY KEY, name VARCHAR(255));");
statement.execute("INSERT INTO test_table (name) VALUES ('test');");
// 运行查询并验证结果
try (ResultSet resultSet = statement.executeQuery("SELECT name FROM test_table WHERE id = 1;")) {
assertEquals(true, resultSet.next());
assertEquals("test", resultSet.getString("name"));
assertEquals(false, resultSet.next());
}
}
}
}
}这段代码首先使用@BeforeClass注解启动了一个嵌入式PostgreSQL服务器，然后定义了一个@Test注解的方法，该方法创建了一个数据库连接，执行了一个简单的查询，并断言查询结果是否如预期。最后，在@AfterClass注解的方法中停止了PostgreSQL服务器。这个过程模拟了在实际开发中进行单元测试的情况，确保数据库操作在不影响系统其余部分的情况下能正确运行。