public void should_not_send_messages_to_persistence_twice_if_persistence_goes_up_and_down() { using (MessageId.PauseIdGeneration()) { PeerDirectory.Setup(directory => directory.GetPeersHandlingMessage(new MessageBinding(new MessageTypeId(typeof(PersistMessageCommand)), BindingKey.Empty))) .Returns(new List <Peer> { PersistencePeer }); // Stopping persistence InnerTransport.RaiseMessageReceived(new TransportMessage(MessageTypeId.PersistenceStopping, new byte[0], PersistencePeer)); InnerTransport.Messages.Clear(); var ackedMessage = new FakeCommand(456).ToTransportMessage(); Transport.AckMessage(ackedMessage); InnerTransport.AckedMessages.ShouldBeEmpty(); // starting persistence - should send enqueued messages Transport.OnPeerUpdated(PersistencePeer.Id, PeerUpdateAction.Started); InnerTransport.ExpectExactly(new[] { new TransportMessageSent(new MessageHandled(ackedMessage.Id).ToTransportMessage(Self), PersistencePeer) }); // Stopping persistence again InnerTransport.RaiseMessageReceived(new TransportMessage(MessageTypeId.PersistenceStopping, new byte[0], PersistencePeer)); InnerTransport.Messages.Clear(); // starting persistence again - should not have anything to send Transport.OnPeerUpdated(PersistencePeer.Id, PeerUpdateAction.Started); InnerTransport.ExpectNothing(); } }
public void should_not_lose_messages_when_persistent_transport_goes_down_and_comes_back_up() { using (MessageId.PauseIdGeneration()) { PeerDirectory.Setup(directory => directory.GetPeersHandlingMessage(new MessageBinding(new MessageTypeId(typeof(PersistMessageCommand)), BindingKey.Empty))) .Returns(new List <Peer> { PersistencePeer }); // Stopping persistence InnerTransport.RaiseMessageReceived(new TransportMessage(MessageTypeId.PersistenceStopping, new byte[0], PersistencePeer)); InnerTransport.ExpectExactly(new TransportMessageSent(new TransportMessage(MessageTypeId.PersistenceStoppingAck, new byte[0], Self), PersistencePeer)); InnerTransport.Messages.Clear(); // should enqueue messages to persistence var message = new FakeCommand(123).ToTransportMessage(); var ackedMessage = new FakeCommand(456).ToTransportMessage(); Transport.Send(message, new[] { AnotherPersistentPeer }); Transport.AckMessage(ackedMessage); InnerTransport.AckedMessages.ShouldBeEmpty(); InnerTransport.ExpectExactly(new TransportMessageSent(message, AnotherPersistentPeer)); InnerTransport.Messages.Clear(); // starting persistence - should send enqueued messages Transport.OnPeerUpdated(PersistencePeer.Id, PeerUpdateAction.Started); InnerTransport.ExpectExactly(new[] { new TransportMessageSent(new PersistMessageCommand(message, new[] { AnotherPersistentPeer.Id }).ToTransportMessage(Self), PersistencePeer), new TransportMessageSent(new MessageHandled(ackedMessage.Id).ToTransportMessage(Self), PersistencePeer) }); InnerTransport.Messages.Clear(); InnerTransport.AckedMessages.Clear(); // should send messages without going through the queue Transport.Send(message, new[] { AnotherPersistentPeer }); Transport.AckMessage(ackedMessage); InnerTransport.ExpectExactly(new[] { new TransportMessageSent(message, AnotherPersistentPeer), new TransportMessageSent(new PersistMessageCommand(message, new[] { AnotherPersistentPeer.Id }).ToTransportMessage(Self), PersistencePeer), new TransportMessageSent(new MessageHandled(ackedMessage.Id).ToTransportMessage(Self), PersistencePeer), }); } }
public void should_discard_messages_waiting_for_persistence_on_stop() { using (MessageId.PauseIdGeneration()) { Transport.Start(); PeerDirectory.Setup(directory => directory.GetPeersHandlingMessage(new MessageBinding(MessageTypeId.PersistenceStoppingAck, BindingKey.Empty))) .Returns(new List <Peer> { PersistencePeer }); // Stopping persistence InnerTransport.RaiseMessageReceived(new TransportMessage(MessageTypeId.PersistenceStopping, new byte[0], PersistencePeer)); InnerTransport.Messages.Clear(); var ackedMessage = new FakeCommand(456).ToTransportMessage(); Transport.AckMessage(ackedMessage); Transport.Stop(); Transport.Start(); } }