Switch generators to use data abstraction layer

EasySourceGenerators.Examples/PiExample.cs

@@ -3,21 +3,23 @@
 
 namespace EasySourceGenerators.Examples;
 
-public static partial class PiExample
-{
-    public static partial int GetPiDecimal(int decimalNumber);
-
-    [GeneratesMethod(nameof(GetPiDecimal))]
-    [SwitchCase(arg1: 0)]
-    [SwitchCase(arg1: 1)]
-    [SwitchCase(arg1: 2)]
-    static int GetPiDecimal_Generator_Specialized(int decimalNumber) =>
-        SlowMath.CalculatePiDecimal(decimalNumber);
-
-    [GeneratesMethod(nameof(GetPiDecimal))]
-    [SwitchDefault]
-    static Func<int, int> GetPiDecimal_Generator_Fallback() => decimalNumber => SlowMath.CalculatePiDecimal(decimalNumber);
-}
+// NOTE: [SwitchCase] attribute-based generation is commented out pending replacement
+// with a data-driven approach. See PiExampleFluent.cs for the fluent API equivalent.
+// public static partial class PiExample
+// {
+//     public static partial int GetPiDecimal(int decimalNumber);
+//
+//     [GeneratesMethod(nameof(GetPiDecimal))]
+//     [SwitchCase(arg1: 0)]
+//     [SwitchCase(arg1: 1)]
+//     [SwitchCase(arg1: 2)]
+//     static int GetPiDecimal_Generator_Specialized(int decimalNumber) =>
+//         SlowMath.CalculatePiDecimal(decimalNumber);
+//     
+//     [GeneratesMethod(nameof(GetPiDecimal))]
+//     [SwitchDefault]
+//     static Func<int, int> GetPiDecimal_Generator_Fallback() => decimalNumber => SlowMath.CalculatePiDecimal(decimalNumber);
+// }
 
 /*
  This will generate the following method:

EasySourceGenerators.GeneratorTests/DataGeneratorsFactoryTests.cs

@@ -0,0 +1,117 @@
+using EasySourceGenerators.Generators;
+
+namespace EasySourceGenerators.GeneratorTests;
+
+[TestFixture]
+public class DataGeneratorsFactoryTests
+{
+    [Test]
+    public void CreateSimpleReturnBody_WithValue_CreatesCorrectData()
+    {
+        DataSimpleReturnBody result = DataGeneratorsFactory.CreateSimpleReturnBody("hello");
+
+        Assert.That(result.ReturnValue, Is.EqualTo("hello"));
+    }
+
+    [Test]
+    public void CreateSimpleReturnBody_WithNull_CreatesDataWithNullValue()
+    {
+        DataSimpleReturnBody result = DataGeneratorsFactory.CreateSimpleReturnBody(null);
+
+        Assert.That(result.ReturnValue, Is.Null);
+    }
+
+    [Test]
+    public void CreateSwitchBodyFromFluentData_WithCasesAndDefault_CreatesCorrectData()
+    {
+        SwitchBodyData fluentData = new SwitchBodyData(
+            CasePairs: new List<(object key, string value)> { (1, "2"), (2, "4"), (3, "6") },
+            HasDefaultCase: true);
+
+        DataSwitchBody result = DataGeneratorsFactory.CreateSwitchBodyFromFluentData(fluentData, "0");
+
+        Assert.That(result.Cases, Has.Count.EqualTo(3));
+        Assert.That(result.Cases[0].Key, Is.EqualTo(1));
+        Assert.That(result.Cases[0].FormattedValue, Is.EqualTo("2"));
+        Assert.That(result.Cases[1].Key, Is.EqualTo(2));
+        Assert.That(result.Cases[1].FormattedValue, Is.EqualTo("4"));
+        Assert.That(result.Cases[2].Key, Is.EqualTo(3));
+        Assert.That(result.Cases[2].FormattedValue, Is.EqualTo("6"));
+        Assert.That(result.DefaultCase, Is.Not.Null);
+        Assert.That(result.DefaultCase!.Expression, Is.EqualTo("0"));
+    }
+
+    [Test]
+    public void CreateSwitchBodyFromFluentData_WithoutDefault_CreatesDataWithNullDefaultCase()
+    {
+        SwitchBodyData fluentData = new SwitchBodyData(
+            CasePairs: new List<(object key, string value)> { (1, "one") },
+            HasDefaultCase: false);
+
+        DataSwitchBody result = DataGeneratorsFactory.CreateSwitchBodyFromFluentData(fluentData, null);
+
+        Assert.That(result.Cases, Has.Count.EqualTo(1));
+        Assert.That(result.DefaultCase, Is.Null);
+    }
+
+    [Test]
+    public void CreateSwitchBodyFromFluentData_WithEmptyCases_CreatesEmptyData()
+    {
+        SwitchBodyData fluentData = new SwitchBodyData(
+            CasePairs: new List<(object key, string value)>(),
+            HasDefaultCase: false);
+
+        DataSwitchBody result = DataGeneratorsFactory.CreateSwitchBodyFromFluentData(fluentData, null);
+
+        Assert.That(result.Cases, Is.Empty);
+        Assert.That(result.DefaultCase, Is.Null);
+    }
+
+    [Test]
+    public void CreateSwitchBodyFromFluentData_WithDefaultOnly_CreatesDataWithDefaultAndNoCases()
+    {
+        SwitchBodyData fluentData = new SwitchBodyData(
+            CasePairs: new List<(object key, string value)>(),
+            HasDefaultCase: true);
+
+        DataSwitchBody result = DataGeneratorsFactory.CreateSwitchBodyFromFluentData(fluentData, "888");
+
+        Assert.That(result.Cases, Is.Empty);
+        Assert.That(result.DefaultCase, Is.Not.Null);
+        Assert.That(result.DefaultCase!.Expression, Is.EqualTo("888"));
+    }
+
+    [Test]
+    public void CreateSwitchBodyFromFluentData_WithThrowExpression_PreservesExpression()
+    {
+        SwitchBodyData fluentData = new SwitchBodyData(
+            CasePairs: new List<(object key, string value)> { (1, "\"Dog\"") },
+            HasDefaultCase: true);
+
+        DataSwitchBody result = DataGeneratorsFactory.CreateSwitchBodyFromFluentData(
+            fluentData,
+            "throw new ArgumentException(\"Unknown\")");
+
+        Assert.That(result.DefaultCase, Is.Not.Null);
+        Assert.That(result.DefaultCase!.Expression, Is.EqualTo("throw new ArgumentException(\"Unknown\")"));
+    }
+
+    [Test]
+    public void CreateSwitchBodyFromFluentData_PreservesKeyTypes()
+    {
+        SwitchBodyData fluentData = new SwitchBodyData(
+            CasePairs: new List<(object key, string value)>
+            {
+                (true, "\"Yes\""),
+                ("hello", "\"World\""),
+                (42, "\"Answer\"")
+            },
+            HasDefaultCase: false);
+
+        DataSwitchBody result = DataGeneratorsFactory.CreateSwitchBodyFromFluentData(fluentData, null);
+
+        Assert.That(result.Cases[0].Key, Is.TypeOf<bool>());
+        Assert.That(result.Cases[1].Key, Is.TypeOf<string>());
+        Assert.That(result.Cases[2].Key, Is.TypeOf<int>());
+    }
+}