[#4125] Add MockConnectionProvider and MockConfiguration for convenience
This commit is contained in:
Lukas Eder
parent
63c8c48a80
commit
6534a9cd7f
269
jOOQ/src/main/java/org/jooq/tools/jdbc/MockConfiguration.java
Normal file
269
jOOQ/src/main/java/org/jooq/tools/jdbc/MockConfiguration.java
Normal file
@ -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.
|
||||
* <p>
|
||||
* This {@link Configuration} wraps a delegate <code>Configuration</code> 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<Object, Object> 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);
|
||||
}
|
||||
|
||||
}
|
||||
@ -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.
|
||||
* <p>
|
||||
* This {@link ConnectionProvider} wraps a delegate
|
||||
* <code>ConnectionProvider</code> 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;
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user