Steve Pham
38261d0800
* Updated all copyright headers to split the longer original copyright line into 2 shorter lines Signed-off-by: Steve Pham <spham@amazon.com>
72 lines
2.7 KiB
C++
72 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) Contributors to the Open 3D Engine Project.
|
|
* For complete copyright and license terms please see the LICENSE at the root of this distribution.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 OR MIT
|
|
*
|
|
*/
|
|
|
|
#include <AzTest/AzTest.h>
|
|
#include <AzCore/UnitTest/TestTypes.h>
|
|
#include <AWSClientAuthGemMock.h>
|
|
#include <Authorization/AWSClientAuthPersistentCognitoIdentityProvider.h>
|
|
|
|
|
|
class AWSClientAuthPersistentCognitoIdentityProviderTest : public AWSClientAuthUnitTest::AWSClientAuthGemAllocatorFixture
|
|
{
|
|
protected:
|
|
void SetUp() override
|
|
{
|
|
AWSClientAuthUnitTest::AWSClientAuthGemAllocatorFixture::SetUp();
|
|
|
|
}
|
|
|
|
void TearDown() override
|
|
{
|
|
AWSClientAuthUnitTest::AWSClientAuthGemAllocatorFixture::TearDown();
|
|
}
|
|
|
|
public:
|
|
|
|
};
|
|
|
|
TEST_F(AWSClientAuthPersistentCognitoIdentityProviderTest, Initialize_AndPersisteIdentityId_Success)
|
|
{
|
|
AWSClientAuth::AWSClientAuthPersistentCognitoIdentityProvider provider;
|
|
provider.Initialize(AWSClientAuthUnitTest::TEST_ACCOUNT_ID, AWSClientAuthUnitTest::TEST_IDENTITY_POOL_ID);
|
|
|
|
ASSERT_TRUE(provider.GetAccountId() == AWSClientAuthUnitTest::TEST_ACCOUNT_ID);
|
|
ASSERT_TRUE(provider.GetIdentityPoolId() == AWSClientAuthUnitTest::TEST_IDENTITY_POOL_ID);
|
|
|
|
provider.PersistIdentityId(AWSClientAuthUnitTest::TEST_IDENTITY_ID);
|
|
ASSERT_TRUE(provider.GetIdentityId() == AWSClientAuthUnitTest::TEST_IDENTITY_ID);
|
|
}
|
|
|
|
TEST_F(AWSClientAuthPersistentCognitoIdentityProviderTest, AddRemoveLogins_Success)
|
|
{
|
|
AWSClientAuth::AWSClientAuthPersistentCognitoIdentityProvider provider;
|
|
provider.Initialize(AWSClientAuthUnitTest::TEST_ACCOUNT_ID, AWSClientAuthUnitTest::TEST_IDENTITY_POOL_ID);
|
|
|
|
ASSERT_TRUE(provider.HasLogins() == false);
|
|
Aws::Map<Aws::String, Aws::Auth::LoginAccessTokens> logins;
|
|
Aws::Auth::LoginAccessTokens tokens;
|
|
tokens.accessToken = "TestToken";
|
|
logins.insert(std::pair<Aws::String, Aws::Auth::LoginAccessTokens>("TestLoginKey1", tokens));
|
|
logins.insert(std::pair<Aws::String, Aws::Auth::LoginAccessTokens>("TestLoginKey2", tokens));
|
|
provider.PersistLogins(logins);
|
|
|
|
ASSERT_TRUE(provider.HasLogins() == true);
|
|
ASSERT_TRUE(provider.GetLogins().size() == 2);
|
|
ASSERT_TRUE(provider.GetLogins()["TestLoginKey1"].accessToken == tokens.accessToken);
|
|
ASSERT_TRUE(provider.GetLogins()["TestLoginKey2"].accessToken == tokens.accessToken);
|
|
|
|
provider.RemoveLogin("TestLoginKey1");
|
|
ASSERT_TRUE(provider.HasLogins() == true);
|
|
ASSERT_TRUE(provider.GetLogins().size() == 1);
|
|
ASSERT_TRUE(provider.GetLogins()["TestLoginKey2"].accessToken == tokens.accessToken);
|
|
|
|
provider.RemoveLogin("TestLoginKey2");
|
|
ASSERT_TRUE(provider.HasLogins() == false);
|
|
ASSERT_TRUE(provider.GetLogins().size() == 0);
|
|
}
|