Test handling consume packets

skycoin · Dec 12, 2019 · e6dc9a2 · e6dc9a2
1 parent 5b7a985
commit e6dc9a2
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/pkg/router/router_test.go b/pkg/router/router_test.go
@@ -290,14 +290,25 @@ func TestRouter_handleTransportPacket(t *testing.T) {
 			Reverse: cnsmRule,
 		})
 
-		packet := routing.MakeDataPacket(intFwdRtID[0], []byte("test"))
+		packet := routing.MakeDataPacket(intFwdRtID[0], []byte("test intermediary forward"))
 		require.NoError(t, r0.handleTransportPacket(context.TODO(), packet))
 
 		recvPacket, err := r1.tm.ReadPacket()
 		assert.NoError(t, err)
 		assert.Equal(t, packet.Size(), recvPacket.Size())
 		assert.Equal(t, packet.Payload(), recvPacket.Payload())
 		assert.Equal(t, dstRtIDs[1], recvPacket.RouteID())
+
+		consumeMsg := []byte("test_consume")
+		packet = routing.MakeDataPacket(dstRtIDs[1], consumeMsg)
+		require.NoError(t, r1.handleTransportPacket(context.TODO(), packet))
+
+		rg, ok := r1.routeGroup(fwdRtDesc.Invert())
+		require.True(t, ok)
+		require.NotNil(t, rg)
+
+		data := <-rg.readCh
+		require.Equal(t, consumeMsg, data)
 	})
 
 	t.Run("handlePacket_close", func(t *testing.T) {