add struct constraint and enum constraint test

Author: hi-xiaoxin

src/SourceGeneration.Reflection.SourceGenerator.Test/UnitTest1.cs

@@ -8,20 +8,24 @@ public class UnitTest1
     [TestMethod]
     public void TestMethod1()
     {
-        string source = @"
-//[assembly: SourceReflectionAttribute<SourceGeneration.Reflection.Sample2.EnumTestObject>]
-//[assembly: SourceReflectionType(typeof(System.Collections.Generic.List<string>))]
-//[assembly: SourceReflectionType<int[]>]
-using System;
-using System.Threading.Tasks;
-using SourceGeneration.Reflection;
+        string source = """
+            //[assembly: SourceReflectionAttribute<SourceGeneration.Reflection.Sample2.EnumTestObject>]
+            //[assembly: SourceReflectionType(typeof(System.Collections.Generic.List<string>))]
+            //[assembly: SourceReflectionType<int[]>]
+            using System;
+            using System.Threading.Tasks;
+            using SourceGeneration.Reflection;
+
+            namespace SourceGeneration.Reflection.Sample2
+            {
+                [SourceReflection]
+                public class TestClass
+                {
+                    public T InvokeStructConstraint<T>(T t) where T: struct => t;
+                }
+            }
+            """;
 
-namespace SourceGeneration.Reflection.Sample2
-{
-public class BaseObject { }
-[SourceReflection] public class InheritedObject1 : BaseObject { }
-[SourceReflection] public class InheritedObject2 : BaseObject { }
-";
         var result = CSharpTestGenerator.Generate<ReflectionSourceGenerator>(source, typeof(SourceReflectionAttribute).Assembly, typeof(UnitTest1).Assembly);
         var script = result.RunResult.GeneratedTrees.FirstOrDefault()?.GetText();
         var script2 = result.RunResult.GeneratedTrees.LastOrDefault()?.GetText();

src/SourceGeneration.Reflection.Test/GenericMethodTest.cs

@@ -68,6 +68,44 @@ public void NotNullConstraint()
         Assert.AreEqual(null, method.Invoke(instance, [null]));
     }
 
+    [TestMethod]
+    public void StructConstraint()
+    {
+        SourceTypeInfo type = SourceReflector.GetRequiredType<GenericMethodTestObject>();
+
+        var method = type.GetMethod("InvokeStructConstraint")!;
+
+        GenericMethodTestObject instance = new();
+
+        Assert.ThrowsException<InvalidOperationException>(() =>
+        {
+            Assert.AreEqual(1, method.Invoke(instance, [1]));
+        });
+
+        var value = method.MethodInfo.MakeGenericMethod(typeof(int)).Invoke(instance, [1]);
+        Assert.AreEqual(1, value);
+    }
+
+    [TestMethod]
+    public void EnumConstraint()
+    {
+        SourceTypeInfo type = SourceReflector.GetRequiredType<GenericMethodTestObject>();
+
+        var method = type.GetMethod("InvokeEnumConstraint")!;
+
+        GenericMethodTestObject instance = new();
+
+        
+        var aa = method.Invoke(instance, [EnumTestObject.B]);
+
+        Assert.AreEqual(EnumTestObject.A, method.Invoke(instance, [EnumTestObject.A]));
+        Assert.AreEqual(EnumTestObject.B, method.Invoke(instance, [EnumTestObject.B]));
+        Assert.ThrowsException<InvalidCastException>(() =>
+        {
+            Assert.AreEqual("abc", method.Invoke(instance, ["abc"]));
+        });
+    }
+
     [TestMethod]
     public void DoubleInternfaceConstraint()
     {
@@ -87,7 +125,6 @@ public void DoubleInternfaceConstraint()
         });
     }
 
-
     [TestMethod]
     public void DoubleTypeParameter()
     {
@@ -138,6 +175,10 @@ public class GenericMethodTestObject
     public T Invoke5<T>(T t) where T : ICloneable, IComparable => t;
     public T Invoke6<T, K>(T t, K k) where T : ICloneable where K : IComparable => t;
     public T[] InvokeArray1<T>(T[] t) => t;
+
+    public T InvokeStructConstraint<T>(T t) where T: struct => t;
+    public T InvokeEnumConstraint<T>(T t) where T : Enum => t;
+
 }