org.jooq.tools.jdbc.MockConnectionProvider Maven / Gradle / Ivy
/*
* 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
*
* https://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: https://www.jooq.org/legal/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}.
*
* Disclaimer: The general idea of mocking a JDBC connection with this
* jOOQ API is to provide quick workarounds, injection points, etc. using a very
* simple JDBC abstraction. It is NOT RECOMMENDED to emulate an entire database
* (including complex state transitions, transactions, locking, etc.) using this
* mock API. Once you have this requirement, please consider using an actual
* database instead for integration testing (e.g. using
* https://www.testcontainers.org),
* rather than implementing your test database inside of a
* MockDataProvider.
*
* @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 w)
delegate.release(w.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;
}
}
}