Added enum equivalence classes

Author: robdmoore

NTestDataBuilder.Tests/EquivalenceClasses/EnumEquivalenceClassesTests.cs

@@ -0,0 +1,53 @@
+using NTestDataBuilder.EquivalenceClasses;
+using Shouldly;
+using Xunit;
+
+namespace NTestDataBuilder.Tests.EquivalenceClasses
+{
+    public class EnumEquivalenceClassesTests
+    {
+        public AnonymousValueFixture Any { get; private set; }
+
+        public EnumEquivalenceClassesTests()
+        {
+            Any = new AnonymousValueFixture();
+        }
+
+        [Fact]
+        public void WhenGettingAnyOfEnum_ThenReturnSequentialtEnumValueEveryTimeUntilItWrapsAroundAgain()
+        {
+            var values = new[]
+            {
+                Any.Of<TestEnum>(),
+                Any.Of<TestEnum>(),
+                Any.Of<TestEnum>(),
+                Any.Of<TestEnum>(),
+                Any.Of<TestEnum>()
+            };
+
+            values.ShouldBe(new []{TestEnum.One, TestEnum.Two, TestEnum.Three, TestEnum.Four, TestEnum.One});
+        }
+
+        [Fact]
+        public void WhenGettingAnyEnumExceptOneValue_ThenReturnSequentialEnumValueExceptTheExceptionEveryTimeUntilItWrapsAroundAgain()
+        {
+            var values = new[]
+            {
+                Any.Except(TestEnum.Two),
+                Any.Except(TestEnum.Two),
+                Any.Except(TestEnum.Two),
+                Any.Except(TestEnum.Two)
+            };
+
+            values.ShouldBe(new[] { TestEnum.One, TestEnum.Three, TestEnum.Four, TestEnum.One });
+        }
+    }
+
+    enum TestEnum
+    {
+        One,
+        Two,
+        Three,
+        Four
+    }
+}

NTestDataBuilder.Tests/NTestDataBuilder.Tests.csproj

@@ -60,6 +60,7 @@
     <Compile Include="DataSources\DataSourceConventionTests.cs" />
     <Compile Include="EquivalenceClasses\PersonEquivalenceClassesTests.cs" />
     <Compile Include="EquivalenceClasses\IntegerEquivalenceClassesTests.cs" />
+    <Compile Include="EquivalenceClasses\EnumEquivalenceClassesTests.cs" />
     <Compile Include="EquivalenceClasses\StringEquivalenceClassesTests.cs" />
     <Compile Include="AsProxyTests.cs" />
     <Compile Include="BuildListTests.cs" />

NTestDataBuilder/EquivalenceClasses/EnumEquivalenceClasses.cs

@@ -0,0 +1,39 @@
+using System;
+using System.Linq;
+using Ploeh.AutoFixture;
+
+namespace NTestDataBuilder.EquivalenceClasses
+{
+    /// <summary>
+    /// Extension methods that describe equivalence classes for generating anonymous enum values.
+    /// </summary>
+    public static class EnumEquivalenceClasses
+    {
+        /// <summary>
+        /// Generate and return an enum from any available values in the enum.
+        /// </summary>
+        /// <param name="fixture">The fixture to generate an enum for</param>
+        /// <returns>The generated enum</returns>
+        public static TEnum Of<TEnum>(this AnonymousValueFixture fixture)
+            where TEnum : struct,  IComparable, IFormattable, IConvertible
+        {
+            return fixture.Fixture.Create<TEnum>();
+        }
+
+        /// <summary>
+        /// Generate and return an enum from any values in the enum except the provided exceptions.
+        /// </summary>
+        /// <param name="fixture">The fixture to generate an enum for</param>
+        /// <param name="except">A list of exceptions; recommend you specify by label for readability</param>
+        /// <returns>The generated enum</returns>
+        public static TEnum Except<TEnum>(this AnonymousValueFixture fixture, params TEnum[] except)
+            where TEnum : struct,  IComparable, IFormattable, IConvertible
+        {
+            var value = fixture.Fixture.Create<TEnum>();
+            while (except.Contains(value))
+                value = fixture.Fixture.Create<TEnum>();
+
+            return value;
+        }
+    }
+}

NTestDataBuilder/NTestDataBuilder.csproj

@@ -72,6 +72,7 @@
     <Compile Include="DataSources\Person\PersonNameSuffixSource.cs" />
     <Compile Include="DataSources\Person\PersonNameTitleSource.cs" />
     <Compile Include="DummyContext.cs" />
+    <Compile Include="EquivalenceClasses\EnumEquivalenceClasses.cs" />
     <Compile Include="EquivalenceClasses\GeographyEquivalenceClassescs.cs" />
     <Compile Include="EquivalenceClasses\IntegerEquivalenceClasses.cs" />
     <Compile Include="EquivalenceClasses\PersonEquivalenceClasses.cs" />