diff --git a/jOOQ/src/main/java/org/jooq/tools/jdbc/MockConfiguration.java b/jOOQ/src/main/java/org/jooq/tools/jdbc/MockConfiguration.java new file mode 100644 index 0000000000..8f916388d7 --- /dev/null +++ b/jOOQ/src/main/java/org/jooq/tools/jdbc/MockConfiguration.java @@ -0,0 +1,269 @@ +/** + * Copyright (c) 2009-2015, Data Geekery GmbH (http://www.datageekery.com) + * All rights reserved. + * + * This work is dual-licensed + * - under the Apache Software License 2.0 (the "ASL") + * - under the jOOQ License and Maintenance Agreement (the "jOOQ License") + * ============================================================================= + * You may choose which license applies to you: + * + * - If you're using this work with Open Source databases, you may choose + * either ASL or jOOQ License. + * - If you're using this work with at least one commercial database, you must + * choose jOOQ License + * + * For more information, please visit http://www.jooq.org/licenses + * + * Apache Software License 2.0: + * ----------------------------------------------------------------------------- + * 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. + * + * jOOQ License and Maintenance Agreement: + * ----------------------------------------------------------------------------- + * Data Geekery grants the Customer the non-exclusive, timely limited and + * non-transferable license to install and use the Software under the terms of + * the jOOQ License and Maintenance Agreement. + * + * This library is distributed with a LIMITED WARRANTY. See the jOOQ License + * and Maintenance Agreement for more details: http://www.jooq.org/licensing + */ +package org.jooq.tools.jdbc; + +import java.sql.Connection; +import java.util.Map; + +import javax.sql.DataSource; + +import org.jooq.Configuration; +import org.jooq.ConnectionProvider; +import org.jooq.ConverterProvider; +import org.jooq.ExecuteListenerProvider; +import org.jooq.RecordListenerProvider; +import org.jooq.RecordMapperProvider; +import org.jooq.SQLDialect; +import org.jooq.TransactionProvider; +import org.jooq.VisitListenerProvider; +import org.jooq.conf.Settings; + +/** + * A mock configuration. + *

+ * This {@link Configuration} wraps a delegate Configuration and + * wraps all {@link ConnectionProvider} references in + * {@link MockConnectionProvider}. + * + * @author Lukas Eder + */ +public class MockConfiguration implements Configuration { + + /** + * Generated UID + */ + private static final long serialVersionUID = 2600901130544049995L; + + private final Configuration delegate; + private final MockDataProvider provider; + + public MockConfiguration(Configuration delegate, MockDataProvider provider) { + this.delegate = delegate; + this.provider = provider; + } + + @Override + public Map data() { + return delegate.data(); + } + + @Override + public Object data(Object key) { + return delegate.data(key); + } + + @Override + public Object data(Object key, Object value) { + return delegate.data(key, value); + } + + @Override + public ConnectionProvider connectionProvider() { + return new MockConnectionProvider(delegate.connectionProvider(), provider); + } + + @Override + public TransactionProvider transactionProvider() { + return delegate.transactionProvider(); + } + + @Override + public RecordMapperProvider recordMapperProvider() { + return delegate.recordMapperProvider(); + } + + @Override + public RecordListenerProvider[] recordListenerProviders() { + return delegate.recordListenerProviders(); + } + + @Override + public ExecuteListenerProvider[] executeListenerProviders() { + return delegate.executeListenerProviders(); + } + + @Override + public VisitListenerProvider[] visitListenerProviders() { + return delegate.visitListenerProviders(); + } + + @Override + public ConverterProvider converterProvider() { + return delegate.converterProvider(); + } + + @SuppressWarnings("deprecation") + @Override + public org.jooq.SchemaMapping schemaMapping() { + return delegate.schemaMapping(); + } + + @Override + public SQLDialect dialect() { + return delegate.dialect(); + } + + @Override + public SQLDialect family() { + return delegate.family(); + } + + @Override + public Settings settings() { + return delegate.settings(); + } + + @Override + public Configuration set(ConnectionProvider newConnectionProvider) { + return delegate.set(newConnectionProvider); + } + + @Override + public Configuration set(Connection newConnection) { + return delegate.set(newConnection); + } + + @Override + public Configuration set(DataSource newDataSource) { + return delegate.set(newDataSource); + } + + @Override + public Configuration set(TransactionProvider newTransactionProvider) { + return delegate.set(newTransactionProvider); + } + + @Override + public Configuration set(RecordMapperProvider newRecordMapperProvider) { + return delegate.set(newRecordMapperProvider); + } + + @Override + public Configuration set(RecordListenerProvider... newRecordListenerProviders) { + return delegate.set(newRecordListenerProviders); + } + + @Override + public Configuration set(ExecuteListenerProvider... newExecuteListenerProviders) { + return delegate.set(newExecuteListenerProviders); + } + + @Override + public Configuration set(VisitListenerProvider... newVisitListenerProviders) { + return delegate.set(newVisitListenerProviders); + } + + @Override + public Configuration set(ConverterProvider newConverterProvider) { + return delegate.set(newConverterProvider); + } + + @Override + public Configuration set(SQLDialect newDialect) { + return delegate.set(newDialect); + } + + @Override + public Configuration set(Settings newSettings) { + return delegate.set(newSettings); + } + + @Override + public Configuration derive() { + return delegate.derive(); + } + + @Override + public Configuration derive(ConnectionProvider newConnectionProvider) { + return delegate.derive(newConnectionProvider); + } + + @Override + public Configuration derive(Connection newConnection) { + return delegate.derive(newConnection); + } + + @Override + public Configuration derive(DataSource newDataSource) { + return delegate.derive(newDataSource); + } + + @Override + public Configuration derive(TransactionProvider newTransactionProvider) { + return delegate.derive(newTransactionProvider); + } + + @Override + public Configuration derive(RecordMapperProvider newRecordMapperProvider) { + return delegate.derive(newRecordMapperProvider); + } + + @Override + public Configuration derive(RecordListenerProvider... newRecordListenerProviders) { + return delegate.derive(newRecordListenerProviders); + } + + @Override + public Configuration derive(ExecuteListenerProvider... newExecuteListenerProviders) { + return delegate.derive(newExecuteListenerProviders); + } + + @Override + public Configuration derive(VisitListenerProvider... newVisitListenerProviders) { + return delegate.derive(newVisitListenerProviders); + } + + @Override + public Configuration derive(ConverterProvider newConverterProvider) { + return delegate.derive(newConverterProvider); + } + + @Override + public Configuration derive(SQLDialect newDialect) { + return delegate.derive(newDialect); + } + + @Override + public Configuration derive(Settings newSettings) { + return delegate.derive(newSettings); + } + +} diff --git a/jOOQ/src/main/java/org/jooq/tools/jdbc/MockConnectionProvider.java b/jOOQ/src/main/java/org/jooq/tools/jdbc/MockConnectionProvider.java new file mode 100644 index 0000000000..17a2a5cd4a --- /dev/null +++ b/jOOQ/src/main/java/org/jooq/tools/jdbc/MockConnectionProvider.java @@ -0,0 +1,87 @@ +/** + * Copyright (c) 2009-2015, Data Geekery GmbH (http://www.datageekery.com) + * All rights reserved. + * + * This work is dual-licensed + * - under the Apache Software License 2.0 (the "ASL") + * - under the jOOQ License and Maintenance Agreement (the "jOOQ License") + * ============================================================================= + * You may choose which license applies to you: + * + * - If you're using this work with Open Source databases, you may choose + * either ASL or jOOQ License. + * - If you're using this work with at least one commercial database, you must + * choose jOOQ License + * + * For more information, please visit http://www.jooq.org/licenses + * + * Apache Software License 2.0: + * ----------------------------------------------------------------------------- + * 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. + * + * jOOQ License and Maintenance Agreement: + * ----------------------------------------------------------------------------- + * Data Geekery grants the Customer the non-exclusive, timely limited and + * non-transferable license to install and use the Software under the terms of + * the jOOQ License and Maintenance Agreement. + * + * This library is distributed with a LIMITED WARRANTY. See the jOOQ License + * and Maintenance Agreement for more details: http://www.jooq.org/licensing + */ +package org.jooq.tools.jdbc; + +import java.sql.Connection; + +import org.jooq.ConnectionProvider; + +/** + * A mock connection provider. + *

+ * This {@link ConnectionProvider} wraps a delegate + * ConnectionProvider and wraps all acquired {@link Connection} + * references in {@link MockConnection}. + * + * @author Lukas Eder + */ +public class MockConnectionProvider implements ConnectionProvider { + + private final ConnectionProvider delegate; + private final MockDataProvider provider; + + public MockConnectionProvider(ConnectionProvider delegate, MockDataProvider provider) { + this.delegate = delegate; + this.provider = provider; + } + + @Override + public final Connection acquire() { + return new MockConnectionWrapper(delegate.acquire()); + } + + @Override + public final void release(Connection connection) { + if (connection instanceof MockConnectionWrapper) + delegate.release(((MockConnectionWrapper) connection).connection); + else + throw new IllegalArgumentException("Argument connection must be a MockConnectionWrapper"); + } + + private class MockConnectionWrapper extends MockConnection { + final Connection connection; + + public MockConnectionWrapper(Connection connection) { + super(provider); + this.connection = connection; + } + } +}