Fix macros (#18)

Sage · Oct 24, 2024 · 065f4eb · 065f4eb
1 parent b7fa48b
commit 065f4eb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/Sources/SageSwiftKitMacros/CodableMacros/Builders/DecodeVariableBuild.swift b/Sources/SageSwiftKitMacros/CodableMacros/Builders/DecodeVariableBuild.swift
@@ -186,7 +186,7 @@ struct DecodeVariableBuild {
     }
 
     func buildStringOrInt(attribute: AttributeSyntax) -> CodeBlockItemSyntaxBuilder {
-        guard type == "String" else {
+        guard type == "String?" else {
             return buildBasicDecode()
         }
 
@@ -221,7 +221,7 @@ struct DecodeVariableBuild {
     }
 
     func buildStringOrDouble(attribute: AttributeSyntax) -> CodeBlockItemSyntaxBuilder {
-        guard type == "String" else {
+        guard type == "String?" else {
             return buildBasicDecode()
         }
 

diff --git a/Tests/SageSwiftKitTests/CodableMacrosTests/CustomDecodableTests.swift b/Tests/SageSwiftKitTests/CodableMacrosTests/CustomDecodableTests.swift
@@ -136,12 +136,12 @@ final class CustomDecodableTests: XCTestCase {
     @CustomCodable
     struct PlayingObject {
         @StringOrInt
-        var value: String
+        var value: String?
     }
     """,
     expandedSource: """
     struct PlayingObject {
-        var value: String
+        var value: String?
     
         enum CodingKeys: String, CodingKey {
             case value
@@ -180,12 +180,12 @@ final class CustomDecodableTests: XCTestCase {
     @CustomCodable
     struct PlayingObject {
         @StringOrDouble
-        var value: String
+        var value: String?
     }
     """,
     expandedSource: """
     struct PlayingObject {
-        var value: String
+        var value: String?
     
         enum CodingKeys: String, CodingKey {
             case value