Examples with BoltProtocol org.neo4j.bolt.BoltProtocol used on opensource projects

Example 6 with BoltProtocol

use of org.neo4j.bolt.BoltProtocol in project neo4j by neo4j.

the class ProtocolHandshakerTest method shouldFailOutOfRangeProtocol.

@ParameterizedTest
@MethodSource("protocolVersionProviderOutOfRange")
void shouldFailOutOfRangeProtocol(int major, int minor) {
    // Given
    int noVersion = 0;
    BoltProtocol protocol = newBoltProtocol(major, minor);
    BoltProtocolFactory handlerFactory = newProtocolFactory(major, minor, protocol);
    var memoryTracker = mock(MemoryTracker.class);
    var scopedTracker = mock(MemoryTracker.class);
    when(memoryTracker.getScopedMemoryTracker()).thenReturn(scopedTracker);
    EmbeddedChannel channel = new EmbeddedChannel(new ProtocolHandshaker(handlerFactory, boltChannel, logProvider, false, true, mock(ChannelProtector.class), memoryTracker));
    // When
    ByteBuf input = // create handshake data
    Unpooled.wrappedBuffer(// preamble
    new byte[] { (byte) 0x60, (byte) 0x60, (byte) 0xB0, (byte) 0x17 }, // first choice - 4.0 no range
    new byte[] { 0, 0, 0, 4 }, // second choice - 4.4 -> 4.2 inclusive range of 2
    new byte[] { 0, 2, 4, 4 }, // No protocol
    new byte[] { 0, 0, 0, 0 }, // No protocol
    new byte[] { 0, 0, 0, 0 });
    channel.writeInbound(input);
    // Then
    assertEquals(1, channel.outboundMessages().size());
    assertByteBufEquals(Unpooled.buffer().writeInt(noVersion), channel.readOutbound());
    assertThrows(NoSuchElementException.class, () -> channel.pipeline().remove(ProtocolHandshaker.class));
    assertFalse(channel.isActive());
}

Example 7 with BoltProtocol

use of org.neo4j.bolt.BoltProtocol in project neo4j by neo4j.

the class ProtocolHandshakerTest method shouldHandleProtocolRanges.

@ParameterizedTest
@MethodSource("protocolVersionProviderInRange")
void shouldHandleProtocolRanges(int major, int minor) {
    // Given
    int packedVersion = new BoltProtocolVersion(major, minor).toInt();
    BoltProtocol protocol = newBoltProtocol(major, minor);
    BoltProtocolFactory handlerFactory = newProtocolFactory(major, minor, protocol);
    var memoryTracker = mock(MemoryTracker.class, RETURNS_MOCKS);
    EmbeddedChannel channel = new EmbeddedChannel(new ProtocolHandshaker(handlerFactory, boltChannel, logProvider, false, true, mock(ChannelProtector.class), memoryTracker));
    // When
    ByteBuf input = // create handshake data
    Unpooled.wrappedBuffer(// preamble
    new byte[] { (byte) 0x60, (byte) 0x60, (byte) 0xB0, (byte) 0x17 }, // first choice - 4.0 no range
    new byte[] { 0, 0, 0, 4 }, // second choice - 4.3 -> 4.1 inclusive range of 2
    new byte[] { 0, 2, 3, 4 }, // No protocol
    new byte[] { 0, 0, 0, 0 }, // No protocol
    new byte[] { 0, 0, 0, 0 });
    channel.writeInbound(input);
    // Then
    assertEquals(1, channel.outboundMessages().size());
    assertByteBufEquals(Unpooled.buffer().writeInt(packedVersion), channel.readOutbound());
    assertThrows(NoSuchElementException.class, () -> channel.pipeline().remove(ProtocolHandshaker.class));
    assertTrue(channel.isActive());
}

Example 8 with BoltProtocol

use of org.neo4j.bolt.BoltProtocol in project neo4j by neo4j.

the class ProtocolHandshakerTest method newBoltProtocol.

private static BoltProtocol newBoltProtocol(int majorVersion, int minorVersion) {
    BoltProtocol handler = mock(BoltProtocol.class);
    when(handler.version()).thenReturn(new BoltProtocolVersion(majorVersion, minorVersion));
    return handler;
}

Example 9 with BoltProtocol

use of org.neo4j.bolt.BoltProtocol in project neo4j by neo4j.

the class ProtocolHandshakerTest method shouldHandleFragmentedMessage.

@Test
void shouldHandleFragmentedMessage() {
    // Given
    BoltProtocol protocol = newBoltProtocol(1, 0);
    BoltProtocolFactory handlerFactory = newProtocolFactory(1, 0, protocol);
    var memoryTracker = mock(MemoryTracker.class, RETURNS_MOCKS);
    EmbeddedChannel channel = new EmbeddedChannel(new ProtocolHandshaker(handlerFactory, boltChannel, logProvider, false, true, mock(ChannelProtector.class), memoryTracker));
    // When
    channel.writeInbound(Unpooled.wrappedBuffer(new byte[] { (byte) 0x60, (byte) 0x60, (byte) 0xB0 }));
    assertEquals(0, channel.outboundMessages().size());
    channel.writeInbound(Unpooled.wrappedBuffer(new byte[] { (byte) 0x17, 0, 0, 0 }));
    assertEquals(0, channel.outboundMessages().size());
    channel.writeInbound(Unpooled.wrappedBuffer(new byte[] { 0, 0, 0 }));
    assertEquals(0, channel.outboundMessages().size());
    channel.writeInbound(Unpooled.wrappedBuffer(new byte[] { 0, 1, 0, 0, 0 }));
    assertEquals(0, channel.outboundMessages().size());
    channel.writeInbound(Unpooled.wrappedBuffer(new byte[] { 0, 0, 0 }));
    assertEquals(0, channel.outboundMessages().size());
    channel.writeInbound(Unpooled.wrappedBuffer(new byte[] { 0, 0 }));
    // Then
    assertEquals(1, channel.outboundMessages().size());
    assertByteBufEquals(Unpooled.buffer().writeInt(1), channel.readOutbound());
    assertThrows(NoSuchElementException.class, () -> channel.pipeline().remove(ProtocolHandshaker.class));
    assertTrue(channel.isActive());
    verify(protocol).install();
}

Example 10 with BoltProtocol

use of org.neo4j.bolt.BoltProtocol in project neo4j by neo4j.

the class ProtocolHandshakerTest method shouldFallbackToNoProtocolIfNoMatch.

@Test
void shouldFallbackToNoProtocolIfNoMatch() {
    // Given
    BoltProtocol protocol = newBoltProtocol(1, 0);
    BoltProtocolFactory handlerFactory = newProtocolFactory(1, 0, protocol);
    var memoryTracker = mock(MemoryTracker.class, RETURNS_MOCKS);
    EmbeddedChannel channel = new EmbeddedChannel(new ProtocolHandshaker(handlerFactory, boltChannel, logProvider, false, true, mock(ChannelProtector.class), memoryTracker));
    // When
    ByteBuf input = // create handshake data
    Unpooled.wrappedBuffer(// preamble
    new byte[] { (byte) 0x60, (byte) 0x60, (byte) 0xB0, (byte) 0x17 }, // first choice - no protocol
    new byte[] { 0, 0, 0, 0 }, // second choice - protocol 1
    new byte[] { 0, 0, 0, 2 }, // third choice - no protocol
    new byte[] { 0, 0, 0, 3 }, // fourth choice - no protocol
    new byte[] { 0, 0, 0, 4 });
    channel.writeInbound(input);
    // Then
    assertEquals(1, channel.outboundMessages().size());
    assertByteBufEquals(Unpooled.buffer().writeInt(0), channel.readOutbound());
    assertFalse(channel.isActive());
    verify(protocol, never()).install();
}