Solve primite argument

Author: inputfalken

src/DynamoDBGenerator.SourceGenerator/Generations/AttributeExpressionValue.cs

@@ -86,23 +86,25 @@ internal static IEnumerable<string> CreateExpressionAttributeValue(IEnumerable<D
     }
     private static CodeFactory CreateStruct(ITypeSymbol typeSymbol, Func<ITypeSymbol, ImmutableArray<DynamoDbDataMember>> fn, MarshallerOptions options)
     {
-        // TODO
-        if (options.IsConvertable(typeSymbol))
-        {
-            
-        }
-        var dataMembers = fn(typeSymbol)
-            .Select(x =>
-            {
-                return (
-                    IsUnknown: !options.IsConvertable(x.DataMember.Type) && x.DataMember.Type.TypeIdentifier() is UnknownType,
-                    DDB: x,
-                    ValueRef: $"_{x.DataMember.Name}ValueRef",
-                    AttributeReference: TypeName(x.DataMember.Type),
-                    AttributeInterfaceName: $"{Constants.DynamoDBGenerator.Marshaller.AttributeExpressionValueTrackerInterface}<{x.DataMember.Type.Representation().annotated}>"
-                );
-            })
-            .ToArray();
+        var dataMembers =
+            options.IsConvertable(typeSymbol)
+                ? Array
+                    .Empty<(bool IsUnknown, DynamoDbDataMember DDB, string ValueRef, string AttributeReference, string
+                        AttributeInterfaceName)>()
+                : fn(typeSymbol)
+                    .Select(x =>
+                    {
+                        return (
+                            IsUnknown: !options.IsConvertable(x.DataMember.Type) &&
+                                       x.DataMember.Type.TypeIdentifier() is UnknownType,
+                            DDB: x,
+                            ValueRef: $"_{x.DataMember.Name}ValueRef",
+                            AttributeReference: TypeName(x.DataMember.Type),
+                            AttributeInterfaceName:
+                            $"{Constants.DynamoDBGenerator.Marshaller.AttributeExpressionValueTrackerInterface}<{x.DataMember.Type.Representation().annotated}>"
+                        );
+                    })
+                    .ToArray();
 
         var structName = TypeName(typeSymbol);
         var interfaceName = $"{Constants.DynamoDBGenerator.Marshaller.AttributeExpressionValueTrackerInterface}<{typeSymbol.Representation().annotated}>";

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

@@ -1,14 +1,20 @@
+using Amazon.DynamoDBv2.Model;
 using DynamoDBGenerator.Attributes;
 
 namespace DynamoDBGenerator.SourceGenerator.Tests.DynamoDBDocumentTests;
 
 [DynamoDBMarshaller(typeof((Guid, Guid)), PropertyName = "PrimitiveArgument", ArgumentType = typeof(string))]
 public partial class PrimitiveArgumentTests
 {
-    
     [Fact]
-    public void Test()
+    public void ExpressionValueTracker_String_ShouldBeExpandedCorrectly()
     {
-        PrimitiveArgument.
+        IAttributeExpressionValueTracker<string> valueTracker = PrimitiveArgument.AttributeExpressionValueTracker();
+        valueTracker.ToString().Should().Be(":p1");
+
+        valueTracker.AccessedValues("hello").Should().BeEquivalentTo(new Dictionary<string, AttributeValue>
+        {
+            { ":p1", new AttributeValue { S = "hello" } }
+        });
     }
 }