Made TypesToVerify IEnumerable<Type>

Author: khellang

TestStack.ConventionTests/ConventionData/Types.cs

@@ -1,6 +1,7 @@
 namespace TestStack.ConventionTests.ConventionData
 {
     using System;
+    using System.Collections;
     using System.Collections.Generic;
     using System.ComponentModel;
     using System.Linq;
@@ -9,15 +10,15 @@
     /// <summary>
     ///     This is where we set what our convention is all about.
     /// </summary>
-    public class Types : IConventionData
+    public class Types : IConventionData, IEnumerable<Type>
     {
         private Types(IEnumerable<Type> typesToVerify, string descriptionOfTypes)
         {
-            TypesToVerify = types.ToArray();
+            TypesToVerify = typesToVerify;
             Description = descriptionOfTypes;
         }
 
-        public Type[] TypesToVerify { get; private set; }
+        public IEnumerable<Type> TypesToVerify { get; private set; }
 
         public string Description { get; private set; }
 
@@ -202,6 +203,16 @@ public static Types InCollection(IEnumerable<Type> types, string descriptionOfTy
             return new Types(types, descriptionOfTypes);
         }
 
+        public IEnumerator<Type> GetEnumerator()
+        {
+            return TypesToVerify.GetEnumerator();
+        }
+
+        IEnumerator IEnumerable.GetEnumerator()
+        {
+            return GetEnumerator();
+        }
+
         private static string GetAssemblyName(Assembly assembly)
         {
             return assembly.GetName().Name;