diff --git a/jOOQ/src/main/java/org/jooq/tools/jdbc/Mock.java b/jOOQ/src/main/java/org/jooq/tools/jdbc/Mock.java new file mode 100644 index 0000000000..66f1a1c748 --- /dev/null +++ b/jOOQ/src/main/java/org/jooq/tools/jdbc/Mock.java @@ -0,0 +1,105 @@ +/** + * Copyright (c) 2009-2016, Data Geekery GmbH (http://www.datageekery.com) + * All rights reserved. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + * + * Other licenses: + * ----------------------------------------------------------------------------- + * Commercial licenses for this work are available. These replace the above + * ASL 2.0 and offer limited warranties, support, maintenance, and commercial + * database integrations. + * + * For more information, please visit: http://www.jooq.org/licenses + * + * + * + * + * + * + * + * + * + * + * + * + * + * + * + * + */ +package org.jooq.tools.jdbc; + +import static org.jooq.impl.DSL.using; + +import org.jooq.AttachableInternal; +import org.jooq.Configuration; +import org.jooq.Record; +import org.jooq.Result; +import org.jooq.impl.DefaultConfiguration; + +/** + * Various utilities related to {@link MockDataProvider}. + * + * @author Lukas Eder + */ +public final class Mock { + + /** + * Create a new {@link MockDataProvider} that always returns a single record + * for all queries. + */ + public static final MockDataProvider of(Record record) { + return of(result(record)); + } + + /** + * Create a new {@link MockDataProvider} that always returns the same result + * for all queries. + */ + public static final MockDataProvider of(Result result) { + return of(new MockResult(result.size(), result)); + } + + /** + * Create a new {@link MockDataProvider} that always returns the same mock + * results for all queries. + */ + public static final MockDataProvider of(final MockResult... result) { + return new MockDataProvider() { + @Override + public MockResult[] execute(MockExecuteContext ctx) { + return result; + } + }; + } + + /** + * Wrap a record in a result. + */ + static final Result result(Record data) { + Configuration configuration = data instanceof AttachableInternal + ? ((AttachableInternal) data).configuration() + : new DefaultConfiguration(); + + Result result = using(configuration).newResult(data.fields()); + result.add(data); + + return result; + } + + /** + * No instances + */ + private Mock() {} +} diff --git a/jOOQ/src/main/java/org/jooq/tools/jdbc/MockResult.java b/jOOQ/src/main/java/org/jooq/tools/jdbc/MockResult.java index 7e90df786b..f117502f9a 100644 --- a/jOOQ/src/main/java/org/jooq/tools/jdbc/MockResult.java +++ b/jOOQ/src/main/java/org/jooq/tools/jdbc/MockResult.java @@ -40,15 +40,10 @@ */ package org.jooq.tools.jdbc; -import static org.jooq.impl.DSL.using; - import java.sql.Statement; -import org.jooq.AttachableInternal; -import org.jooq.Configuration; import org.jooq.Record; import org.jooq.Result; -import org.jooq.impl.DefaultConfiguration; /** * A mock result. @@ -99,7 +94,7 @@ public class MockResult { * @param data The single record in this result. */ public MockResult(Record data) { - this(1, result(data)); + this(1, Mock.result(data)); } /** @@ -113,17 +108,6 @@ public class MockResult { this.data = data; } - private static final Result result(Record data) { - Configuration configuration = data instanceof AttachableInternal - ? ((AttachableInternal) data).configuration() - : new DefaultConfiguration(); - - Result result = using(configuration).newResult(data.fields()); - result.add(data); - - return result; - } - @Override public String toString() { return (data != null) ? data.toString() : ("" + rows);