Extra validation for command and query

smbadiwe · Aug 12, 2016 · 617d736 · 617d736
1 parent 64cefa1
commit 617d736
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/MultiTenancyFramework.Core/Commands/CommandProcessor.cs b/MultiTenancyFramework.Core/Commands/CommandProcessor.cs
@@ -14,6 +14,8 @@ public CommandProcessor(IServiceProvider serviceProvider)
         [DebuggerStepThrough]
         public void Process(ICommand command)
         {
+            if (command == null) throw new ArgumentNullException("command");
+
             var handlerType =
                 typeof(ICommandHandler<>).MakeGenericType(command.GetType());
 

diff --git a/MultiTenancyFramework.Core/Data/Queries/DbQueryProcessor.cs b/MultiTenancyFramework.Core/Data/Queries/DbQueryProcessor.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Diagnostics;
 
 namespace MultiTenancyFramework.Data.Queries
 {
@@ -12,9 +13,11 @@ public DbQueryProcessor(IServiceProvider serviceProvider)
 
         public string InstitutionCode { get; set; }
 
-        //[DebuggerStepThrough]
+        [DebuggerStepThrough]
         public TResult Process<TResult>(IDbQuery<TResult> query)
         {
+            if (query == null) throw new ArgumentNullException("query");
+
             var handlerType =
                 typeof(IDbQueryHandler<,>).MakeGenericType(query.GetType(), typeof(TResult));