Example usage for org.springframework.jdbc.datasource TestDataSourceWrapper setTarget

Introduction

In this page you can find the example usage for org.springframework.jdbc.datasource TestDataSourceWrapper setTarget.

Prototype

public void setTarget(DataSource target) 

Source Link

Usage

From source file:org.springframework.jdbc.object.GenericSqlQueryTests.java

@Before
public void setUp() throws Exception {
    super.setUp();
    mockPreparedStatement = createMock(PreparedStatement.class);
    mockResultSet = createMock(ResultSet.class);
    this.bf = new XmlBeanFactory(
            new ClassPathResource("org/springframework/jdbc/object/GenericSqlQueryTests-context.xml"));
    TestDataSourceWrapper testDataSource = (TestDataSourceWrapper) bf.getBean("dataSource");
    testDataSource.setTarget(mockDataSource);
}

---

From source file:org.springframework.jdbc.object.GenericStoredProcedureTests.java

@Test
public void testAddInvoices() throws Exception {
    DefaultListableBeanFactory bf = new DefaultListableBeanFactory();
    new XmlBeanDefinitionReader(bf).loadBeanDefinitions(
            new ClassPathResource("org/springframework/jdbc/object/GenericStoredProcedureTests-context.xml"));
    Connection connection = mock(Connection.class);
    DataSource dataSource = mock(DataSource.class);
    given(dataSource.getConnection()).willReturn(connection);
    CallableStatement callableStatement = mock(CallableStatement.class);
    TestDataSourceWrapper testDataSource = (TestDataSourceWrapper) bf.getBean("dataSource");
    testDataSource.setTarget(dataSource);

    given(callableStatement.execute()).willReturn(false);
    given(callableStatement.getUpdateCount()).willReturn(-1);
    given(callableStatement.getObject(3)).willReturn(new Integer(4));

    given(connection.prepareCall("{call " + "add_invoice" + "(?, ?, ?)}")).willReturn(callableStatement);

    StoredProcedure adder = (StoredProcedure) bf.getBean("genericProcedure");
    Map<String, Object> in = new HashMap<String, Object>(2);
    in.put("amount", 1106);
    in.put("custid", 3);
    Map out = adder.execute(in);/*from   w w  w.ja  v a  2 s  .c  om*/
    Integer id = (Integer) out.get("newid");
    assertEquals(4, id.intValue());

    verify(callableStatement).setObject(1, new Integer(1106), Types.INTEGER);
    verify(callableStatement).setObject(2, new Integer(3), Types.INTEGER);
    verify(callableStatement).registerOutParameter(3, Types.INTEGER);
    verify(callableStatement).close();
}

---