throw exceptions (#21)

Co-authored-by: Ben Wesson <[email protected]>

Author: wessonben

DbMocker.Tests/DatabaseCommandTests.cs

@@ -379,5 +379,48 @@ public void Mock_ExecuteDataset_WithFunction_Test()
 
         }
 
+        [TestMethod]
+        public void Mock_ThrowException_Test()
+        {
+            var conn = new MockDbConnection();
+
+            conn.Mocks
+                .WhenAny()
+                .Throws(new Exception("TestException"));
+
+            using (var cmd = new DatabaseCommand(conn))
+            {
+                var exception = Assert.ThrowsException<Exception>(() =>
+                {
+                    cmd.CommandText.AppendLine("SELECT ...");
+                    cmd.ExecuteScalar<int>();
+                });
+
+                Assert.IsNotNull(exception);
+                Assert.AreEqual("TestException", exception.Message);
+            }
+        }
+
+        [TestMethod]
+        public void Mock_ThrowGenericException_Test()
+        {
+            var conn = new MockDbConnection();
+
+            conn.Mocks
+                .WhenAny()
+                .Throws<Exception>();
+
+            using (var cmd = new DatabaseCommand(conn))
+            {
+                var exception = Assert.ThrowsException<Exception>(() =>
+                {
+                    cmd.CommandText.AppendLine("SELECT ...");
+                    cmd.ExecuteScalar<int>();
+                });
+
+                Assert.IsNotNull(exception);
+            }
+        }
+
     }
 }

DbMocker/MockReturns.cs

@@ -114,6 +114,24 @@ public void ReturnsRow(MockTable returns)
             ReturnsFunction = (cmd) => new[] { returns };
         }
 
+        /// <summary>
+        /// Throws the specified exception.
+        /// </summary>
+        /// <param name="exception">The exception.</param>
+        public void Throws(Exception exception)
+        {
+            ReturnsFunction = command => throw exception;
+        }
+
+        /// <summary>
+        /// Throws the specified exception type.
+        /// </summary>
+        /// <typeparam name="TException">The type of the exception.</typeparam>
+        public void Throws<TException>() where TException : Exception, new()
+        {
+            ReturnsFunction = command => throw new TException();
+        }
+
         /// <summary>
         /// Set output value for dbParameter, when the condition occured.
         /// </summary>