MySQL HEX/UNHEX函数在内存数据库中无法测试的替代方案咨询

我之前在测试带HEX/UNHEX的MySQL查询时也遇到过一模一样的困扰——不想把十六进制转换逻辑从SQL移到代码里，又想用轻量的内存数据库做单元测试。下面几个方案亲测有效，完全不用修改你的业务SQL：

方案1：给H2数据库自定义兼容函数 #

H2支持自定义Java函数，我们可以手动实现MySQL风格的HEX和UNHEX，然后注册到H2中：

1. 先写一个包含静态方法的工具类，实现转换逻辑：

import java.util.HexFormat;

public class MySqlHexFunctions {
    // 对应MySQL的HEX(byte[]) -> String
    public static String hex(byte[] input) {
        return input == null ? null : HexFormat.of().formatHex(input);
    }

    // 对应MySQL的UNHEX(String) -> byte[]
    public static byte[] unhex(String input) {
        return input == null ? null : HexFormat.of().parseHex(input);
    }
}

2. 在H2数据库初始化时注册这些函数，有两种方式：

   • 方式A：通过JDBC URL参数
     在连接URL里添加INIT参数，直接执行注册语句：

     jdbc:h2:mem:testdb;DB_CLOSE_DELAY=-1;INIT=CREATE ALIAS IF NOT EXISTS HEX FOR 'com.yourpackage.MySqlHexFunctions.hex'\;CREATE ALIAS IF NOT EXISTS UNHEX FOR 'com.yourpackage.MySqlHexFunctions.unhex'
     

     注意分号要用反斜杠转义，避免被JDBC URL解析器拆分。

   • 方式B：在测试初始化代码中执行
     如果你用Spring Boot或者其他框架，可以在测试启动后执行SQL语句：

     CREATE ALIAS IF NOT EXISTS HEX FOR "com.yourpackage.MySqlHexFunctions.hex";
     CREATE ALIAS IF NOT EXISTS UNHEX FOR "com.yourpackage.MySqlHexFunctions.unhex";
     

这样你的原有SQL里的HEX()和UNHEX()就会调用我们自定义的方法，和MySQL的行为一致。

方案2：用Testcontainers启动真实MySQL容器（非内存，但轻量） #

如果你的SQL逻辑特别复杂，内存数据库的兼容始终有问题，Testcontainers是个绝佳替代——它会在测试时启动一个真实的MySQL容器，测试结束后自动销毁，完全模拟生产环境的MySQL行为，不用改任何SQL：

以JUnit 5为例，配置步骤如下：

1. 添加Testcontainers MySQL依赖（比如Maven）：

<dependency>
    <groupId>org.testcontainers</groupId>
    <artifactId>mysql</artifactId>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.testcontainers</groupId>
    <artifactId>junit-jupiter</artifactId>
    <scope>test</scope>
</dependency>

2. 在测试类中配置容器：

import org.junit.jupiter.api.Test;
import org.testcontainers.containers.MySQLContainer;
import org.testcontainers.junit.jupiter.Container;
import org.testcontainers.junit.jupiter.Testcontainers;

@Testcontainers
public class HexQueryTest {
    // 启动MySQL 8.0容器，自动拉取镜像
    @Container
    private static final MySQLContainer<?> MYSQL_CONTAINER = new MySQLContainer<>("mysql:8.0")
            .withDatabaseName("test_db")
            .withUsername("test_user")
            .withPassword("test_pass");

    @Test
    void testHexUnhexQuery() {
        // 获取容器的JDBC连接信息
        String jdbcUrl = MYSQL_CONTAINER.getJdbcUrl();
        String username = MYSQL_CONTAINER.getUsername();
        String password = MYSQL_CONTAINER.getPassword();
        
        // 这里用JDBC或者ORM框架执行你的原有SQL即可，完全支持HEX/UNHEX
    }
}

这个方案虽然不是内存数据库，但启动速度很快（首次拉取镜像慢，之后复用），而且完全兼容MySQL的所有特性，适合复杂场景。

方案3：扩展JPA/Hibernate方言（如果用ORM框架） #

如果你用Hibernate作为ORM框架，可以自定义H2方言，注册HEX和UNHEX函数，让Hibernate自动适配：

1. 自定义方言类：

import org.hibernate.dialect.H2Dialect;
import org.hibernate.dialect.function.StandardSQLFunction;
import org.hibernate.type.StandardBasicTypes;

public class CustomH2Dialect extends H2Dialect {
    public CustomH2Dialect() {
        super();
        // 注册HEX函数，映射到H2的原生函数（如果参数类型兼容），或者自定义函数
        registerFunction("hex", new StandardSQLFunction("hex", StandardBasicTypes.STRING));
        // 注册UNHEX函数
        registerFunction("unhex", new StandardSQLFunction("unhex", StandardBasicTypes.BINARY));
    }
}

2. 在配置文件中指定这个方言：
   比如Spring Boot的application-test.properties：

spring.jpa.properties.hibernate.dialect=com.yourpackage.CustomH2Dialect

这样Hibernate在生成SQL或者执行原生SQL时，会自动识别HEX/UNHEX函数，适配H2数据库。

小贴士：如果H2原生的HEX/UNHEX和MySQL的行为有差异（比如参数类型处理），还是推荐方案1的自定义函数，确保行为完全一致。

内容的提问来源于stack exchange，提问作者Gagan