Merge pull request #11 from Tynamix/FEATURE_Enumerator_Plugin

Feature enumerator plugin

Author: Tynamix

ObjectFiller.Test/ListFillingTest.cs

@@ -1,4 +1,6 @@
+using System;
 using System.Collections.Generic;
+using System.Linq;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using ObjectFiller.Test.TestPoco.ListTest;
 using Tynamix.ObjectFiller;
@@ -24,6 +26,27 @@ public void TestFillAllListsExceptArray()
             Assert.IsNotNull(entity.EntityIList);
         }
 
+        [TestMethod]
+        public void TestUseEnumerable()
+        {
+            Filler<EntityCollection> eFiller = new Filler<EntityCollection>();
+            eFiller.Setup()
+                .ListItemCount(20)
+                .OnProperty(x => x.EntityArray, x => x.EntityICollection,
+                            x => x.EntityIList, x => x.ObservableCollection,
+                            x => x.EntityIEnumerable).IgnoreIt()
+                .SetupFor<Entity>()
+                .OnProperty(x => x.Id).Use(Enumerable.Range(1, 22).Select(x => (int)Math.Pow(2, x)));
+
+            EntityCollection ec = eFiller.Create();
+
+            for (int i = 0; i < ec.EntityList.Count; i++)
+            {
+                int lastPowNum = (int)Math.Pow(2, i + 1);
+                Assert.AreEqual(lastPowNum, ec.EntityList[i].Id);
+            }
+        }
+
         [TestMethod]
         public void TestFillList()
         {

ObjectFiller.Test/PersonFillingTest.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using ObjectFiller.Test.TestPoco.Person;
 using Tynamix.ObjectFiller;
@@ -26,6 +27,25 @@ public void TestFillPerson()
 
         }
 
+        [TestMethod]
+        public void TestFillPersonWithEnumerable()
+        {
+            Filler<Person> pFiller = new Filler<Person>();
+
+            pFiller.Setup()
+                .OnType<IAddress>().CreateInstanceOf<Address>()
+                .OnProperty(x => x.Age).Use(Enumerable.Range(18, 60));
+                
+
+            Person filledPerson = pFiller.Create();
+
+            Assert.IsNotNull(filledPerson.Address);
+            Assert.IsNotNull(filledPerson.Addresses);
+            Assert.IsNotNull(filledPerson.StringToIAddress);
+            Assert.IsNotNull(filledPerson.SureNames);
+
+        }
+
         [TestMethod]
         public void TestNameListStringRandomizer()
         {

ObjectFiller/ObjectFiller.csproj

@@ -70,6 +70,9 @@
       <SubType>Designer</SubType>
     </EmbeddedResource>
   </ItemGroup>
+  <ItemGroup>
+    <Compile Include="Plugins\EnumeratorPlugin.cs" />
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.

ObjectFiller/Plugins/EnumeratorPlugin.cs

@@ -0,0 +1,41 @@
+using System;
+using System.Collections.Generic;
+
+namespace Tynamix.ObjectFiller
+{
+    internal class EnumeratorPlugin<T> : IRandomizerPlugin<T>
+    {
+        private readonly IEnumerable<T> _enumerable;
+        private IEnumerator<T> _enumerator;
+
+
+        public EnumeratorPlugin(IEnumerable<T> enumerable)
+        {
+            _enumerable = enumerable;
+        }
+
+        public T GetValue()
+        {
+            // First time?
+            if (_enumerator == null)
+            {
+                _enumerator = _enumerable.GetEnumerator();
+            }
+
+            // Advance, try to recover if we hit end-of-enumeration
+            var hasNext = _enumerator.MoveNext();
+            if (!hasNext)
+            {
+                _enumerator = _enumerable.GetEnumerator();
+                hasNext = _enumerator.MoveNext();
+
+                if (!hasNext)
+                {
+                    throw new Exception("Unable to get next value from enumeration " + _enumerable);
+                }
+            }
+
+            return _enumerator.Current;
+        }
+    }
+}

ObjectFiller/Setup/FluentPropertyApi.cs

@@ -71,6 +71,18 @@ public FluentFillerApi<TTargetObject> Use(IRandomizerPlugin<TTargetType> randomi
             return Use(randomizerPlugin.GetValue);
         }
 
+        /// <summary>
+        /// Use this function if you want to use an IEnumerable for the datageneration.
+        /// With that you can generate random data in a specific order, with include, exclude and all the other stuff
+        /// what is possible with IEnumerables and linq
+        /// </summary>
+        /// <param name="enumerable">An IEnumerable with items of type <typeparam name="TTargetObject"/> which will be used to fill the data.</param>
+        /// <returns>Main FluentFiller API</returns>
+        public FluentFillerApi<TTargetObject> Use(IEnumerable<TTargetType> enumerable)
+        {
+            return Use(new EnumeratorPlugin<TTargetType>(enumerable));
+        }
+
         /// <summary>
         /// Ignores the entity for which the fluent setup is made (Type, Property)
         /// </summary>

ObjectFiller/Setup/FluentTypeApi.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 
 namespace Tynamix.ObjectFiller
 {
@@ -39,6 +40,11 @@ public FluentFillerApi<TTargetObject> Use(IRandomizerPlugin<TTargetType> randomi
             return _callback;
         }
 
+        public FluentFillerApi<TTargetObject> Use(IEnumerable<TTargetType> enumerable)
+        {
+            return Use(new EnumeratorPlugin<TTargetType>(enumerable));
+        }
+
         /// <summary>
         /// Ignores the entity for which the fluent setup is made (Type, Property)
         /// </summary>

ObjectFiller/Setup/IFluentApi.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 
 namespace Tynamix.ObjectFiller
 {
@@ -24,6 +25,15 @@ internal interface IFluentApi<TTargetObject, TTargetType> where TTargetObject :
         /// <returns>Main FluentFiller API</returns>
         FluentFillerApi<TTargetObject> Use(IRandomizerPlugin<TTargetType> randomizerPlugin);
 
+        /// <summary>
+        /// Use this function if you want to use an IEnumerable for the datageneration.
+        /// With that you can generate random data in a specific order, with include, exclude and all the other stuff
+        /// what is possible with IEnumerables and linq
+        /// </summary>
+        /// <param name="enumerable">An IEnumerable with items of type <typeparam name="TTargetObject"/> which will be used to fill the data.</param>
+        /// <returns>Main FluentFiller API</returns>
+        FluentFillerApi<TTargetObject> Use(IEnumerable<TTargetType> enumerable);
+
         /// <summary>
         /// Ignores the entity for which the fluent setup is made (Type, Property)
         /// </summary>