Examples with MockConnection com.mockrunner.mock.jdbc.MockConnection used on opensource projects

Example 1 with MockConnection

use of com.mockrunner.mock.jdbc.MockConnection in project cayenne by apache.

the class LocalDateTypeTest method testSetJdbcObject.

public void testSetJdbcObject() throws Exception {
    PreparedStatement statement = new MockPreparedStatement(new MockConnection(), "update t set c = ?");
    LocalDate savedObject = new LocalDate(0);
    type.setJdbcObject(statement, savedObject, 1, Types.DATE, 0);
    Object object = ((MockPreparedStatement) statement).getParameter(1);
    assertEquals(Date.class, object.getClass());
    assertEquals(savedObject.toDate().getTime(), ((Date) object).getTime());
    type.setJdbcObject(statement, savedObject, 1, Types.TIMESTAMP, 0);
    object = ((MockPreparedStatement) statement).getParameter(1);
    assertEquals(Timestamp.class, object.getClass());
    assertEquals(savedObject.toDate().getTime(), ((Timestamp) object).getTime());
    type.setJdbcObject(statement, null, 1, Types.TIMESTAMP, 0);
    object = ((MockPreparedStatement) statement).getParameter(1);
    assertNull(object);
}

Example 2 with MockConnection

use of com.mockrunner.mock.jdbc.MockConnection in project cayenne by apache.

the class LocalTimeTypeTest method testSetJdbcObject.

public void testSetJdbcObject() throws Exception {
    PreparedStatement statement = new MockPreparedStatement(new MockConnection(), "update t set c = ?");
    LocalTime date = new LocalTime(0);
    type.setJdbcObject(statement, date, 1, Types.TIME, 0);
    Object object = ((MockPreparedStatement) statement).getParameter(1);
    assertEquals(Time.class, object.getClass());
    assertEquals(new LocalDate(0, DateTimeZone.UTC).toDateTime(date).getMillis(), ((Time) object).getTime());
    type.setJdbcObject(statement, date, 1, Types.TIMESTAMP, 0);
    object = ((MockPreparedStatement) statement).getParameter(1);
    assertEquals(Timestamp.class, object.getClass());
    assertEquals(new LocalDate(0, DateTimeZone.UTC).toDateTime(date).getMillis(), ((Timestamp) object).getTime());
    type.setJdbcObject(statement, null, 1, Types.TIMESTAMP, 0);
    object = ((MockPreparedStatement) statement).getParameter(1);
    assertNull(object);
}

Example 3 with MockConnection

use of com.mockrunner.mock.jdbc.MockConnection in project cayenne by apache.

the class BatchActionLockingIT method testRunAsIndividualQueriesOptimisticLockingFailure.

@Test
public void testRunAsIndividualQueriesOptimisticLockingFailure() throws Exception {
    EntityResolver resolver = runtime.getDataDomain().getEntityResolver();
    // test with adapter that supports keys...
    JdbcAdapter adapter = buildAdapter(true);
    DbEntity dbEntity = resolver.getObjEntity(SimpleLockingTestEntity.class).getDbEntity();
    List<DbAttribute> qualifierAttributes = Arrays.asList(dbEntity.getAttribute("LOCKING_TEST_ID"), dbEntity.getAttribute("NAME"));
    Collection<String> nullAttributeNames = Collections.singleton("NAME");
    Map<String, Object> qualifierSnapshot = new HashMap<>();
    qualifierSnapshot.put("LOCKING_TEST_ID", 1);
    DeleteBatchQuery batchQuery = new DeleteBatchQuery(dbEntity, qualifierAttributes, nullAttributeNames, 5);
    batchQuery.setUsingOptimisticLocking(true);
    batchQuery.add(qualifierSnapshot);
    DeleteBatchTranslator batchQueryBuilder = new DeleteBatchTranslator(batchQuery, adapter, null);
    MockConnection mockConnection = new MockConnection();
    PreparedStatementResultSetHandler preparedStatementResultSetHandler = mockConnection.getPreparedStatementResultSetHandler();
    preparedStatementResultSetHandler.setExactMatch(false);
    preparedStatementResultSetHandler.setCaseSensitive(false);
    preparedStatementResultSetHandler.prepareUpdateCount("DELETE", 0);
    boolean generatesKeys = false;
    DataNode node = new DataNode();
    node.setAdapter(adapter);
    node.setEntityResolver(resolver);
    node.setRowReaderFactory(mock(RowReaderFactory.class));
    BatchAction action = new BatchAction(batchQuery, node, false);
    try {
        action.runAsIndividualQueries(mockConnection, batchQueryBuilder, new MockOperationObserver(), generatesKeys);
        fail("No OptimisticLockingFailureException thrown.");
    } catch (OptimisticLockException e) {
    }
    assertEquals(0, mockConnection.getNumberCommits());
    assertEquals(0, mockConnection.getNumberRollbacks());
}

Example 4 with MockConnection

use of com.mockrunner.mock.jdbc.MockConnection in project cayenne by apache.

the class BatchActionLockingIT method testRunAsIndividualQueriesSuccess.

@Test
public void testRunAsIndividualQueriesSuccess() throws Exception {
    EntityResolver resolver = runtime.getDataDomain().getEntityResolver();
    // test with adapter that supports keys...
    JdbcAdapter adapter = buildAdapter(true);
    DbEntity dbEntity = resolver.getObjEntity(SimpleLockingTestEntity.class).getDbEntity();
    List<DbAttribute> qualifierAttributes = Arrays.asList(dbEntity.getAttribute("LOCKING_TEST_ID"), dbEntity.getAttribute("NAME"));
    Collection<String> nullAttributeNames = Collections.singleton("NAME");
    Map<String, Object> qualifierSnapshot = new HashMap<>();
    qualifierSnapshot.put("LOCKING_TEST_ID", new Integer(1));
    DeleteBatchQuery batchQuery = new DeleteBatchQuery(dbEntity, qualifierAttributes, nullAttributeNames, 5);
    batchQuery.setUsingOptimisticLocking(true);
    batchQuery.add(qualifierSnapshot);
    DeleteBatchTranslator batchQueryBuilder = new DeleteBatchTranslator(batchQuery, adapter, null);
    MockConnection mockConnection = new MockConnection();
    PreparedStatementResultSetHandler preparedStatementResultSetHandler = mockConnection.getPreparedStatementResultSetHandler();
    preparedStatementResultSetHandler.setExactMatch(false);
    preparedStatementResultSetHandler.setCaseSensitive(false);
    preparedStatementResultSetHandler.prepareUpdateCount("DELETE", 1);
    boolean generatesKeys = false;
    DataNode node = new DataNode();
    node.setAdapter(adapter);
    node.setEntityResolver(resolver);
    node.setRowReaderFactory(mock(RowReaderFactory.class));
    BatchAction action = new BatchAction(batchQuery, node, false);
    action.runAsIndividualQueries(mockConnection, batchQueryBuilder, new MockOperationObserver(), generatesKeys);
    assertEquals(0, mockConnection.getNumberCommits());
    assertEquals(0, mockConnection.getNumberRollbacks());
}

Example 5 with MockConnection

use of com.mockrunner.mock.jdbc.MockConnection in project cayenne by apache.

the class JDBCResultIteratorTest method testNextDataRow.

@Test
public void testNextDataRow() throws Exception {
    Connection c = new MockConnection();
    Statement s = new MockStatement(c);
    MockResultSet rs = new MockResultSet("rs");
    rs.addColumn("a", new Object[] { "1", "2", "3" });
    RowDescriptor descriptor = new RowDescriptorBuilder().setResultSet(rs).getDescriptor(new ExtendedTypeMap());
    RowReader<?> rowReader = new DefaultRowReaderFactory().rowReader(descriptor, new MockQueryMetadata(), mock(DbAdapter.class), Collections.<ObjAttribute, ColumnDescriptor>emptyMap());
    JDBCResultIterator it = new JDBCResultIterator(s, rs, rowReader);
    DataRow row = (DataRow) it.nextRow();
    assertNotNull(row);
    assertEquals(1, row.size());
    assertEquals("1", row.get("a"));
}