Unit tests

Signed-off-by: Zeynel Koca <z.koca@primevision.com>

Author: ZeynelKoca

state/aws/dynamodb/dynamodb.go

@@ -493,26 +493,7 @@ func (d *StateStore) Multi(ctx context.Context, request *state.TransactionalStat
 			if err != nil {
 				return fmt.Errorf("dynamodb error: failed to marshal value for key %s: %w", req.Key, err)
 			}
-			ttl, err := d.parseTTL(&req)
-			if err != nil {
-				return fmt.Errorf("dynamodb error: failed to parse ttlInSeconds: %w", err)
-			}
-			twi.Put = &types.Put{
-				TableName: ptr.Of(d.table),
-				Item: map[string]types.AttributeValue{
-					d.partitionKey: &types.AttributeValueMemberS{
-						Value: req.Key,
-					},
-					"value": &types.AttributeValueMemberS{
-						Value: value,
-					},
-				},
-			}
-			if ttl != nil {
-				twi.Put.Item[d.ttlAttributeName] = &types.AttributeValueMemberN{
-					Value: strconv.FormatInt(*ttl, 10),
-				}
-			}
+			twi.Put = pd.ToPut()
 
 		case state.DeleteRequest:
 			twi.Delete = &types.Delete{

state/aws/dynamodb/dynamodb_test.go

@@ -1027,7 +1027,6 @@ func TestParseTTLWithDefault(t *testing.T) {
 		ttl, err := s.parseTTL(req)
 		require.NoError(t, err)
 		require.NotNil(t, ttl)
-
 		// Should use explicit value (300), not default (600)
 		expectedTime := time.Now().Unix() + 300
 		assert.InDelta(t, expectedTime, *ttl, 2) // Allow 2 second tolerance