How to use JdbcMessageHeaders method of com.consol.citrus.jdbc.message.JdbcMessageHeaders class

Best Citrus code snippet using com.consol.citrus.jdbc.message.JdbcMessageHeaders.JdbcMessageHeaders

Source:JdbcEndpointAdapterControllerTest.java

...19import com.consol.citrus.db.server.JdbcServerException;20import com.consol.citrus.endpoint.EndpointAdapter;21import com.consol.citrus.jdbc.data.DataSetCreator;22import com.consol.citrus.jdbc.message.JdbcMessage;23import com.consol.citrus.jdbc.message.JdbcMessageHeaders;24import com.consol.citrus.jdbc.model.*;25import com.consol.citrus.message.*;26import org.springframework.xml.transform.StringResult;27import org.testng.annotations.*;28import java.util.HashMap;29import java.util.Random;30import static org.mockito.ArgumentMatchers.any;31import static org.mockito.ArgumentMatchers.anyString;32import static org.mockito.ArgumentMatchers.eq;33import static org.mockito.Mockito.*;34import static org.testng.Assert.assertEquals;35import static org.testng.Assert.assertTrue;36public class JdbcEndpointAdapterControllerTest {37    private final JdbcEndpointConfiguration jdbcEndpointConfiguration = mock(JdbcEndpointConfiguration.class);38    private final EndpointAdapter endpointAdapter = mock(EndpointAdapter.class);39    private JdbcEndpointAdapterController jdbcEndpointAdapterController;40    @BeforeMethod41    public void setup(){42        final JdbcServerConfiguration serverConfiguration = mock(JdbcServerConfiguration.class);43        when(serverConfiguration.getMaxConnections()).thenReturn(1);44        when(jdbcEndpointConfiguration.getServerConfiguration()).thenReturn(serverConfiguration);45        when(jdbcEndpointConfiguration.getAutoHandleQueries()).thenReturn(new JdbcEndpointConfiguration().getAutoHandleQueries());46        jdbcEndpointAdapterController = new JdbcEndpointAdapterController(jdbcEndpointConfiguration, endpointAdapter);47    }48    @Test49    public void testHandleMessage(){50        //GIVEN51        final Message request = mock(Message.class);52        final Message expectedResponse = mock(Message.class);53        when(endpointAdapter.handleMessage(request)).thenReturn(expectedResponse);54        //WHEN55        final Message response = jdbcEndpointAdapterController.handleMessage(request);56        //THEN57        assertEquals(response, expectedResponse);58    }59    @Test60    public void testHandleMessageWhenEndpointAdapterReturnsNull(){61        //GIVEN62        final Message request = mock(Message.class);63        final Message expectedResponse = JdbcMessage.success();64        when(endpointAdapter.handleMessage(request)).thenReturn(null);65        //WHEN66        final Message response = jdbcEndpointAdapterController.handleMessage(request);67        //THEN68        assertEquals(response.getPayload(), expectedResponse.getPayload());69    }70    @Test71    public void testHandleMessageOperationPayloadConversion(){72        //GIVEN73        final Message request = mock(Message.class);74        final Operation payload = mock(Operation.class);75        when(request.getPayload()).thenReturn(payload);76        when(request.getPayload(Operation.class)).thenReturn(payload);77        final JdbcMarshaller jdbcMarshallerMock = mock(JdbcMarshaller.class);78        when(jdbcEndpointConfiguration.getMarshaller()).thenReturn(jdbcMarshallerMock);79        //WHEN80        jdbcEndpointAdapterController.handleMessage(request);81        //THEN82        verify(jdbcMarshallerMock).marshal(eq(payload), any(StringResult.class));83        verify(request).setPayload(anyString());84    }85    @Test86    public void testOpenConnection(){87        //GIVEN88        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);89        when(jdbcEndpointConfiguration.isAutoConnect()).thenReturn(true);90        jdbcEndpointAdapterController.getConnections().set(0);91        //WHEN92        final int before = jdbcEndpointAdapterController.getConnections().get();93        jdbcEndpointAdapterController.openConnection(new HashMap<>());94        final int after = jdbcEndpointAdapterController.getConnections().get();95        //THEN96        assertEquals(before + 1, after);97        verify(jdbcEndpointAdapterController, never()).handleMessage(any());98    }99    @Test100    public void testOpenConnectionWithoutAutoConnect(){101        //GIVEN102        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);103        jdbcEndpointAdapterController.getConnections().set(0);104        when(jdbcEndpointConfiguration.isAutoConnect()).thenReturn(false);105        //WHEN106        final int before = jdbcEndpointAdapterController.getConnections().get();107        jdbcEndpointAdapterController.openConnection(new HashMap<>());108        final int after = jdbcEndpointAdapterController.getConnections().get();109        //THEN110        verify(jdbcEndpointAdapterController).handleMessage(any());111        assertEquals(before + 1, after);112    }113    @Test(expectedExceptions = JdbcServerException.class)114    public void testOpenConnectionWithoutAutoConnectAndInvalidProperties(){115        //GIVEN116        when(jdbcEndpointConfiguration.isAutoConnect()).thenReturn(false);117        jdbcEndpointAdapterController.getConnections().set(0);118        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);119        final Message errorMessage = mock(Message.class);120        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");121        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());122        //WHEN123        jdbcEndpointAdapterController.openConnection(new HashMap<>());124        //THEN125        //Exception is Thrown126    }127    @Test(expectedExceptions = JdbcServerException.class)128    public void testOpenConnectionMaximumConnectionsReached(){129        //GIVEN130        when(jdbcEndpointConfiguration.isAutoConnect()).thenReturn(true);131        jdbcEndpointAdapterController.getConnections().set(1);132        //WHEN133        jdbcEndpointAdapterController.openConnection(new HashMap<>());134        //THEN135        //Exception is Thrown136    }137    @Test138    public void testCloseConnection(){139        //GIVEN140        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);141        when(jdbcEndpointConfiguration.isAutoConnect()).thenReturn(true);142        jdbcEndpointAdapterController.getConnections().set(1);143        //WHEN144        final int before = jdbcEndpointAdapterController.getConnections().get();145        jdbcEndpointAdapterController.closeConnection();146        final int after = jdbcEndpointAdapterController.getConnections().get();147        //THEN148        verify(jdbcEndpointAdapterController, never()).handleMessage(any());149        assertEquals(before -1 , after);150    }151    @Test152    public void testCloseConnectionWithoutAutoConnect(){153        //GIVEN154        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);155        jdbcEndpointAdapterController.getConnections().set(1);156        when(jdbcEndpointConfiguration.isAutoConnect()).thenReturn(false);157        //WHEN158        final int before = jdbcEndpointAdapterController.getConnections().get();159        jdbcEndpointAdapterController.closeConnection();160        final int after = jdbcEndpointAdapterController.getConnections().get();161        //THEN162        verify(jdbcEndpointAdapterController).handleMessage(any());163        assertEquals(before - 1, after);164    }165    @Test(expectedExceptions = JdbcServerException.class)166    public void testErroneousCloseConnection(){167        //GIVEN168        when(jdbcEndpointConfiguration.isAutoConnect()).thenReturn(false);169        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);170        final Message errorMessage = mock(Message.class);171        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");172        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());173        //WHEN174        jdbcEndpointAdapterController.closeConnection();175        //THEN176        //Exception is Thrown177    }178    @Test179    public void testCloseConnectionWithoutOpenConnectionIsSuccessful(){180        //GIVEN181        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);182        when(jdbcEndpointConfiguration.isAutoConnect()).thenReturn(true);183        jdbcEndpointAdapterController.getConnections().set(0);184        //WHEN185        jdbcEndpointAdapterController.closeConnection();186        //THEN187        verify(jdbcEndpointAdapterController, never()).handleMessage(any());188        assertEquals(jdbcEndpointAdapterController.getConnections().get(), 0);189    }190    @Test191    public void testCreatePreparedStatementWithAutoCreateStatement(){192        //GIVEN193        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);194        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(true);195        //WHEN196        jdbcEndpointAdapterController.createPreparedStatement("some statement");197        //THEN198        verify(jdbcEndpointAdapterController, never()).handleMessage(any());199    }200    @Test201    public void testCreatePreparedStatementWithoutAutoCreateStatement(){202        //GIVEN203        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);204        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(false);205        //WHEN206        jdbcEndpointAdapterController.createPreparedStatement("some statement");207        //THEN208        verify(jdbcEndpointAdapterController).handleMessage(any());209    }210    @Test(expectedExceptions = JdbcServerException.class)211    public void testCreatePreparedStatementWithoutAutoCreateStatementAndFailure(){212        //GIVEN213        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);214        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(false);215        final Message errorMessage = mock(Message.class);216        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");217        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());218        //WHEN219        jdbcEndpointAdapterController.createPreparedStatement("some statement");220        //THEN221        //Exception is thrown222    }223    @Test224    public void testCreateStatementWithAutoCreateStatement(){225        //GIVEN226        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);227        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(true);228        //WHEN229        jdbcEndpointAdapterController.createStatement();230        //THEN231        verify(jdbcEndpointAdapterController, never()).handleMessage(any());232    }233    @Test234    public void testCreateStatementWithoutAutoCreateStatement(){235        //GIVEN236        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);237        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(false);238        //WHEN239        jdbcEndpointAdapterController.createStatement();240        //THEN241        verify(jdbcEndpointAdapterController).handleMessage(any());242    }243    @Test(expectedExceptions = JdbcServerException.class)244    public void testCreateStatementWithoutAutoCreateStatementAndFailure(){245        //GIVEN246        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);247        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(false);248        final Message errorMessage = mock(Message.class);249        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");250        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());251        //WHEN252        jdbcEndpointAdapterController.createStatement();253        //THEN254        //Exception is thrown255    }256    @Test257    public void testExecuteQuery(){258        //GIVEN259        final DataSet expectedDataSet = mock(DataSet.class);260        final DataSetCreator dataSetCreator = mock(DataSetCreator.class);261        when(dataSetCreator.createDataSet(any(), any())).thenReturn(expectedDataSet);262        final JdbcEndpointAdapterController jdbcEndpointAdapterController =263                spy(new JdbcEndpointAdapterController(jdbcEndpointConfiguration, endpointAdapter, dataSetCreator));264        final Message messageToMarshal = mock(Message.class);265        when(messageToMarshal.getHeader(MessageHeaders.MESSAGE_TYPE)).thenReturn(MessageType.JSON.toString());266        doReturn(messageToMarshal).when(jdbcEndpointAdapterController).handleMessage(any());267        final String query = "some query";268        //WHEN269        final DataSet dataSet = jdbcEndpointAdapterController.executeQuery(query);270        //THEN271        verify(jdbcEndpointAdapterController).handleMessage(any());272        verify(dataSetCreator).createDataSet(messageToMarshal, MessageType.JSON);273        assertEquals(dataSet, expectedDataSet);274    }275    @Test(expectedExceptions = JdbcServerException.class)276    public void testExecuteQueryForwardsException(){277        //GIVEN278        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);279        final Message errorMessage = mock(Message.class);280        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");281        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());282        final String query = "some query";283        //WHEN284        jdbcEndpointAdapterController.executeQuery(query);285        //THEN286        //Exception is thrown287    }288    @Test289    public void testExecute(){290        //GIVEN291        final DataSet expectedDataSet = mock(DataSet.class);292        final DataSetCreator dataSetCreator = mock(DataSetCreator.class);293        when(dataSetCreator.createDataSet(any(), any())).thenReturn(expectedDataSet);294        final JdbcEndpointAdapterController jdbcEndpointAdapterController =295                spy(new JdbcEndpointAdapterController(jdbcEndpointConfiguration, endpointAdapter, dataSetCreator));296        final Message messageToMarshal = mock(Message.class);297        when(messageToMarshal.getHeader(MessageHeaders.MESSAGE_TYPE)).thenReturn(MessageType.JSON.toString());298        doReturn(messageToMarshal).when(jdbcEndpointAdapterController).handleMessage(any());299        //WHEN300        jdbcEndpointAdapterController.executeStatement("statement");301        //THEN302        verify(jdbcEndpointAdapterController).handleMessage(any());303    }304    @Test(expectedExceptions = JdbcServerException.class)305    public void testExecuteWithFailure(){306        //GIVEN307        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);308        final Message errorMessage = mock(Message.class);309        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");310        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());311        //WHEN312        jdbcEndpointAdapterController.executeStatement("statement");313        //THEN314        //Exception is thrown315    }316    @Test317    public void testExecuteUpdate(){318        //GIVEN319        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);320        final Message errorMessage = mock(Message.class);321        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_ROWS_UPDATED)).thenReturn("2");322        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());323        //WHEN324        final int rowsUpdated = jdbcEndpointAdapterController.executeUpdate("statement");325        //THEN326        verify(jdbcEndpointAdapterController).handleMessage(any());327        assertEquals(rowsUpdated, 2);328    }329    @Test(expectedExceptions = JdbcServerException.class)330    public void testExecuteUpdateWithFailure(){331        //GIVEN332        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);333        final Message errorMessage = mock(Message.class);334        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");335        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());336        //WHEN337        jdbcEndpointAdapterController.executeUpdate("statement");338        //THEN339        //Exception is thrown340    }341    @Test342    public void testCloseStatementWithAutoCreateStatement(){343        //GIVEN344        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);345        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(true);346        //WHEN347        jdbcEndpointAdapterController.closeStatement();348        //THEN349        verify(jdbcEndpointAdapterController, never()).handleMessage(any());350    }351    @Test352    public void testCloseStatementWithoutAutoCreateStatement(){353        //GIVEN354        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);355        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(false);356        //WHEN357        jdbcEndpointAdapterController.closeStatement();358        //THEN359        verify(jdbcEndpointAdapterController).handleMessage(any());360    }361    @Test(expectedExceptions = JdbcServerException.class)362    public void testCloseStatementWithoutAutoCreateStatementAndFailure(){363        //GIVEN364        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);365        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(false);366        final Message errorMessage = mock(Message.class);367        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");368        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());369        //WHEN370        jdbcEndpointAdapterController.closeStatement();371        //THEN372        //Exception is thrown373    }374    @Test375    public void testSetTransactionState(){376        //GIVEN377        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);378        final boolean expectedBoolean = new Random().nextBoolean();379        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(true);380        //WHEN381        jdbcEndpointAdapterController.setTransactionState(expectedBoolean);382        //THEN383        verify(jdbcEndpointAdapterController, never()).handleMessage(any());384        assertEquals(jdbcEndpointAdapterController.getTransactionState(), expectedBoolean);385    }386    @Test387    public void testSetTransactionStateWithoutAutoTransactionHandling(){388        //GIVEN389        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);390        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(false);391        //WHEN392        jdbcEndpointAdapterController.setTransactionState(true);393        //THEN394        verify(jdbcEndpointAdapterController).handleMessage(any());395    }396    @Test397    public void testSetTransactionStateVerifyMessageOnlyIfTransactionHasBeenStarted(){398        //GIVEN399        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);400        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(false);401        //WHEN402        jdbcEndpointAdapterController.setTransactionState(false);403        //THEN404        verify(jdbcEndpointAdapterController, never()).handleMessage(any());405    }406    @Test407    public void testCommitStatementsWithAutoCreateStatement(){408        //GIVEN409        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);410        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(true);411        //WHEN412        jdbcEndpointAdapterController.commitStatements();413        //THEN414        verify(jdbcEndpointAdapterController, never()).handleMessage(any());415    }416    @Test417    public void testCommitStatementsWithoutAutoCreateStatement(){418        //GIVEN419        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);420        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(false);421        //WHEN422        jdbcEndpointAdapterController.commitStatements();423        //THEN424        verify(jdbcEndpointAdapterController).handleMessage(any());425    }426    @Test(expectedExceptions = JdbcServerException.class)427    public void testCommitStatementsWithoutAutoCreateStatementAndFailure(){428        //GIVEN429        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);430        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(false);431        final Message errorMessage = mock(Message.class);432        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");433        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());434        //WHEN435        jdbcEndpointAdapterController.commitStatements();436        //THEN437        //Exception is thrown438    }439    @Test440    public void testRollbackStatementsWithAutoCreateStatement(){441        //GIVEN442        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);443        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(true);444        //WHEN445        jdbcEndpointAdapterController.rollbackStatements();446        //THEN447        verify(jdbcEndpointAdapterController, never()).handleMessage(any());448    }449    @Test450    public void testRollbackStatementsWithoutAutoCreateStatement(){451        //GIVEN452        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);453        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(false);454        //WHEN455        jdbcEndpointAdapterController.rollbackStatements();456        //THEN457        verify(jdbcEndpointAdapterController).handleMessage(any());458    }459    @Test(expectedExceptions = JdbcServerException.class)460    public void testRollbackStatementsWithoutAutoCreateStatementAndFailure(){461        //GIVEN462        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);463        when(jdbcEndpointConfiguration.isAutoTransactionHandling()).thenReturn(false);464        final Message errorMessage = mock(Message.class);465        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");466        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());467        //WHEN468        jdbcEndpointAdapterController.rollbackStatements();469        //THEN470        //Exception is thrown471    }472    @Test473    public void testCreateCallableStatementWithAutoCreateStatement(){474        //GIVEN475        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);476        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(true);477        //WHEN478        jdbcEndpointAdapterController.createCallableStatement("some statement");479        //THEN480        verify(jdbcEndpointAdapterController, never()).handleMessage(any());481    }482    @Test483    public void testCreateCallableStatementWithoutAutoCreateStatement(){484        //GIVEN485        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);486        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(false);487        //WHEN488        jdbcEndpointAdapterController.createCallableStatement("some statement");489        //THEN490        verify(jdbcEndpointAdapterController).handleMessage(any());491    }492    @Test(expectedExceptions = JdbcServerException.class)493    public void testCreateCallableStatementWithoutAutoCreateStatementAndFailure(){494        //GIVEN495        final JdbcEndpointAdapterController jdbcEndpointAdapterController = spy(this.jdbcEndpointAdapterController);496        when(jdbcEndpointConfiguration.isAutoCreateStatement()).thenReturn(false);497        final Message errorMessage = mock(Message.class);498        when(errorMessage.getHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS)).thenReturn("false");499        doReturn(errorMessage).when(jdbcEndpointAdapterController).handleMessage(any());500        //WHEN501        jdbcEndpointAdapterController.createCallableStatement("some statement");502        //THEN503        //Exception is thrown504    }505    @Test506    public void testHandleMessageWithAutoHandleQueriesEmptyOperation(){507        //GIVEN508        final Message request = mock(Message.class);509        when(request.getPayload(Operation.class)).thenReturn(null);510        final Message expectedResponse = mock(Message.class);;511        when(endpointAdapter.handleMessage(request)).thenReturn(expectedResponse);512        //WHEN...

Source:JdbcMessage.java

...91    private static JdbcMessage result(boolean success) {92        OperationResult operationResult = new OperationResult();93        operationResult.setSuccess(success);94        JdbcMessage message = new JdbcMessage(operationResult);95        message.setHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS, success);96        return message;97    }98    public static JdbcMessage result(OperationResult operationResult) {99        JdbcMessage message = new JdbcMessage(operationResult);100        message.setHeader(JdbcMessageHeaders.JDBC_SERVER_SUCCESS, operationResult.isSuccess());101        if (!operationResult.isSuccess()) {102            message.exception(operationResult.getException());103        }104        return message;105    }106    public JdbcMessage exception(String message) {107        if (operationResult == null) {108            throw new CitrusRuntimeException("Invalid access to operation result exception for JDBC message");109        }110        if (operationResult.isSuccess()) {111            throw new CitrusRuntimeException("Unable to set operation result exception on 'success' JDBC result");112        }113        setHeader(JdbcMessageHeaders.JDBC_SERVER_EXCEPTION, message);114        return this;115    }116    public JdbcMessage rowsUpdated(int number) {117        if (operationResult == null) {118            throw new CitrusRuntimeException("Invalid access to operation result exception for JDBC message");119        }120        operationResult.setAffectedRows(number);121        setHeader(JdbcMessageHeaders.JDBC_ROWS_UPDATED, number);122        return this;123    }124    public JdbcMessage dataSet(String dataSet) {125        if (operationResult == null) {126            throw new CitrusRuntimeException("Invalid access to operation result exception for JDBC message");127        }128        operationResult.setDataSet(dataSet);129        return this;130    }131    public JdbcMessage dataSet(Resource dataSet) {132        try {133            dataSet(FileUtils.readToString(dataSet));134        } catch (IOException e) {135            throw new CitrusRuntimeException("Failed to read data set file", e);...

Source:JdbcMessageHeaders.java

...18/**19 * @author Christoph Deppisch20 * @since 2.7.421 */22public abstract class JdbcMessageHeaders {23    /**24     * Prevent instantiation.25     */26    private JdbcMessageHeaders() {27    }28    /** Special header prefix for jdbc transport headers */29    public static final String JDBC_PREFIX = MessageHeaders.PREFIX + "jdbc_";30    public static final String JDBC_SERVER_PREFIX = JDBC_PREFIX + "server_";31    public static final String JDBC_ROWS_UPDATED = JDBC_PREFIX + "rows_updated";32    public static final String JDBC_SERVER_SUCCESS = JDBC_SERVER_PREFIX + "success";33    public static final String JDBC_SERVER_EXCEPTION = JDBC_SERVER_PREFIX + "exception";34}...

JdbcMessageHeaders

Using AI Code Generation

1import com.consol.citrus.jdbc.message.JdbcMessageHeaders;2import com.consol.citrus.message.MessageHeaders;3import com.consol.citrus.message.MessageType;4import com.consol.citrus.message.builder.DefaultMessageBuilder;5import com.consol.citrus.message.builder.ObjectPayloadMessageBuilder;6import com.consol.citrus.message.builder.PayloadTemplateMessageBuilder;7import com.consol.citrus.message.builder.StaticMessageContentBuilder;8import com.consol.citrus.message.builder.TextMessageBuilder;9import com.consol.citrus.message.builder.XMLMessageBuilder;10import com.consol.citrus.message.builder.XpathMessageBuilder;11import com.consol.citrus.message.builder.XpathPayloadTemplateMessageBuilder;12import com.consol.citrus.message.builder.XpathResultMessageBuilder;13import com.consol.citrus.message.builder.XpathResultPayloadTemplateMessageBuilder;14import com.consol.citrus.message.selector.MessageSelectorBuilder;15import com.consol.citrus.message.selector.ObjectMessageSelectorBuilder;16import com.consol.citrus.message.selector.XPathMessageSelectorBuilder;17import java.util.HashMap;18import java.util.Map;19import org.springframework.core.io.ClassPathResource;20import org.springframework.core.io.Resource;21import org.springframework.oxm.Marshaller;22import org.springframework.oxm.xstream.XStreamMarshaller;23import org.springframework.xml.transform.StringSource;24import org.springframework.xml.transform.StringResult;25import org.springframework.xml.transform.StringSource;26import org.springframework.xml.transform.StringResult;27import org.springframework.util.StringUtils;28import org.springframework.util.Assert;29import com.consol.citrus.context.TestContext;30import com.consol.citrus.exceptions.CitrusRuntimeException;31import com.consol.citrus.message.Message;32import com.consol.citrus.message.MessageHeaderType;33import com.consol.citrus.message.MessageHeaders;34import com.consol.citrus.message.MessageType;35import com.consol.citrus.message.builder.DefaultMessageBuilder;36import com.consol.citrus.message.builder.ObjectPayloadMessageBuilder;37import com.consol.citrus.message.builder.PayloadTemplateMessageBuilder;38import com.consol.citrus.message.builder.StaticMessageContentBuilder;39import com.consol.citrus.message.builder.TextMessageBuilder;40import com.consol.citrus.message.builder.XMLMessageBuilder;41import com.consol.citrus.message.builder.XpathMessageBuilder;42import com.consol.citrus.message.builder.XpathPayloadTemplateMessageBuilder;43import com.consol.cit

JdbcMessageHeaders

Using AI Code Generation

1package com.consol.citrus.jdbc.message;2import java.util.HashMap;3import java.util.Map;4import org.testng.Assert;5import org.testng.annotations.Test;6public class JdbcMessageHeadersTest {7    public void testJdbcMessageHeaders() {8        Map<String, Object> headersMap = new HashMap<String, Object>();9        headersMap.put("header1", "value1");10        headersMap.put("header2", "value2");11        headersMap.put("header3", "value3");12        JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders(headersMap);13        Assert.assertEquals(jdbcMessageHeaders.get("header1"), "value1");14        Assert.assertEquals(jdbcMessageHeaders.get("header2"), "value2");15        Assert.assertEquals(jdbcMessageHeaders.get("header3"), "value3");16    }17}

JdbcMessageHeaders

Using AI Code Generation

1package com.consol.citrus.jdbc.message;2import org.testng.Assert;3import org.testng.annotations.Test;4import org.springframework.jdbc.core.JdbcTemplate;5import org.springframework.jdbc.datasource.DriverManagerDataSource;6import java.util.List;7import java.util.Map;8public class JdbcMessageHeadersTest {9    public void testJdbcMessageHeaders() {10        DriverManagerDataSource dataSource = new DriverManagerDataSource();11        dataSource.setDriverClassName("org.hsqldb.jdbcDriver");12        dataSource.setUrl("jdbc:hsqldb:mem:testdb");13        dataSource.setUsername("sa");14        dataSource.setPassword("");15        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);16        jdbcTemplate.execute("CREATE TABLE TEST_TABLE(ID INT, NAME VARCHAR(20))");17        jdbcTemplate.execute("INSERT INTO TEST_TABLE VALUES(1, 'TestName1')");18        jdbcTemplate.execute("INSERT INTO TEST_TABLE VALUES(2, 'TestName2')");19        jdbcTemplate.execute("INSERT INTO TEST_TABLE VALUES(3, 'TestName3')");20        JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders(dataSource, "SELECT * FROM TEST_TABLE");21        List<Map<String, Object>> result = jdbcMessageHeaders.getHeaders();22        Assert.assertEquals(result.size(), 3);23        Assert.assertEquals(result.get(0).get("ID"), 1);24        Assert.assertEquals(result.get(0).get("NAME"), "TestName1");25        Assert.assertEquals(result.get(1).get("ID"), 2);26        Assert.assertEquals(result.get(1).get("NAME"), "TestName2");27        Assert.assertEquals(result.get(2).get("ID"), 3);28        Assert.assertEquals(result.get(2).get("NAME"), "TestName3");29    }30}31package com.consol.citrus.jdbc.message;32import org.testng.Assert;33import org.testng.annotations.Test;34import org.springframework.jdbc.core.JdbcTemplate;35import org.springframework.jdbc.datasource.DriverManagerDataSource;36import java.util.List;37import java.util.Map;38public class JdbcMessageHeadersTest {39    public void testJdbcMessageHeaders() {40        DriverManagerDataSource dataSource = new DriverManagerDataSource();41        dataSource.setDriverClassName("org.hsqldb.jdbcDriver");42        dataSource.setUrl("jdbc:hsqldb:mem:testdb");43        dataSource.setUsername("sa");44        dataSource.setPassword("");45        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);

JdbcMessageHeaders

Using AI Code Generation

1public class JdbcMessageHeadersTest {2    public static void main(String[] args) {3        JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();4        jdbcMessageHeaders.setDatabaseProductName("databaseProductName");5        jdbcMessageHeaders.setDatabaseProductVersion("databaseProductVersion");6        jdbcMessageHeaders.setDriverName("driverName");7        jdbcMessageHeaders.setDriverVersion("driverVersion");8        jdbcMessageHeaders.setJdbcUrl("jdbcUrl");9        jdbcMessageHeaders.setUserName("userName");10        jdbcMessageHeaders.setSql("sql");11        jdbcMessageHeaders.setUpdateCount(1);12        jdbcMessageHeaders.setTableName("tableName");13        jdbcMessageHeaders.setTableType("tableType");14        jdbcMessageHeaders.setSchemaName("schemaName");15        jdbcMessageHeaders.setCatalogName("catalogName");16        jdbcMessageHeaders.setColumnName("columnName");17        jdbcMessageHeaders.setColumnType(1);18        jdbcMessageHeaders.setColumnTypeName("columnTypeName");19        jdbcMessageHeaders.setColumnClassName("columnClassName");20        jdbcMessageHeaders.setColumnDisplaySize(1);21        jdbcMessageHeaders.setColumnPrecision(1);22        jdbcMessageHeaders.setColumnScale(1);23        jdbcMessageHeaders.setNullable(1);24        jdbcMessageHeaders.setCaseSensitive(false);25        jdbcMessageHeaders.setCurrency(false);26        jdbcMessageHeaders.setAutoIncrement(false);27        jdbcMessageHeaders.setReadOnly(false);28        jdbcMessageHeaders.setSearchable(false);29        jdbcMessageHeaders.setSigned(false);30        jdbcMessageHeaders.setWritable(false);31        jdbcMessageHeaders.setDefinitelyWritable(false);32        jdbcMessageHeaders.setIndexName("indexName");33        jdbcMessageHeaders.setIndexType(1);34        jdbcMessageHeaders.setIndexColumnName("indexColumnName");35        jdbcMessageHeaders.setIndexColumnSequence(1);36        jdbcMessageHeaders.setPrimaryKeyName("primaryKeyName");37        jdbcMessageHeaders.setPrimaryKeyColumnName("primaryKeyColumnName");38        jdbcMessageHeaders.setPrimaryKeyColumnSequence(1);39        jdbcMessageHeaders.setForeignKeyName("foreignKeyName");40        jdbcMessageHeaders.setForeignKeyColumnName("foreignKeyColumnName");41        jdbcMessageHeaders.setForeignKeyColumnSequence(1);42        jdbcMessageHeaders.setForeignKeyTableName("foreignKeyTableName");43        jdbcMessageHeaders.setForeignKeyTableSchema("foreignKeyTableSchema");44        jdbcMessageHeaders.setForeignKeyTableCatalog("foreignKeyTableCatalog");45        jdbcMessageHeaders.setForeignKeyColumnIndex(1);46        jdbcMessageHeaders.setForeignKeyColumnSequence(1);47        jdbcMessageHeaders.setForeignKeyColumnUpdateRule(1);

JdbcMessageHeaders

Using AI Code Generation

1public class JdbcMessageHeadersTest {2    public void testJdbcMessageHeaders() {3        JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();4        jdbcMessageHeaders.setHeaders(Arrays.asList("header1", "header2"));5        Assert.assertEquals(jdbcMessageHeaders.getHeaders(), Arrays.asList("header1", "header2"));6    }7}8public class JdbcMessageHeadersTest {9    public void testJdbcMessageHeaders() {10        JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();11        jdbcMessageHeaders.setHeaders(Arrays.asList("header1", "header2"));12        Assert.assertEquals(jdbcMessageHeaders.getHeaders(), Arrays.asList("header1", "header2"));13    }14}15public class JdbcMessageHeadersTest {16    public void testJdbcMessageHeaders() {17        JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();18        jdbcMessageHeaders.setHeaders(Arrays.asList("header1", "header2"));19        Assert.assertEquals(jdbcMessageHeaders.getHeaders(), Arrays.asList("header1", "header2"));20    }21}22public class JdbcMessageHeadersTest {23    public void testJdbcMessageHeaders() {24        JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();25        jdbcMessageHeaders.setHeaders(Arrays.asList("header1", "header2"));26        Assert.assertEquals(jdbcMessageHeaders.getHeaders(), Arrays.asList("header1", "header2"));27    }28}29public class JdbcMessageHeadersTest {30    public void testJdbcMessageHeaders() {31        JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();32        jdbcMessageHeaders.setHeaders(Arrays.asList("header1", "header2"));33        Assert.assertEquals(jdbcMessageHeaders.getHeaders(), Arrays.asList("header1", "

JdbcMessageHeaders

Using AI Code Generation

1public void testJdbcMessageHeaders() {2    JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();3    Map<String, Object> headers = jdbcMessageHeaders.getHeaders();4    System.out.println("Headers: " + headers);5}6public void testJdbcMessageHeaders() {7    JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();8    Map<String, Object> headers = jdbcMessageHeaders.getHeaders();9    System.out.println("Headers: " + headers);10}11public void testJdbcMessageHeaders() {12    JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();13    Map<String, Object> headers = jdbcMessageHeaders.getHeaders();14    System.out.println("Headers: " + headers);15}16public void testJdbcMessageHeaders() {17    JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();18    Map<String, Object> headers = jdbcMessageHeaders.getHeaders();19    System.out.println("Headers: " + headers);20}21public void testJdbcMessageHeaders() {22    JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();23    Map<String, Object> headers = jdbcMessageHeaders.getHeaders();24    System.out.println("Headers: " + headers);25}26public void testJdbcMessageHeaders() {27    JdbcMessageHeaders jdbcMessageHeaders = new JdbcMessageHeaders();

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.