@BeforeMethod private void setup() { state = new State(); state.setProp(JdbcPublisher.JDBC_PUBLISHER_FINAL_TABLE_NAME, destinationTable); state.setProp(JdbcPublisher.JDBC_PUBLISHER_DATABASE_NAME, database); commands = mock(JdbcWriterCommands.class); factory = mock(JdbcWriterCommandsFactory.class); conn = mock(Connection.class); when(factory.newInstance(state, conn)).thenReturn(commands); workUnitStates = new ArrayList<>(); workUnitState = mock(WorkUnitState.class); when(workUnitState.getProp(ConfigurationKeys.WRITER_STAGING_TABLE)).thenReturn(stagingTable); workUnitStates.add(workUnitState); publisher = new JdbcPublisher(state, factory); publisher = spy(publisher); doReturn(conn).when(publisher).createConnection(); }