Search in sources :

Example 11 with BufferedChannelOutput

use of org.neo4j.bolt.packstream.BufferedChannelOutput in project neo4j by neo4j.

the class TransportErrorIT method shouldHandleUnknownMessages.

@ParameterizedTest(name = "{displayName} {2}")
@MethodSource("argumentsProvider")
public void shouldHandleUnknownMessages(Class<? extends TransportConnection> connectionClass, Neo4jPack neo4jPack, String name) throws Exception {
    initParameters(connectionClass, neo4jPack, name);
    // Given I send a message with an invalid type
    final RecordingByteChannel rawData = new RecordingByteChannel();
    final PackStream.Packer packer = new PackStream.Packer(new BufferedChannelOutput(rawData));
    // Invalid message type
    packer.packStructHeader(1, (byte) 0x66);
    packer.pack(1234);
    packer.flush();
    byte[] invalidMessage = rawData.getBytes();
    // When
    connection.connect(address).send(util.defaultAcceptedVersions()).send(util.chunk(32, invalidMessage));
    // Then
    assertThat(connection).satisfies(util.eventuallyReceivesSelectedProtocolVersion());
    assertThat(connection).satisfies(eventuallyDisconnects());
}
Also used : PackStream(org.neo4j.bolt.packstream.PackStream) RecordingByteChannel(org.neo4j.bolt.messaging.RecordingByteChannel) BufferedChannelOutput(org.neo4j.bolt.packstream.BufferedChannelOutput) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest) MethodSource(org.junit.jupiter.params.provider.MethodSource)

Aggregations

BufferedChannelOutput (org.neo4j.bolt.packstream.BufferedChannelOutput)11 RecordingByteChannel (org.neo4j.bolt.messaging.RecordingByteChannel)10 BoltRequestMessageWriter (org.neo4j.bolt.messaging.BoltRequestMessageWriter)6 Neo4jPack (org.neo4j.bolt.packstream.Neo4jPack)6 RequestMessage (org.neo4j.bolt.messaging.RequestMessage)5 ParameterizedTest (org.junit.jupiter.params.ParameterizedTest)3 MethodSource (org.junit.jupiter.params.provider.MethodSource)3 PackStream (org.neo4j.bolt.packstream.PackStream)3 BoltResponseMessageWriterV3 (org.neo4j.bolt.v3.messaging.BoltResponseMessageWriterV3)2 BoltResponseMessageWriter (org.neo4j.bolt.messaging.BoltResponseMessageWriter)1 ResponseMessage (org.neo4j.bolt.messaging.ResponseMessage)1 BufferedChannelInput (org.neo4j.bolt.packstream.BufferedChannelInput)1 BoltProtocolV3ComponentFactory.newNeo4jPack (org.neo4j.bolt.v3.BoltProtocolV3ComponentFactory.newNeo4jPack)1 BoltRequestMessageWriterV4 (org.neo4j.bolt.v4.BoltRequestMessageWriterV4)1