🔧 allow multiple usages of [RequireNonZero] and [RequireUnique]

fluffynuts · Dec 12, 2024 · 8113e91 · 8113e91
1 parent 08422e3
commit 8113e91
Show file tree

Hide file tree

Showing 3 changed files with 98 additions and 80 deletions.
diff --git a/source/TestUtils/PeanutButter.RandomGenerators/RequireNonZero.cs b/source/TestUtils/PeanutButter.RandomGenerators/RequireNonZero.cs
@@ -4,7 +4,6 @@
 namespace Imported.PeanutButter.RandomGenerators;
 #else
 using System;
-using PeanutButter.DuckTyping.AutoConversion;
 using PeanutButter.Utils;
 
 namespace PeanutButter.RandomGenerators;
@@ -13,6 +12,10 @@ namespace PeanutButter.RandomGenerators;
 /// <summary>
 /// Requires that the named property is randomized to a non-zero value
 /// </summary>
+[AttributeUsage(
+    AttributeTargets.Class,
+    AllowMultiple = true
+)]
 #if BUILD_PEANUTBUTTER_INTERNAL
 internal
 #else
@@ -34,83 +37,4 @@ ref object target
     {
         propInfo.SetValue(target, RandomValueGen.GetRandomInt(1));
     }
-}
-
-/// <summary>
-/// Restricts the possible values for a property
-/// to the provided list
-/// </summary>
-[AttributeUsage(
-    validOn: AttributeTargets.Class,
-    AllowMultiple = true
-)]
-#if BUILD_PEANUTBUTTER_INTERNAL
-internal
-#else
-public
-#endif
-    class Restrict : RandomizerAttribute
-{
-    /// <summary>
-    /// The restricted values for this property
-    /// </summary>
-    public object[] Values { get; }
-
-    /// <inheritdoc />
-    public Restrict(
-        string propertyName,
-        object value,
-        params object[] moreValues
-    ) : base(propertyName)
-    {
-        Values = new[]
-        {
-            value
-        }.And(moreValues);
-    }
-
-    /// <inheritdoc />
-    public override void SetRandomValue(
-        PropertyOrField propInfo,
-        ref object target
-    )
-    {
-        var toSet = RandomValueGen.GetRandomFrom(Values);
-        if (toSet is null)
-        {
-            propInfo.SetValue(
-                target,
-                propInfo.Type.IsNullableType()
-                    ? null
-                    : propInfo.Type.DefaultValue()
-            );
-
-            return;
-        }
-
-        var valueType = toSet.GetType();
-        if (valueType == propInfo.Type)
-        {
-            propInfo.SetValue(target, toSet);
-            return;
-        }
-
-        var converter = ConverterLocator.TryFindConverter(
-            valueType,
-            propInfo.Type
-        );
-        if (converter is null)
-        {
-            throw new NotImplementedException(
-                $"""
-                 There is no known converter to convert between {valueType} and {propInfo.Type}.
-                 You may implement IConverter<{valueType}, {propInfo.Type}> to resolve this:
-                   ConverterLocator should pick it up.
-                 """
-            );
-        }
-
-        var converted = converter.Convert(toSet);
-        propInfo.SetValue(target, converted);
-    }
 }
diff --git a/source/TestUtils/PeanutButter.RandomGenerators/RequireUnique.cs b/source/TestUtils/PeanutButter.RandomGenerators/RequireUnique.cs
@@ -14,6 +14,10 @@ namespace PeanutButter.RandomGenerators;
 /// <summary>
 /// Abstract class to require uniqueness on a property or field by name
 /// </summary>
+[AttributeUsage(
+    AttributeTargets.Class,
+    AllowMultiple = true
+)]
 #if BUILD_PEANUTBUTTER_INTERNAL
 internal
 #else

diff --git a/source/TestUtils/PeanutButter.RandomGenerators/Restrict.cs b/source/TestUtils/PeanutButter.RandomGenerators/Restrict.cs
@@ -0,0 +1,90 @@
+#if BUILD_PEANUTBUTTER_INTERNAL
+using Imported.PeanutButter.Utils;
+
+namespace Imported.PeanutButter.RandomGenerators;
+#else
+using System;
+using PeanutButter.DuckTyping.AutoConversion;
+using PeanutButter.Utils;
+
+namespace PeanutButter.RandomGenerators;
+#endif
+
+/// <summary>
+/// Restricts the possible values for a property
+/// to the provided list
+/// </summary>
+[AttributeUsage(
+    validOn: AttributeTargets.Class,
+    AllowMultiple = true
+)]
+#if BUILD_PEANUTBUTTER_INTERNAL
+internal
+#else
+public
+#endif
+    class Restrict : RandomizerAttribute
+{
+    /// <summary>
+    /// The restricted values for this property
+    /// </summary>
+    public object[] Values { get; }
+
+    /// <inheritdoc />
+    public Restrict(
+        string propertyName,
+        object value,
+        params object[] moreValues
+    ) : base(propertyName)
+    {
+        Values = new[]
+        {
+            value
+        }.And(moreValues);
+    }
+
+    /// <inheritdoc />
+    public override void SetRandomValue(
+        PropertyOrField propInfo,
+        ref object target
+    )
+    {
+        var toSet = RandomValueGen.GetRandomFrom(Values);
+        if (toSet is null)
+        {
+            propInfo.SetValue(
+                target,
+                propInfo.Type.IsNullableType()
+                    ? null
+                    : propInfo.Type.DefaultValue()
+            );
+
+            return;
+        }
+
+        var valueType = toSet.GetType();
+        if (valueType == propInfo.Type)
+        {
+            propInfo.SetValue(target, toSet);
+            return;
+        }
+
+        var converter = ConverterLocator.TryFindConverter(
+            valueType,
+            propInfo.Type
+        );
+        if (converter is null)
+        {
+            throw new NotImplementedException(
+                $"""
+                 There is no known converter to convert between {valueType} and {propInfo.Type}.
+                 You may implement IConverter<{valueType}, {propInfo.Type}> to resolve this:
+                   ConverterLocator should pick it up.
+                 """
+            );
+        }
+
+        var converted = converter.Convert(toSet);
+        propInfo.SetValue(target, converted);
+    }
+}