//
// Copyright Amazon.com Inc. or its affiliates.
// All Rights Reserved.
//
// SPDX-License-Identifier: Apache-2.0
//

import Amplify
import Foundation

@testable import AWSCognitoAuthPlugin

extension AWSAuthCognitoSession {
    static var testData: AWSAuthCognitoSession {
        AWSAuthCognitoSession(isSignedIn: true,
                              identityIdResult: .success("identityId"),
                              awsCredentialsResult: .success(AuthAWSCognitoCredentials.testData),
                              cognitoTokensResult: .success(AWSCognitoUserPoolTokens.testData))
    }
}

extension AuthAWSCognitoCredentials {
    static var testData: AuthAWSCognitoCredentials {
        AuthAWSCognitoCredentials(accessKeyId: "accessKey",
                                  secretAccessKey: "secretAccessKey",
                                  sessionToken: "sessionToken",
                                  expiration: Date() + 121)
    }

    static var expiredTestData: AuthAWSCognitoCredentials {
        AuthAWSCognitoCredentials(accessKeyId: "accessKey",
                                  secretAccessKey: "secretAccessKey",
                                  sessionToken: "sessionToken",
                                  expiration: Date() - 10000)
    }
}

extension FederatedToken {
    static var testData: FederatedToken {
        return  .init(token: "token", provider: .facebook)
    }
}

extension AmplifyCredentials {
    static var testData: AmplifyCredentials {
        AmplifyCredentials.userPoolAndIdentityPool(signedInData: .testData,
                                                   identityID: "identityId",
                                                   credentials: AuthAWSCognitoCredentials.testData)
    }

    static var testDataIdentityPool: AmplifyCredentials {
        AmplifyCredentials.identityPoolOnly(identityID: "someId",
                                            credentials: .testData)
    }

    static var testDataWithExpiredTokens: AmplifyCredentials {
        AmplifyCredentials.userPoolAndIdentityPool(signedInData: .expiredTestData,
                                                   identityID: "identityId",
                                                   credentials: AuthAWSCognitoCredentials.testData)
    }

    static var testDataWithExpiredAWSCredentials: AmplifyCredentials {
        AmplifyCredentials.userPoolAndIdentityPool(signedInData: .testData,
                                                   identityID: "identityId",
                                                   credentials: AuthAWSCognitoCredentials.expiredTestData)
    }

    static var testDataIdentityPoolWithExpiredTokens: AmplifyCredentials {
        AmplifyCredentials.identityPoolOnly(identityID: "identityId",
                                            credentials: AuthAWSCognitoCredentials.testData)
    }
}