Added IT to verify statemachine v3

neo4j · Jul 20, 2018 · b7c2916 · b7c2916
1 parent 6b8a782
commit b7c2916
Show file tree

Hide file tree

Showing 3 changed files with 737 additions and 12 deletions.
diff --git a/community/bolt/src/main/java/org/neo4j/bolt/v3/runtime/ExtraMetaDataConnectedState.java b/community/bolt/src/main/java/org/neo4j/bolt/v3/runtime/ExtraMetaDataConnectedState.java
@@ -24,6 +24,7 @@
 import org.neo4j.bolt.runtime.BoltStateMachineState;
 import org.neo4j.bolt.runtime.StateMachineContext;
 import org.neo4j.bolt.v1.runtime.ConnectedState;
+import org.neo4j.bolt.v3.messaging.HelloMessage;
 import org.neo4j.values.storable.Values;
 
 /**
@@ -42,9 +43,13 @@ public class ExtraMetaDataConnectedState extends ConnectedState
     @Override
     public BoltStateMachineState process( RequestMessage message, StateMachineContext context ) throws BoltConnectionFatality
     {
-        BoltStateMachineState processResult = super.process( message, context );
-        context.connectionState().onMetadata( ROUTING_TABLE_KEY, Values.stringValue( ROUTING_TABLE_VALUE ) );
-        context.connectionState().onMetadata( CONNECTION_ID_KEY, Values.stringValue( context.connectionId() ) );
-        return processResult;
+        if ( message instanceof HelloMessage )
+        {
+            BoltStateMachineState processResult = super.process( message, context );
+            context.connectionState().onMetadata( ROUTING_TABLE_KEY, Values.stringValue( ROUTING_TABLE_VALUE ) );
+            context.connectionState().onMetadata( CONNECTION_ID_KEY, Values.stringValue( context.connectionId() ) );
+            return processResult;
+        }
+        return null;
     }
 }