org.anystub.jdbc.StubParameterMetaData Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of anystub Show documentation
the library allows to create handy stubs for any tests
There is a newer version: 1.0.5
package org.anystub.jdbc;

import org.anystub.Supplier;

import java.sql.ParameterMetaData;
import java.sql.SQLException;

public class StubParameterMetaData implements ParameterMetaData {

    private final StubConnection stubConnection;
    private final StubPreparedStatement stubPreparedStatement;
    private ParameterMetaData realParameterMetaData = null;

    public StubParameterMetaData(StubConnection stubConnection, StubPreparedStatement stubPreparedStatement) throws SQLException {
        this.stubConnection = stubConnection;
        this.stubPreparedStatement = stubPreparedStatement;

        stubConnection.add(() -> {
            realParameterMetaData = stubPreparedStatement.getRealStatement().getParameterMetaData();
        });
    }

    @Override
    public int getParameterCount() throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .requestI(new Supplier() {
                    @Override
                    public Integer get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.getParameterCount();
                    }
                }, stubPreparedStatement.getSql(), "#getParameterCount");
    }

    @Override
    public int isNullable(int i) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .requestI(new Supplier() {
                    @Override
                    public Integer get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.isNullable(i);
                    }
                }, useKeys("isNullable", i));
    }

    @Override
    public boolean isSigned(int i) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .requestB(new Supplier() {
                    @Override
                    public Boolean get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.isSigned(i);
                    }
                }, useKeys("isSigned", i));
    }

    @Override
    public int getPrecision(int i) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .requestI(new Supplier() {
                    @Override
                    public Integer get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.getPrecision(i);
                    }
                }, useKeys("getPrecision", i));
    }

    @Override
    public int getScale(int i) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .requestI(new Supplier() {
                    @Override
                    public Integer get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.getScale(i);
                    }
                }, useKeys("getScale", i));
    }

    @Override
    public int getParameterType(int i) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .requestI(new Supplier() {
                    @Override
                    public Integer get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.getParameterType(i);
                    }
                }, useKeys("getParameterType", i));
    }

    @Override
    public String getParameterTypeName(int i) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .request(new Supplier() {
                    @Override
                    public String get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.getParameterTypeName(i);
                    }
                }, useKeys("getParameterTypeName", i));
    }

    @Override
    public String getParameterClassName(int i) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .request(new Supplier() {
                    @Override
                    public String get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.getParameterClassName(i);
                    }
                }, useKeys("getParameterClassName", i));
    }

    @Override
    public int getParameterMode(int i) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .requestI(new Supplier() {
                    @Override
                    public Integer get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.getParameterMode(i);
                    }
                }, useKeys("#getParameterMode", i));
    }

    @Override
    public  T unwrap(Class aClass) throws SQLException {
        return null;
    }

    @Override
    public boolean isWrapperFor(Class aClass) throws SQLException {
        return stubConnection
                .getStubDataSource()
                .getBase()
                .requestB(new Supplier() {
                    @Override
                    public Boolean get() throws SQLException {
                        stubConnection.runSql();
                        return realParameterMetaData.isWrapperFor(aClass);
                    }
                }, stubPreparedStatement.getSql(), "#isWrapperFor", aClass.getCanonicalName());
    }

    private String[] useKeys(String name, int i) {
        return new String[]{stubPreparedStatement.getSql(), "#", name, String.valueOf(i)};
    }
}