Add tests to verify primitive arguments

Author: inputfalken

tests/DynamoDBGenerator.SourceGenerator.Tests/DynamoDBDocumentTests/PrimitiveArgumentTests.cs

@@ -1,20 +1,48 @@
 using Amazon.DynamoDBv2.Model;
 using DynamoDBGenerator.Attributes;
+using DynamoDBGenerator.Extensions;
 
 namespace DynamoDBGenerator.SourceGenerator.Tests.DynamoDBDocumentTests;
 
-[DynamoDBMarshaller(typeof((Guid, Guid)), PropertyName = "PrimitiveArgument", ArgumentType = typeof(string))]
+[DynamoDBMarshaller(typeof((Guid, Guid)), PropertyName = "STRING", ArgumentType = typeof(string))]
+[DynamoDBMarshaller(typeof((Guid, Guid)), PropertyName = "GUID", ArgumentType = typeof(Guid))]
+[DynamoDBMarshaller(typeof((Guid, Guid)), PropertyName = "INT", ArgumentType = typeof(int))]
 public partial class PrimitiveArgumentTests
 {
     [Fact]
-    public void ExpressionValueTracker_String_ShouldBeExpandedCorrectly()
+    public void ExpressionValueTracker_STRING_ShouldBeExpandedCorrectly()
     {
-        IAttributeExpressionValueTracker<string> valueTracker = PrimitiveArgument.AttributeExpressionValueTracker();
+        IAttributeExpressionValueTracker<string> valueTracker = STRING.AttributeExpressionValueTracker();
         valueTracker.ToString().Should().Be(":p1");
 
         valueTracker.AccessedValues("hello").Should().BeEquivalentTo(new Dictionary<string, AttributeValue>
         {
             { ":p1", new AttributeValue { S = "hello" } }
         });
     }
+
+    [Fact]
+    public void ExpressionValueTracker_INT_ShouldBeExpandedCorrectly()
+    {
+        IAttributeExpressionValueTracker<int> valueTracker = INT.AttributeExpressionValueTracker();
+        valueTracker.ToString().Should().Be(":p1");
+
+        valueTracker.AccessedValues(2).Should().BeEquivalentTo(new Dictionary<string, AttributeValue>
+        {
+            { ":p1", new AttributeValue { N = "2" } }
+        });
+    }
+
+    [Fact]
+    public void ExpressionValueTracker_GUID_ShouldBeExpandedCorrectly()
+    {
+        IAttributeExpressionValueTracker<Guid> valueTracker = GUID.AttributeExpressionValueTracker();
+        valueTracker.ToString().Should().Be(":p1");
+
+        var guid = Guid.NewGuid();
+        valueTracker.AccessedValues(guid).Should().BeEquivalentTo(new Dictionary<string, AttributeValue>
+        {
+            { ":p1", new AttributeValue { S = guid.ToString() } }
+        });
+    }
 }