Ejemplo n.º 1
0
        public async Task ShouldCloseGracefully()
        {
            var transportConfiguration = new TransportConfiguration()
            {
                InboundEndPoint = "tcp://localhost:7979",
                PeerId          = new PeerId("TestTransport")
            };

            var receiver = new Peer(transportConfiguration.PeerId, transportConfiguration.InboundEndPoint);

            var serializer = new JsonMessageSerializer();
            var logger     = new MockLogger();
            var transport  = new Transport(serializer, logger);

            transport.Configure(transportConfiguration.PeerId, transportConfiguration.InboundEndPoint);

            transport.Start();

            transport.MessageReceived += (msg) =>
            {
                TestTransportContext.Increment();
            };

            var task = Task.Run(() =>
            {
                for (var i = 0; i < 100; i++)
                {
                    var testTransportMessage = new TestTransportMessage();
                    var message          = serializer.Serialize(testTransportMessage);
                    var transportMessage = new TransportMessage(new MessageTypeId(typeof(TestTransportMessage)), message, transportConfiguration.PeerId);
                    transport.Send(transportMessage, new[] { receiver });
                }
            });

            while (!task.IsCompleted)
            {
                await Task.Delay(200);
            }

            Assert.AreEqual(true, task.IsCompletedSuccessfully);

            transport.Stop();

            Assert.AreEqual(100, TestTransportContext.Get());
        }
Ejemplo n.º 2
0
        public async Task ShouldMakeARoundTrip()
        {
            var transportConfiguration = new TransportConfiguration()
            {
                InboundEndPoint = "tcp://localhost:7979",
                PeerId          = new PeerId($"{Guid.NewGuid()}")
            };

            var receiver = new Peer(transportConfiguration.PeerId, transportConfiguration.InboundEndPoint);

            var serializer = new JsonMessageSerializer();
            var logger     = new MockLogger();

            var transport = new Transport(serializer, logger);

            transport.Configure(transportConfiguration.PeerId, transportConfiguration.InboundEndPoint);

            var testTransportMessage = new TestTransportMessage();
            var message          = serializer.Serialize(testTransportMessage);
            var transportMessage = new TransportMessage(new MessageTypeId(typeof(TestTransportMessage)), message, transportConfiguration.PeerId);

            transport.Start();

            transport.MessageReceived += (msg) =>
            {
                if (TestTransportContext.Get() == 0)
                {
                    Assert.AreEqual(transportMessage.Id, msg.Id);
                    Assert.AreEqual(transportMessage.Sender, transportConfiguration.PeerId);

                    var testTransportMessage2 = serializer.Deserialize(msg.Content, Type.GetType(msg.MessageTypeId.FullName)) as TestTransportMessage;
                    Assert.AreEqual(testTransportMessage.Id, testTransportMessage2.Id);
                    Assert.AreEqual(testTransportMessage.Subject, testTransportMessage2.Subject);
                    Assert.AreEqual(testTransportMessage.Content, testTransportMessage2.Content);

                    TestTransportContext.Increment();

                    var ackMessage          = new TestTransportMessage(TestTransportMessage.AcknowledgeSubject);
                    var ackTransportMessage = new TransportMessage(new MessageTypeId(typeof(TestTransportMessage)), serializer.Serialize(ackMessage), transportConfiguration.PeerId);

                    transport.Send(ackTransportMessage, new[] { new Peer(msg.Sender, transportConfiguration.InboundEndPoint) });
                }

                else if (TestTransportContext.Get() == 1)
                {
                    var testTransportMessage2 = serializer.Deserialize(msg.Content, Type.GetType(msg.MessageTypeId.FullName)) as TestTransportMessage;
                    Assert.AreEqual(TestTransportMessage.AcknowledgeSubject, testTransportMessage2.Subject);

                    TestTransportContext.Increment();
                }
            };


            transport.Send(transportMessage, new[] { receiver });

            await Task.Delay(1000);

            Assert.AreEqual(2, TestTransportContext.Get());

            transport.Stop();
        }