AuthenticatorTests.swift

// Copyright 2022 Esri.

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0

// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import XCTest
@testable import ArcGISToolkit
@testable import ArcGIS
import Combine

@MainActor final class AuthenticatorTests: XCTestCase {
    func testInit() {
        let config = OAuthUserConfiguration(
            portalURL: URL(string:"www.arcgis.com")!,
            clientID: "client id",
            redirectURL: URL(string:"myapp://oauth")!
        )
        let authenticator = Authenticator(promptForUntrustedHosts: true, oAuthUserConfigurations: [config])
        XCTAssertTrue(authenticator.promptForUntrustedHosts)
        XCTAssertEqual(authenticator.oAuthUserConfigurations, [config])
    }
    
    func testMakePersistent() async throws {
        // Make sure credential stores are restored.
        addTeardownBlock {
            ArcGISEnvironment.authenticationManager.arcGISCredentialStore = ArcGISCredentialStore()
            await ArcGISEnvironment.authenticationManager.setNetworkCredentialStore(NetworkCredentialStore())
        }
        
        // This tests that calling setupPersistentCredentialStorage tries to sync with the keychain.
        do {
            try await ArcGISEnvironment.authenticationManager.setupPersistentCredentialStorage(access: .whenUnlocked)
            XCTFail("Expected an error to be thrown as unit tests should not have access to the keychain")
        } catch {}
    }
    
    func testClearCredentialStores() async {
        ArcGISEnvironment.authenticationManager.arcGISCredentialStore.add(
            PregeneratedTokenCredential(
                url: URL(string: "www.arcgis.com")!,
                tokenInfo: .init(
                    accessToken: "token",
                    expirationDate: .distantFuture,
                    isSSLRequired: false
                )!
            )
        )
        
        var arcGISCreds = ArcGISEnvironment.authenticationManager.arcGISCredentialStore.credentials
        XCTAssertEqual(arcGISCreds.count, 1)
        
        await ArcGISEnvironment.authenticationManager.clearCredentialStores()
        
        arcGISCreds = ArcGISEnvironment.authenticationManager.arcGISCredentialStore.credentials
        XCTAssertTrue(arcGISCreds.isEmpty)
    }
}