public void RemoveMessageShouldDoIt() { // Arrange _messageQueueHandler.AppendMessage(_message); var initialSize = _messageQueueHandler.MessagesCount; var messageToRemove = _message; // Act _messageQueueHandler.RemoveMessage(messageToRemove); var currentSize = _messageQueueHandler.MessagesCount; // Assert Assert.That(currentSize, Is.EqualTo(initialSize - 1)); }
private void ReplaceMessageInChannel(Channel channel, Message message, IMessageQueueHandler messageQueueHandler) { var isSuccess = TryMoveMessageToChannel(channel, message); messageQueueHandler.RemoveMessage(message); if (!isSuccess) { if (message.MessageType == MessageType.SendingRequest) { var sender = _network.GetNodeById(message.LastTransferNodeId); _messageReceiver.HandleReceivedMessage(sender, message); } else { messageQueueHandler.AddMessageInStart(message); } } else { _handledMessagesInNode.Add(message); } }