test

Groot/Groot.cs

@@ -61,7 +61,7 @@ public static IEnumerable<T> GetObjectFromCsv<T>(string filePath, bool autoMapFo
                          .ToList()
                          .ForEach(prop =>
                          {
-                             prop.SetValue(tElem, Convert.ChangeType(x.Value, prop.PropertyType), null);
+                             prop.SetValue(tElem, ChangeType(prop, x.Value), null);
                          });
 
                  }
@@ -70,6 +70,14 @@ public static IEnumerable<T> GetObjectFromCsv<T>(string filePath, bool autoMapFo
 
          }
 
-
+         private static object ChangeType(PropertyInfo prop, string value)
+         {
+            if (prop.PropertyType.IsEnum)
+              return Enum.Parse(prop.PropertyType, value);
+
+            return Convert.ChangeType(value, prop.PropertyType);
+         }
+
+
      }
  }

GrootUnitTest/GrootUnitTest.csproj

@@ -17,5 +17,8 @@
     <None Update="resources\grootTrees.csv">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Update="resources\item.csv">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
   </ItemGroup>
 </Project>

GrootUnitTest/models/Item.cs

@@ -0,0 +1,16 @@
+using Groot;
+
+namespace GrootUnitTest.models
+{
+    public class Item
+    {
+        [GrootField("name")]
+        public string Name { get; set; }
+
+        [GrootField("amount")]
+        public int Amount { get; set; }
+
+        [GrootField("type")]
+        public ItemType Type { get; set; }
+    }
+}

GrootUnitTest/models/ItemType.cs

@@ -0,0 +1,12 @@
+using Groot;
+
+namespace GrootUnitTest.models
+{
+    public enum ItemType 
+    {
+        Plastic=1,
+        Wood=2,
+        Iron=3,
+        Fabric=4
+    };
+}

GrootUnitTest/resources/item.csv

@@ -0,0 +1,4 @@
+name, amount, type
+itema, 68, 1
+itemb, 42, 2
+itemc, 55, 1

GrootUnitTest/test/GetObjectFromCsvTest.cs

@@ -8,6 +8,9 @@ namespace GrootUnitTest.test
     [TestClass]
     public class GetObjectFromCsvTest
     {
+        private const string itemName = "itema";
+        private const string mappedFieldWithNoCustomAttr = "mappedFieldWithCustomAttr";
+
         [TestMethod]
         public void AutoMapForNoCustomAttr()
         {
@@ -47,5 +50,39 @@ public void NoAutoMapForNoCustomAttr()
             Assert.AreEqual(232.29M, paul.Height, "mappedFieldWithCustomAttr");
 
         }
+
+        [TestMethod]
+        public void AutoMapForNoCustomAttrUsingEnum()
+        {
+            var path = Path.GetFullPath("resources/item.csv");
+            var types = Groot.Groot.GetObjectFromCsv<models.Item>(path);
+
+            Assert.AreEqual(3, types.Count(), "rows amount");
+
+            var itema = types.First(t => t.Name == itemName);
+
+            Assert.IsNotNull(itema);
+            Assert.AreEqual(itemName, itema.Name, mappedFieldWithNoCustomAttr);
+            Assert.AreEqual(68, itema.Amount, mappedFieldWithNoCustomAttr);
+            Assert.AreEqual(models.ItemType.Plastic, itema.Type, mappedFieldWithNoCustomAttr);
+
+        }
+
+        [TestMethod]
+        public void NoAutoMapForNoCustomAttrUsingEnum()
+        {
+            var path = Path.GetFullPath("resources/item.csv");
+            var types = Groot.Groot.GetObjectFromCsv<models.Item>(path, false);
+
+            Assert.AreEqual(3, types.Count(), "rows amount");
+
+            var itema = types.First(t => t.Name == itemName);
+
+            Assert.IsNotNull(itema);
+            Assert.AreEqual(itemName, itema.Name, mappedFieldWithNoCustomAttr);
+            Assert.AreEqual(68, itema.Amount, mappedFieldWithNoCustomAttr);
+            Assert.AreEqual(models.ItemType.Plastic, itema.Type, mappedFieldWithNoCustomAttr);
+
+        }
     }
 }