Examples with MockCallableStatement com.alibaba.druid.mock.MockCallableStatement used on opensource projects

Example 1 with MockCallableStatement

use of com.alibaba.druid.mock.MockCallableStatement in project druid by alibaba.

the class EncodingConvertFilterTest2 method test_stat.

public void test_stat() throws Exception {
    Assert.assertTrue(dataSource.isInited());
    EncodingConvertFilter filter = (EncodingConvertFilter) dataSource.getProxyFilters().get(0);
    DruidPooledConnection conn = dataSource.getConnection();
    final String PARAM_VALUE = "中国";
    CallableStatement stmt = conn.prepareCall("select ?");
    stmt.setString(1, PARAM_VALUE);
    MockCallableStatement raw = stmt.unwrap(MockCallableStatement.class);
    stmt.execute();
    String param1 = (String) raw.getParameters().get(0);
    String C_TEXT = new String(param1.getBytes(SERVER_ENCODING), CLIENT_ENCODING);
    Assert.assertEquals(PARAM_VALUE, C_TEXT);
    Assert.assertFalse(param1.equals(PARAM_VALUE));
    MyResultSet rawRs = new MyResultSet(raw);
    rawRs.setValue(filter.encode((ConnectionProxy) conn.getConnection(), text));
    raw.getOutParameters().add(rawRs);
    ResultSet rs = (ResultSet) stmt.getObject(1);
    rs.next();
    Assert.assertEquals(text, rs.getString(1));
    Assert.assertEquals(text, rs.getString("1"));
    Assert.assertEquals(text, rs.getObject(1));
    Assert.assertEquals(text, rs.getObject("1"));
    Assert.assertEquals(text, rs.getObject(1, Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(text, rs.getObject("1", Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(text, rs.getString(2));
    Assert.assertEquals(text, rs.getString("2"));
    Assert.assertEquals(text, rs.getObject(2));
    Assert.assertEquals(text, rs.getObject("2"));
    Assert.assertEquals(text, rs.getObject(2, Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(text, rs.getObject("2", Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(text, rs.getString(3));
    Assert.assertEquals(text, rs.getString("3"));
    Assert.assertEquals(text, rs.getObject(3));
    Assert.assertEquals(text, rs.getObject("3"));
    Assert.assertEquals(text, rs.getObject(3, Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(text, rs.getObject("3", Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(text, rs.getString(4));
    Assert.assertEquals(text, rs.getString("4"));
    Assert.assertEquals(text, rs.getObject(4));
    Assert.assertEquals(text, rs.getObject("4"));
    Assert.assertEquals(text, rs.getObject(4, Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(text, rs.getObject("4", Collections.<String, Class<?>>emptyMap()));
    stmt.registerOutParameter(2, Types.VARCHAR);
    stmt.registerOutParameter(3, Types.CLOB);
    raw.getOutParameters().add(param1);
    raw.getOutParameters().add(param1);
    Assert.assertEquals(C_TEXT, stmt.getString(4));
    Assert.assertEquals(C_TEXT, stmt.getString("4"));
    Assert.assertEquals(C_TEXT, stmt.getObject(4));
    Assert.assertEquals(C_TEXT, stmt.getObject("4"));
    Assert.assertEquals(C_TEXT, stmt.getObject(4, Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(C_TEXT, stmt.getObject("4", Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(C_TEXT, stmt.getString(5));
    Assert.assertEquals(C_TEXT, stmt.getString("5"));
    Assert.assertEquals(C_TEXT, stmt.getObject(5));
    Assert.assertEquals(C_TEXT, stmt.getObject("5"));
    Assert.assertEquals(C_TEXT, stmt.getObject(5, Collections.<String, Class<?>>emptyMap()));
    Assert.assertEquals(C_TEXT, stmt.getObject("5", Collections.<String, Class<?>>emptyMap()));
    stmt.setObject(1, C_TEXT);
    Assert.assertEquals(param1, raw.getParameters().get(0));
    stmt.setObject(2, new StringReader(C_TEXT));
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(1)));
    stmt.setCharacterStream(3, new StringReader(C_TEXT));
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(2)));
    stmt.setCharacterStream(4, new StringReader(C_TEXT), C_TEXT.length());
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(3)));
    stmt.setCharacterStream(5, new StringReader(C_TEXT), (long) C_TEXT.length());
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(4)));
    stmt.setObject(6, C_TEXT, Types.VARCHAR);
    Assert.assertEquals(param1, raw.getParameters().get(5));
    stmt.setObject(7, new StringReader(C_TEXT), Types.VARCHAR);
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(6)));
    stmt.setObject(8, C_TEXT, Types.VARCHAR, 0);
    Assert.assertEquals(param1, raw.getParameters().get(7));
    stmt.setObject(9, new StringReader(C_TEXT), Types.VARCHAR, 0);
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(8)));
    stmt.setObject(10, 1, Types.INTEGER);
    Assert.assertEquals(1, raw.getParameters().get(9));
    stmt.setObject(11, 2, Types.INTEGER, 0);
    Assert.assertEquals(2, raw.getParameters().get(10));
    stmt.setObject(12, 3);
    Assert.assertEquals(3, raw.getParameters().get(11));
    stmt.setObject("13", C_TEXT, Types.VARCHAR);
    Assert.assertEquals(param1, raw.getParameters().get(12));
    stmt.setObject("14", new StringReader(C_TEXT), Types.VARCHAR);
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(13)));
    stmt.setObject("15", C_TEXT, Types.VARCHAR, 0);
    Assert.assertEquals(param1, raw.getParameters().get(14));
    stmt.setObject("16", new StringReader(C_TEXT), Types.VARCHAR, 0);
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(15)));
    stmt.setObject("17", 1, Types.INTEGER);
    Assert.assertEquals(1, raw.getParameters().get(16));
    stmt.setObject("18", 2, Types.INTEGER, 0);
    Assert.assertEquals(2, raw.getParameters().get(17));
    stmt.setObject("19", 3);
    Assert.assertEquals(3, raw.getParameters().get(18));
    stmt.setCharacterStream("20", new StringReader(C_TEXT));
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(19)));
    stmt.setCharacterStream("21", new StringReader(C_TEXT), C_TEXT.length());
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(20)));
    stmt.setCharacterStream("22", new StringReader(C_TEXT), (long) C_TEXT.length());
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(21)));
    stmt.setObject("23", C_TEXT);
    Assert.assertEquals(param1, raw.getParameters().get(22));
    stmt.setObject("24", new StringReader(C_TEXT));
    Assert.assertEquals(param1, Utils.read((Reader) raw.getParameters().get(23)));
    stmt.setObject("25", 1, Types.INTEGER);
    Assert.assertEquals(1, raw.getParameters().get(24));
    stmt.setObject("26", 2, Types.INTEGER, 0);
    Assert.assertEquals(2, raw.getParameters().get(25));
    stmt.setObject("27", 3);
    Assert.assertEquals(3, raw.getParameters().get(26));
    rs.close();
    stmt.close();
    conn.close();
}

Example 2 with MockCallableStatement

use of com.alibaba.druid.mock.MockCallableStatement in project druid by alibaba.

the class FilterChainTest_Clob_2 method setUp.

protected void setUp() throws Exception {
    dataSource = new DruidDataSource();
    MockCallableStatement mockStmt = new MockCallableStatement(null, "") {

        @Override
        public Object getObject(int parameterIndex) throws SQLException {
            invokeCount++;
            return new MockClob();
        }
    };
    statement = new CallableStatementProxyImpl(new ConnectionProxyImpl(null, null, null, 0), mockStmt, "", 1);
}

Example 3 with MockCallableStatement

use of com.alibaba.druid.mock.MockCallableStatement in project druid by alibaba.

the class FilterChainTest_ResultSet_2 method setUp.

protected void setUp() throws Exception {
    dataSource = new DruidDataSource();
    ConnectionProxyImpl conn = new ConnectionProxyImpl(dataSource, null, new Properties(), 0);
    MockCallableStatement mockStmt = new MockCallableStatement(null, "") {

        @Override
        public Object getObject(int parameterIndex) throws SQLException {
            invokeCount++;
            return new MockResultSet(null);
        }
    };
    statement = new CallableStatementProxyImpl(conn, mockStmt, "", 1);
}

Example 4 with MockCallableStatement

use of com.alibaba.druid.mock.MockCallableStatement in project druid by alibaba.

the class CallableStatmentTest method test_connect.

public void test_connect() throws Exception {
    MockCallableStatement rawStmt = null;
    MockResultSet rawRs = null;
    {
        Connection conn = dataSource.getConnection();
        CallableStatement stmt = conn.prepareCall("select 1");
        stmt.execute();
        rawStmt = stmt.unwrap(MockCallableStatement.class);
        ResultSet rs = (ResultSet) stmt.getObject(0);
        rawRs = rs.unwrap(MockResultSet.class);
        rs.next();
        rs.close();
        stmt.close();
        Assert.assertFalse(rawStmt.isClosed());
        Assert.assertTrue(rawRs.isClosed());
        rawRs = rs.unwrap(MockResultSet.class);
        Assert.assertNotNull(rawRs);
        conn.close();
    }
    {
        Connection conn = dataSource.getConnection();
        CallableStatement stmt = conn.prepareCall("select 1");
        stmt.execute();
        Assert.assertSame(rawStmt, stmt.unwrap(MockCallableStatement.class));
        Assert.assertFalse(rawStmt.isClosed());
        stmt.getObject(0);
        ResultSet rs = (ResultSet) stmt.getObject(0);
        rs.next();
        rs.close();
        stmt.close();
        conn.close();
    }
}

Example 5 with MockCallableStatement

use of com.alibaba.druid.mock.MockCallableStatement in project druid by alibaba.

the class PoolableCallableStatementTest method setUp.

protected void setUp() throws Exception {
    MockConnection mockConn = new MockConnection();
    DruidConnectionHolder connHolder = new DruidConnectionHolder(dataSource, mockConn, 0);
    conn = new DruidPooledConnection(connHolder);
    raw = new MockCallableStatement(null, null);
    stmt = new DruidPooledCallableStatement(conn, new PreparedStatementHolder(new PreparedStatementKey("", null, null, 0, 0, 0), raw)) {

        protected SQLException checkException(Throwable error) throws SQLException {
            if (error instanceof SQLException) {
                return (SQLException) error;
            }
            return new SQLException(error);
        }
    };
    Assert.assertEquals(0, raw.getOutParameters().size());
    stmt.registerOutParameter(1, Types.INTEGER);
    Assert.assertEquals(1, raw.getOutParameters().size());
    stmt.registerOutParameter(2, Types.DECIMAL, 10);
    Assert.assertEquals(2, raw.getOutParameters().size());
}