Adds AllowMissing testing.

* See issue #1.
mrtryhard · Jul 16, 2018 · d5c34e5 · d5c34e5
1 parent 974548c
commit d5c34e5
Show file tree

Hide file tree

Showing 5 changed files with 57 additions and 8 deletions.
diff --git a/DbAutoFillNextUnitTest/Dataset/BasicObject.cs b/DbAutoFillNextUnitTest/Dataset/BasicObject.cs
@@ -1,7 +1,7 @@
 using DbAutoFillStandard.Types;
 using System.Collections.Generic;
 
-namespace DbAutoFillNextCoreUnitTest.Dataset
+namespace DbAutoFillStandardUnitTest.Dataset
 {
     [DbAutoFill(AllowMissing = false)]
     class BasicObject

diff --git a/DbAutoFillNextUnitTest/Dataset/ComplexObject.cs b/DbAutoFillNextUnitTest/Dataset/ComplexObject.cs
@@ -1,6 +1,6 @@
 using DbAutoFillStandard.Types;
 
-namespace DbAutoFillNextCoreUnitTest.Dataset
+namespace DbAutoFillStandardUnitTest.Dataset
 {
     [DbAutoFill(AllowMissing = false,
         ParameterPrefix = "p_")]

diff --git a/DbAutoFillNextUnitTest/Dataset/DatasetGenerator.cs b/DbAutoFillNextUnitTest/Dataset/DatasetGenerator.cs
@@ -1,7 +1,7 @@
 using System;
 using System.Data;
 
-namespace DbAutoFillNextCoreUnitTest.Dataset
+namespace DbAutoFillStandardUnitTest.Dataset
 {
     internal static class DatasetGenerator
     {

diff --git a/DbAutoFillNextUnitTest/Dataset/SampleObject.cs b/DbAutoFillNextUnitTest/Dataset/SampleObject.cs
@@ -0,0 +1,13 @@
+using DbAutoFillStandard.Types;
+
+namespace DbAutoFillStandardUnitTest.Dataset
+{
+    [DbAutoFill]
+    public class SampleObject
+    {
+        [DbAutoFill(AllowMissing = true)]
+        public int MissingField { get; set; }
+
+        public int Mandatory { get; set; }
+    }
+}
diff --git a/DbAutoFillNextUnitTest/DbAutoFillHelperTests.cs b/DbAutoFillNextUnitTest/DbAutoFillHelperTests.cs
@@ -1,6 +1,6 @@
-using DbAutoFillNextCoreUnitTest.Dataset;
 using DbAutoFillStandard;
 using DbAutoFillStandard.Types;
+using DbAutoFillStandardUnitTest.Dataset;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using System;
 using System.Data;
@@ -103,7 +103,7 @@ public void DbResultsToBasicObject()
             const string intPropName = "IntField";
 
             DataTable dt = DatasetGenerator.CreateNewBasicDataTable(
-                new string[] { stringPropName, intPropName }, 
+                new string[] { stringPropName, intPropName },
                 new Type[] { typeof(string), typeof(int) });
 
             using (IDataReader dr = DatasetGenerator.CreateBasicDataReader(dt, expectedStringValue, expectedIntValue))
@@ -136,9 +136,9 @@ public void DbResultToComplexObject()
                 new string[] { nameINField, toDbUuidField, fromDbIdField, expectedAliasedColumnName },
                 new Type[] { typeof(string), typeof(int), typeof(int), typeof(string) });
 
-            using (IDataReader dr = DatasetGenerator.CreateBasicDataReader(dt, 
-                expectedNameINValue, 
-                unexpectedToDbUuidValue, 
+            using (IDataReader dr = DatasetGenerator.CreateBasicDataReader(dt,
+                expectedNameINValue,
+                unexpectedToDbUuidValue,
                 expectedFromDbUuid,
                 expectedAliaseColumnValue))
             {
@@ -157,6 +157,42 @@ public void DbResultToComplexObject()
             }
         }
 
+        [TestMethod]
+        public void DbResultToSampleObjectAllowMissing()
+        {
+            const string mandatoryField = "Mandatory";
+            const int expectedMandatoryValue = 890;
+
+            {
+                DataTable dt = DatasetGenerator.CreateNewBasicDataTable(
+                    new string[] { mandatoryField },
+                    new Type[] { typeof(int) });
+
+                using (IDataReader dr = DatasetGenerator.CreateBasicDataReader(dt, expectedMandatoryValue))
+                {
+                    dr.Read();
+
+                    SampleObject obj = new SampleObject();
+
+                    DbAutoFillHelper.FillObjectFromDataReader(dr, obj);
+
+                    Assert.AreEqual(expectedMandatoryValue, obj.Mandatory);
+                }
+            }
+
+            {
+                DataTable dt = DatasetGenerator.CreateNewBasicDataTable(new string[] { }, new Type[] { });
+
+                using (IDataReader dr = DatasetGenerator.CreateBasicDataReader(dt, new object[] { }))
+                {
+                    dr.Read();
+                    SampleObject obj = new SampleObject();
+
+                    Assert.ThrowsException<MissingFieldException>(() => { DbAutoFillHelper.FillObjectFromDataReader(dr, obj); });
+                }
+            }
+        }
+
         [TestCleanup]
         public void TestCleanp()
         {