예제 #1
0
  @Test
  public void testUpsertWithConnection() throws Exception {
    SavedReviewerSearch search = new SavedReviewerSearch();
    SavedReviewerSearch newSearch = new SavedReviewerSearch();
    Connection connection = EasyMock.createStrictMock(Connection.class);

    SavedReviewerSearchDAO dao =
        EasyMock.createMockBuilder(SavedReviewerSearchDAO.class)
            .withConstructor()
            .addMockedMethod("update", Connection.class, SavedReviewerSearch.class)
            .addMockedMethod("create", Connection.class, SavedReviewerSearch.class)
            .addMockedMethod("isUpdate")
            .createStrictMock();

    EasyMock.expect(dao.isUpdate(connection, search)).andReturn(false);
    EasyMock.expect(dao.create(connection, search)).andReturn(newSearch);
    EasyMock.replay(dao);
    assertEquals(
        "Wrong search returned in create case.", newSearch, dao.upsert(connection, search));
    EasyMock.verify(dao);

    EasyMock.reset(dao);
    EasyMock.expect(dao.isUpdate(connection, search)).andReturn(true);
    EasyMock.expect(dao.update(connection, search)).andReturn(true);
    EasyMock.replay(dao);
    assertEquals("Wrong search returned in update case.", search, dao.upsert(connection, search));
    EasyMock.verify(dao);
  }
예제 #2
0
  @Test
  public void testUpdate() throws Exception {
    SavedReviewerSearch search = new SavedReviewerSearch();
    SavedReviewerSearchDAO dao =
        EasyMock.createMockBuilder(SavedReviewerSearchDAO.class)
            .withConstructor()
            .addMockedMethod("update", Connection.class, SavedReviewerSearch.class)
            .createStrictMock();
    ITransactionManager txMgr = EasyMock.createStrictMock(ITransactionManager.class);
    dao.setTransactionManager(txMgr);
    Connection mockConn = EasyMock.createStrictMock(Connection.class);
    EasyMock.expect(dao.update(mockConn, search)).andReturn(true);

    MockExecutionWithThrow.<Boolean>execute(txMgr, mockConn);
    EasyMock.replay(txMgr, mockConn, dao);
    assertEquals("Wrong search result.", true, dao.update(search));
    EasyMock.verify(txMgr, mockConn, dao);
  }
예제 #3
0
  @Test
  public void testUpdateWithConnection() throws Exception {
    PreparedStatement preparedStatement = EasyMock.createStrictMock(PreparedStatement.class);
    Connection connection = EasyMock.createStrictMock(Connection.class);
    SavedReviewerSearch search = new SavedReviewerSearch();

    SavedReviewerSearchDAO dao =
        EasyMock.createMockBuilder(SavedReviewerSearchDAO.class)
            .withConstructor()
            .addMockedMethod("updatePreparedStatementForUpdate")
            .createStrictMock();

    EasyMock.expect(connection.prepareStatement(SavedReviewerSearchDAO.UPDATE_SQL))
        .andReturn(preparedStatement);
    dao.updatePreparedStatementForUpdate(preparedStatement, search);
    EasyMock.expectLastCall();
    EasyMock.expect(preparedStatement.executeUpdate()).andReturn(1);
    EasyMock.expectLastCall();
    preparedStatement.close();
    EasyMock.replay(preparedStatement, connection, dao);
    assertEquals("Wrong result returned.", true, dao.update(connection, search));
    EasyMock.verify(preparedStatement, connection, dao);

    // failure case
    EasyMock.reset(preparedStatement, connection, dao);
    EasyMock.expect(connection.prepareStatement(SavedReviewerSearchDAO.UPDATE_SQL))
        .andReturn(preparedStatement);
    dao.updatePreparedStatementForUpdate(preparedStatement, search);
    EasyMock.expectLastCall();
    EasyMock.expect(preparedStatement.executeUpdate()).andThrow(new SQLException("blah"));
    EasyMock.expectLastCall();
    preparedStatement.close();
    EasyMock.replay(preparedStatement, connection, dao);
    try {
      dao.update(connection, search);
      fail("Should have thrown a sql exception.");
    } catch (SQLException ex) {
      // expected
    }
    EasyMock.verify(preparedStatement, connection, dao);
  }