amplify-swift/AmplifyPlugins/Auth/Tests/AWSCognitoAuthPluginUnitTests/ResolverTests/AuthorizationState/AuthorizationTestData.swift

73 lines
2.8 KiB
Swift

//
// 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)
}
}