使用 JOOQ 的 MockDataProvider，如何设置返回的 lastId()？

Question

注意：我没有使用 Jooq 的代码生成器

在 DAO 单元测试中，DAO 被测试以确保在插入对象后，DAO 将 Id 设置为由数据库中的 last_insert_id() 返回。由于我使用的是 JOOQ 的 MockConnection 和 MockDataProvider，因此没有联系实际的数据库。

当 DAO 执行以下操作时：

DSLContext ctx = DSL.using(connection, SQLDialect.MYSQL);
//insert
//get id
BigInteger id = ctx.lastId();

JOOQ 执行以下查询：

select last_insert_id() from dual;

在我的 MockDataProvider 中，我检查何时执行此查询并相应地返回结果：

import static org.jooq.impl.DSL.dual;

//other code

@Override
public MockResult[] execute(MockExecuteContext ctx) throws SQLException {
   Field<BigInteger> id = DSL.field("", BigInteger.class);
   Record record = dual().newRecord();
   record.setValue(id, BigInteger.valueOf(1234567));
   return new MockResult[] { new MockResult(record) };             
}

当返回上述 MockResult 时，我得到以下异常

java.lang.IllegalArgumentException: Field () is not contained in Row ()

为 last_insert_id() 查询填充 MockResult 的正确方法是什么？

Answer 1

MockDataProvider这是您的for的有效实现DSLContext.lastID()：

BigInteger expected = BigInteger.valueOf(1234567);
DSLContext ctx = DSL.using(new MockConnection(c -> {
    Field<BigInteger> id = DSL.field("last_insert_id()", BigInteger.class);
    Record record = DSL.using(MYSQL).newRecord(id);
    record.setValue(id, expected);
    return new MockResult[] { new MockResult(record) };
}), SQLDialect.MYSQL);

assertEquals(expected, ctx.lastID());

您的方法本质上有两件事出了问题：

字段名称

您选择的字段的字段名称实际上被称为last_insert_id()（至少在 jOOQ 3.5.3 中），因此您还必须以这种方式命名您的模拟字段：

    Field<BigInteger> id = DSL.field("last_insert_id()", BigInteger.class);

记录

您必须创建一条已包含您的字段的记录。该dual()表包含不同的字段。jOOQ 生成查询的事实与您必须生成包含字段from dual的事实无关：Recordlast_insert_id()

    Record record = DSL.using(MYSQL).newRecord(id);

警告

当然，您对 3.5.3 中的 jOOQ 实现做出了强有力的假设。无法保证生成的确切查询DSLContext.lastID()始终是

select last_insert_id() from dual