JOOQ MockDataProvider - 如何根据某些条件返回不同的模拟？答案

【问题标题】：JOOQ MockDataProvider - how to return different mocks depending on some conditions?JOOQ MockDataProvider - 如何根据某些条件返回不同的模拟？
【发布时间】：2015-12-28 01:22:13
【问题描述】：

最近我正在使用 JOOQ MockDataProvider 实现一个单元测试。当我想在 DAO 中使用我的模拟提供程序进行许多选择时，我不得不使用许多 if-else 语句。

根据：http://www.jooq.org/doc/latest/manual/tools/jdbc-mocking，我只需要检查我的 SQL 是否以某些查询开头。我在 DAO 中使用的 SQL 可以从 3 种不同的方式开始，所以我使用了相当复杂的 if-else 构造。

第二件事 - 我希望我的 MockDataProvider 仅在第一次执行 SQL 时返回一些模拟结果，然后不返回任何结果 - 在循环中迭代 5 次的 DAO 中，每次我的 DAO 应该检查一些东西数据库。我不知道如何模拟这种行为，所以我使用了简单的计数器 - 但它看起来很糟糕，我希望它以一种好的方式实现。这是我的代码：

public class SomeProvider implements MockDataProvider {

private static final String STATEMENT_NOT_SUPPORTED_ = "Statement not supported: ";
private static final String SELECT_META = "select \"myschema\".\"meta\".";
private static final String SELECT_CLIENT = "select \"myschema\".\"client\".";
private static final String SELECT_KEY = "select \"myschema\".\"key\".";
private static final String TEST_SECRET_KEY = "some key";
private static final String KEY = "40sld";
private static final String DROP = "DROP";
private static final String SOME_URL = "something";
private static final String MONKEY = "monkey";
private static final int FIRST_ITERARION_COUNTER_VALUE = 0;
private final Long keyId;
int counter = 0;

public SomeProvider(Long keyId) {
    this.keyId = keyId;
}

@Override
public MockResult[] execute(MockExecuteContext ctx) throws SQLException {

    DSLContext create = DSL.using(SQLDialect.POSTGRES);
    MockResult[] mock = new MockResult[3];
    String sql = ctx.sql();
    if (sql.toUpperCase().startsWith(DROP)) {
        throw new SQLException(STATEMENT_NOT_SUPPORTED_ + sql);
    } else if (sql.startsWith(SELECT_CLIENT)) {

        Result<ClientRecord> result = create.newResult(CLIENT);
        result.add(create.newRecord(CLIENT));
        result.get(0).setValue(CLIENT.ID, 1L);
        result.get(0).setValue(CLIENT.SECRET_KEY, TEST_SECRET_KEY);
        mock[0] = new MockResult(1, result);

    } else if (sql.startsWith(SELECT_META)) {

        Result<MetaRecord> metaResult = create.newResult(META);
        metaResult.add(create.newRecord(META));
        metaResult.get(0).setValue(META.ID, 1L);

        metaResult.get(0).setValue(META.URL, SOME_URL);
        metaResult.get(0).setValue(META.KEY, KEY);
        metaResult.get(0).setValue(META.OPTION, keyId);
        mock[0] = new MockResult(1, metaResult);

    } else if (sql.startsWith(SELECT_KEY)) {

        Result<KeyRecord> keyResult = create.newResult(KEY);
        if (counter == FIRST_ITERARION_COUNTER_VALUE) {
            // first SELECT returns monkey, rest will return no results
            keyResult.add(create.newRecord(KEY));
            keyResult.get(0).setValue(KEY.ID, 1L);
            keyResult.get(0).setValue(KEY.VALUE, MONKEY);
            mock[0] = new MockResult(1, keyResult);
        } else {
            mock[0] = new MockResult(0, keyResult);
        }
        counter++;
    }

    return mock;
}
}

它有效，但看起来很糟糕。总结一下我的问题是：如何根据查询和查询执行次数返回（使用一个提供程序）不同的结果。也许这个类只用于简单的 DSLContext 模拟，而不是模拟整个 DAO，它使用一个 DSLContext 多次使用许多查询。

【问题讨论】：

标签： java unit-testing mocking jooq

【解决方案1】：

我已经这样做了，但这并不好玩 - 模拟最终需要自己的测试...... 另一种方法是使模拟提供者能够配置预期的内容和返回的内容。

provider.when(..).times(5).return(...)

等等...返回值可以是生成器，也可以是字面值。何时可以使用正则表达式以及字符串

【讨论】：