Simple tests for @_package attribute

swiftlang · Feb 4, 2023 · 0f4cfba · 0f4cfba
1 parent 5ee31ce
commit 0f4cfba
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/Tests/SwiftParserTest/AttributeTests.swift b/Tests/SwiftParserTest/AttributeTests.swift
@@ -540,4 +540,9 @@ final class AttributeTests: XCTestCase {
     AssertParse("func takeRepeater(_ f: @MainActor @Sendable @escaping () -> Int) {}")
     AssertParse("takeRepesater { @MainActor @Sendable () -> Int in 0 }")
   }
+
+  func testPackageAttribute() {
+    AssertParse(#"@_package(path: "../my-package") import A"#)
+    AssertParse(#"@_package(url: "https://example.com/package.git", from: "0.0.1") @_exported import A"#)
+  }
 }
diff --git a/Tests/SwiftSyntaxBuilderTest/ImportTests.swift b/Tests/SwiftSyntaxBuilderTest/ImportTests.swift
@@ -26,4 +26,15 @@ final class ImportTests: XCTestCase {
 
     AssertBuildResult(importDecl, "␣import SwiftSyntax")
   }
+
+  func testPackageAttribute() {
+    let builder = ImportDeclSyntax("""
+      @_package(url: "https://github.com/apple/swift-log.git", "1.0.0"..<"2.0.0")
+      import Logging
+      """)
+    AssertBuildResult(builder, """
+      @_package(url: "https://github.com/apple/swift-log.git", "1.0.0" ..< "2.0.0")
+      import Logging
+      """)
+  }
 }