Very basic fake implementation of ConfirmSelect and WaitForConfirms* methods (no messages counting)

Author: odalet

src/AddUp.FakeRabbitMQ.Tests/FakeModelTests.cs

@@ -1,6 +1,6 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
 using System.Diagnostics.CodeAnalysis;
-using System.Globalization;
 using System.Linq;
 using System.Text;
 using RabbitMQ.Client;
@@ -761,5 +761,57 @@ public void BasicGet_Should_Not_Remove_The_Message_From_Queue_If_Not_Acked(bool
                 Assert.Equal(expectedMessageCount, model.WorkingMessages.Count);
             }
         }
+
+        [Fact]
+        public void WaitForConfirms_should_throw_if_ConfirmSelect_was_not_called()
+        {
+            // arrange
+            var node = new RabbitServer();
+            using (var model = new FakeModel(node))
+                // act & assert
+                Assert.Throws<InvalidOperationException>(() => model.WaitForConfirms());
+        }
+
+        [Fact]
+        public void WaitForConfirmsOrDie_returns_true_if_ConfirmSelect_was_called()
+        {
+            // arrange
+            var node = new RabbitServer();
+            using (var model = new FakeModel(node))
+            {
+                // act
+                model.ConfirmSelect();
+                var result = model.WaitForConfirms();
+
+                // assert
+                Assert.True(result);
+            }
+        }
+
+        [Fact]
+        public void WaitForConfirmsOrDie_should_throw_if_ConfirmSelect_was_not_called()
+        {
+            // arrange
+            var node = new RabbitServer();
+            using (var model = new FakeModel(node))
+                // act & assert
+                Assert.Throws<InvalidOperationException>(() => model.WaitForConfirmsOrDie());
+        }
+
+        [Fact]
+        public void WaitForConfirmsOrDie_does_not_throw_if_ConfirmSelect_was_called()
+        {
+            // arrange
+            var node = new RabbitServer();
+            using (var model = new FakeModel(node))
+            {
+                // act
+                model.ConfirmSelect();
+                model.WaitForConfirmsOrDie();
+            }
+
+            // assert
+            Assert.True(true);
+        }
     }
 }

src/AddUp.FakeRabbitMQ/FakeModel.cs

@@ -15,6 +15,7 @@ internal sealed class FakeModel : IModel
         private readonly ConcurrentDictionary<string, IBasicConsumer> consumers = new ConcurrentDictionary<string, IBasicConsumer>();
         private readonly RabbitServer server;
         private long lastDeliveryTag;
+        private bool confirmModeSelected;
 
         public FakeModel(RabbitServer rabbitServer) => server = rabbitServer;
 
@@ -173,12 +174,21 @@ public uint QueueDelete(string queue, bool ifUnused, bool ifEmpty)
             return instance != null ? 1u : 0u;
         }
 
-        public void ConfirmSelect() => throw new NotImplementedException();
-        public bool WaitForConfirms() => throw new NotImplementedException();
-        public bool WaitForConfirms(TimeSpan timeout) => throw new NotImplementedException();
-        public bool WaitForConfirms(TimeSpan timeout, out bool timedOut) => throw new NotImplementedException();
-        public void WaitForConfirmsOrDie() => throw new NotImplementedException();
-        public void WaitForConfirmsOrDie(TimeSpan timeout) => throw new NotImplementedException();
+        public void ConfirmSelect() => confirmModeSelected = true;
+
+        public void WaitForConfirmsOrDie() => WaitForConfirmsOrDie(TimeSpan.Zero);
+        public void WaitForConfirmsOrDie(TimeSpan timeout) => _ = WaitForConfirms(timeout);
+
+        public bool WaitForConfirms() => WaitForConfirms(TimeSpan.Zero);
+        public bool WaitForConfirms(TimeSpan timeout) => WaitForConfirms(timeout, out _);
+        public bool WaitForConfirms(TimeSpan timeout, out bool timedOut)
+        {
+            if (!confirmModeSelected)
+                throw new InvalidOperationException("Confirms not selected");
+
+            timedOut = false;
+            return true;
+        }
 
         public string BasicConsume(string queue, bool autoAck, IBasicConsumer consumer) => BasicConsume(queue, autoAck, Guid.NewGuid().ToString(), true, false, null, consumer);
         public string BasicConsume(string queue, bool autoAck, string consumerTag, IBasicConsumer consumer) => BasicConsume(queue, autoAck, consumerTag, true, false, null, consumer);