Add unit tests for GenericArgumentClauseSyntax

Matejkob · Feb 5, 2024 · b5267c0 · b5267c0
1 parent 2ac9227
commit b5267c0
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/Tests/SpyableMacroTests/Extensions/UT_TypeSyntax+ContainsGenericType.swift b/Tests/SpyableMacroTests/Extensions/UT_TypeSyntax+ContainsGenericType.swift
@@ -51,6 +51,26 @@ final class UT_TypeSyntax_ContainsGenericType: XCTestCase {
     XCTAssertFalse(typeSyntax(with: "String", containsGenericType: ["T"]))
   }
 
+  func testContainsGenericType_WithGenericArgumentClauseSyntax() {
+    func typeSyntax(
+      with identifier: String,
+      containsGenericType genericTypes: Set<String>
+    ) -> Bool {
+      TypeSyntax(
+        IdentifierTypeSyntax(
+          name: .identifier("Array"),
+          genericArgumentClause: GenericArgumentClauseSyntax {
+            GenericArgumentSyntax(argument: TypeSyntax(stringLiteral: identifier))
+          }
+        )
+      )
+      .containsGenericType(from: genericTypes)
+    }
+
+    XCTAssertTrue(typeSyntax(with: "T", containsGenericType: ["T"]))
+    XCTAssertFalse(typeSyntax(with: "String", containsGenericType: ["T"]))
+  }
+
   func testContainsGenericType_WithTupleTypeSyntax() {
     func typeSyntax(
       with identifier: String,

diff --git a/Tests/SpyableMacroTests/Extensions/UT_TypeSyntax+ErasingGenericType.swift b/Tests/SpyableMacroTests/Extensions/UT_TypeSyntax+ErasingGenericType.swift
@@ -49,6 +49,26 @@ final class UT_TypeSyntax_ErasingGenericTypes: XCTestCase {
     XCTAssertEqual(typeSyntaxDescription(with: "String"), " [String] ")
   }
 
+  func testErasingGenericTypes_WithGenericArgumentClauseSyntax() {
+    func typeSyntaxDescription(with identifier: String) -> String {
+      TypeSyntax(
+        IdentifierTypeSyntax(
+          leadingTrivia: .space,
+          name: .identifier("Array"),
+          genericArgumentClause: GenericArgumentClauseSyntax {
+            GenericArgumentSyntax(argument: TypeSyntax(stringLiteral: identifier))
+          },
+          trailingTrivia: .space
+        )
+      )
+      .erasingGenericTypes(["T"])
+      .description
+    }
+
+    XCTAssertEqual(typeSyntaxDescription(with: "T"), " Array<Any> ")
+    XCTAssertEqual(typeSyntaxDescription(with: "String"), " Array<String> ")
+  }
+
   func testErasingGenericTypes_WithTupleTypeSyntax() {
     func typeSyntaxDescription(with identifier: String) -> String {
       TypeSyntax(