[Collections] Auth tokens need to be mutable

rdar://77095793

Author: yim-lee

Sources/PackageCollections/PackageCollections.swift

@@ -21,11 +21,11 @@ public struct PackageCollections: PackageCollectionsProtocol {
     static let isSupportedPlatform = false
     #endif
 
-    private let configuration: Configuration
+    var configuration: Configuration
     private let diagnosticsEngine: DiagnosticsEngine?
     private let storageContainer: (storage: Storage, owned: Bool)
     private let collectionProviders: [Model.CollectionSourceType: PackageCollectionProvider]
-    private let metadataProvider: PackageMetadataProvider
+    var metadataProvider: PackageMetadataProvider
 
     private var storage: Storage {
         self.storageContainer.storage
@@ -67,6 +67,14 @@ public struct PackageCollections: PackageCollectionsProtocol {
         }
     }
 
+    public mutating func updateAuthTokens(_ authTokens: [AuthTokenType: String]?) {
+        self.configuration.authTokens = authTokens
+        if var metadataProvider = self.metadataProvider as? GitHubPackageMetadataProvider {
+            metadataProvider.configuration.authTokens = authTokens
+            self.metadataProvider = metadataProvider
+        }
+    }
+
     // MARK: - Collections
 
     public func listCollections(identifiers: Set<PackageCollectionsModel.CollectionIdentifier>? = nil,

Tests/PackageCollectionsTests/PackageCollectionsTests.swift

@@ -18,6 +18,35 @@ import TSCBasic
 import TSCUtility
 
 final class PackageCollectionsTests: XCTestCase {
+    func testUpdateAuthTokens() throws {
+        let configuration = PackageCollections.Configuration()
+        let storage = makeMockStorage()
+        defer { XCTAssertNoThrow(try storage.close()) }
+
+        let metadataProviderConfig = GitHubPackageMetadataProvider.Configuration(cacheTTLInSeconds: -1)
+        let metadataProvider = GitHubPackageMetadataProvider(configuration: metadataProviderConfig)
+        var packageCollections = PackageCollections(configuration: configuration, storage: storage, collectionProviders: [:], metadataProvider: metadataProvider)
+
+        XCTAssertNil(packageCollections.configuration.authTokens)
+        do {
+            guard let githubMetadataProvider = packageCollections.metadataProvider as? GitHubPackageMetadataProvider else {
+                return XCTFail("Expected GitHubPackageMetadataProvider")
+            }
+            XCTAssertNil(githubMetadataProvider.configuration.authTokens)
+        }
+
+        let authTokens: [AuthTokenType: String] = [.github("github.test"): "topsekret"]
+        packageCollections.updateAuthTokens(authTokens)
+
+        XCTAssertEqual(authTokens, packageCollections.configuration.authTokens)
+        do {
+            guard let githubMetadataProvider = packageCollections.metadataProvider as? GitHubPackageMetadataProvider else {
+                return XCTFail("Expected GitHubPackageMetadataProvider")
+            }
+            XCTAssertEqual(authTokens, githubMetadataProvider.configuration.authTokens)
+        }
+    }
+
     func testBasicRegistration() throws {
         try skipIfUnsupportedPlatform()