时间戳

首页 > 解决方案 > 模拟 jooq 选择请求

java - 模拟 jooq 选择请求

问题描述

我在 jooq select 中有我的服务myMethod()

Result<MyRecord> records = dsl.selectFrom(MyTable).fetch();

我尝试使用 mockito 编写单元测试。我想模拟我的 dsl 并在我打电话时得到 2 个不同的结果service.myMethod()

Mockito.when(dsl.selectFrom(MyTable)).thenReturn(result)

但它不起作用。如何模拟我的选择请求？

标签： javajunitmockitojooq

解决方案

模拟 jOOQ API

如果您想模拟 jOOQ API，模拟的有趣方法是各种fetch()方法或execute()方法，因为它们的行为类似于 jOOQ DSL API 的“终端”方法。所有中间方法都应该产生新的模拟，而不是实际结果。

但是，由于 jOOQ 的 DSL 非常庞大，您可能会忽略各种边缘情况，所以我认为完全模拟它可能效果不佳。

模拟 JDBC API

如果你真的想模拟 jOOQ API，我宁愿在 JDBC 级别模拟它。jOOQ 实现了一个模拟 JDBC Connection，允许使用单个 lambda 表达式模拟整个 JDBC，您可以在其中执行以下操作（来自手册）：

MockDataProvider provider = new MyProvider();
MockConnection connection = new MockConnection(provider);
DSLContext ctx = DSL.using(connection, SQLDialect.ORACLE);
Result<BookRecord> result = ctx.selectFrom(BOOK).where(BOOK.ID.eq(5)).fetch();

然后（简化）：

public class MyProvider implements MockDataProvider {

    @Override
    public MockResult[] execute(MockExecuteContext ctx) throws SQLException {
        DSLContext ctx = DSL.using(SQLDialect.ORACLE);
        MockResult[] mock = new MockResult[1];
        String sql = ctx.sql();

        if (sql.toUpperCase().startsWith("DROP")) {
            throw new SQLException("Statement not supported: " + sql);
        }
        else if (sql.toUpperCase().startsWith("SELECT")) {
            Result<Record2<Integer, String>> result =
                ctx.newResult(AUTHOR.ID, AUTHOR.LAST_NAME);
            result.add(cctx
                .newRecord(AUTHOR.ID, AUTHOR.LAST_NAME)
                .values(1, "Orwell"));
            mock[0] = new MockResult(1, result);
        }

        return mock;
    }
}

这个模拟实现还有一个开箱即用的基于文件的版本MockFileDatabase，它使用正则表达式来匹配 SQL 字符串并提供基于文本的语法来为您的查询构建结果集。

一般模拟数据库

请注意，模拟不是测试数据库交互的好方法 - 在简单的情况下，它只能让你到目前为止。一种更好、更彻底的方法是集成测试。如果必须，请使用内存数据库，但最好通过testcontainers 之类的方式使用您的生产数据库产品（例如 Oracle、PostgreSQL 等） 。

推荐阅读