You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
212 lines
10 KiB
C++
212 lines
10 KiB
C++
/*
|
|
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
|
|
* its licensors.
|
|
*
|
|
* For complete copyright and license terms please see the LICENSE at the root of this
|
|
* distribution (the "License"). All use of this software is governed by the License,
|
|
* or, if provided, by the license below or the license accompanying this file. Do not
|
|
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
*
|
|
*/
|
|
|
|
#include <AzTest/AzTest.h>
|
|
#include <AzCore/UnitTest/TestTypes.h>
|
|
#include <UserManagement/AWSCognitoUserManagementController.h>
|
|
#include <AWSClientAuthGemMock.h>
|
|
#include <aws/cognito-idp/CognitoIdentityProviderErrors.h>
|
|
|
|
namespace AWSClientAuthUnitTest
|
|
{
|
|
class AWSCognitoUserManagementControllerLocalMock
|
|
: public AWSClientAuth::AWSCognitoUserManagementController
|
|
{
|
|
public:
|
|
using AWSClientAuth::AWSCognitoUserManagementController::m_settings;
|
|
};
|
|
}
|
|
|
|
|
|
class AWSCognitoUserManagementControllerTest
|
|
: public AWSClientAuthUnitTest::AWSClientAuthGemAllocatorFixture
|
|
, public AWSCore::AWSCoreRequestBus::Handler
|
|
{
|
|
protected:
|
|
void SetUp() override
|
|
{
|
|
AWSClientAuthUnitTest::AWSClientAuthGemAllocatorFixture::SetUp();
|
|
AWSClientAuth::AWSCognitoUserManagementSetting::Reflect(*m_serializeContext);
|
|
m_mockController = AZStd::make_unique<AWSClientAuthUnitTest::AWSCognitoUserManagementControllerLocalMock>();
|
|
|
|
AWSCore::AWSCoreRequestBus::Handler::BusConnect();
|
|
}
|
|
|
|
void TearDown() override
|
|
{
|
|
AWSCore::AWSCoreRequestBus::Handler::BusDisconnect();
|
|
m_mockController.reset();
|
|
AWSClientAuthUnitTest::AWSClientAuthGemAllocatorFixture::TearDown();
|
|
}
|
|
|
|
// AWSCore::AWSCoreRequestBus overrides
|
|
AZ::JobContext* GetDefaultJobContext() override
|
|
{
|
|
return m_jobContext.get();
|
|
}
|
|
|
|
// Returns the default client configuration setting to use as a starting point in AWS requests
|
|
AWSCore::AwsApiJobConfig* GetDefaultConfig() override
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
public:
|
|
AZStd::unique_ptr<AWSClientAuthUnitTest::AWSCognitoUserManagementControllerLocalMock> m_mockController;
|
|
};
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, Initialize_Success)
|
|
{
|
|
AZStd::string path = AZStd::string::format("%s/%s/awsCognitoUserManagement.setreg",
|
|
m_testFolder->c_str(), AZ::SettingsRegistryInterface::RegistryFolder);
|
|
CreateTestFile("awsCognitoUserManagement.setreg"
|
|
, R"({"AWS":
|
|
{
|
|
"CognitoUserPool":
|
|
{
|
|
"AppClientId": "TestClientId",
|
|
"SignUpConfirmationType": "email"
|
|
}
|
|
}
|
|
})");
|
|
|
|
ASSERT_TRUE(m_mockController->Initialize(path));
|
|
ASSERT_EQ(m_mockController->m_settings->m_appClientId, "TestClientId");
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, EmailSignUp_Success)
|
|
{
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, SignUp(testing::_)).Times(1);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnEmailSignUpSuccess(testing::_)).Times(1);
|
|
m_mockController->EmailSignUpAsync(AWSClientAuthUnitTest::TEST_USERNAME, AWSClientAuthUnitTest::TEST_PASSWORD, AWSClientAuthUnitTest::TEST_EMAIL);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, EmailSignUp_Fail_SignUpError)
|
|
{
|
|
Aws::Client::AWSError<Aws::CognitoIdentityProvider::CognitoIdentityProviderErrors> error;
|
|
error.SetExceptionName(AWSClientAuthUnitTest::TEST_EXCEPTION);
|
|
Aws::CognitoIdentityProvider::Model::SignUpOutcome outcome(error);
|
|
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, SignUp(testing::_)).Times(1)
|
|
.WillOnce(testing::Return(outcome));
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnEmailSignUpSuccess(testing::_)).Times(0);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnEmailSignUpFail(testing::_)).Times(1);
|
|
m_mockController->EmailSignUpAsync(AWSClientAuthUnitTest::TEST_USERNAME, AWSClientAuthUnitTest::TEST_PASSWORD, AWSClientAuthUnitTest::TEST_EMAIL);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, PhoneSignUp_Success)
|
|
{
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, SignUp(testing::_)).Times(1);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnPhoneSignUpSuccess(testing::_)).Times(1);
|
|
m_mockController->PhoneSignUpAsync(AWSClientAuthUnitTest::TEST_USERNAME, AWSClientAuthUnitTest::TEST_PASSWORD, AWSClientAuthUnitTest::TEST_PHONE);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, PhoneSignUp_Fail_SignUpError)
|
|
{
|
|
Aws::Client::AWSError<Aws::CognitoIdentityProvider::CognitoIdentityProviderErrors> error;
|
|
error.SetExceptionName(AWSClientAuthUnitTest::TEST_EXCEPTION);
|
|
Aws::CognitoIdentityProvider::Model::SignUpOutcome outcome(error);
|
|
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, SignUp(testing::_)).Times(1)
|
|
.WillOnce(testing::Return(outcome));
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnPhoneSignUpSuccess(testing::_)).Times(0);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnPhoneSignUpFail(testing::_)).Times(1);
|
|
m_mockController->PhoneSignUpAsync(AWSClientAuthUnitTest::TEST_USERNAME, AWSClientAuthUnitTest::TEST_PASSWORD, AWSClientAuthUnitTest::TEST_PHONE);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, ConfirmSignUp_Success)
|
|
{
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, ConfirmSignUp(testing::_)).Times(1);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnConfirmSignUpSuccess()).Times(1);
|
|
m_mockController->ConfirmSignUpAsync(AWSClientAuthUnitTest::TEST_USERNAME, AWSClientAuthUnitTest::TEST_CODE);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, ConfirmSignUp_Fail_ConfirmSignUpError)
|
|
{
|
|
Aws::Client::AWSError<Aws::CognitoIdentityProvider::CognitoIdentityProviderErrors> error;
|
|
error.SetExceptionName(AWSClientAuthUnitTest::TEST_EXCEPTION);
|
|
Aws::CognitoIdentityProvider::Model::ConfirmSignUpOutcome outcome(error);
|
|
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, ConfirmSignUp(testing::_)).Times(1)
|
|
.WillOnce(testing::Return(outcome));
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnConfirmSignUpSuccess()).Times(0);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnConfirmSignUpFail(testing::_)).Times(1);
|
|
m_mockController->ConfirmSignUpAsync(AWSClientAuthUnitTest::TEST_USERNAME, AWSClientAuthUnitTest::TEST_CODE);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, EnableMFA_Success)
|
|
{
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, SetUserMFAPreference(testing::_)).Times(1);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnEnableMFASuccess()).Times(1);
|
|
m_mockController->EnableMFAAsync(AWSClientAuthUnitTest::TEST_TOKEN);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, EnableMFA_Fail_SetUserMFAPreferenceError)
|
|
{
|
|
Aws::Client::AWSError<Aws::CognitoIdentityProvider::CognitoIdentityProviderErrors> error;
|
|
error.SetExceptionName(AWSClientAuthUnitTest::TEST_EXCEPTION);
|
|
Aws::CognitoIdentityProvider::Model::SetUserMFAPreferenceOutcome outcome(error);
|
|
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, SetUserMFAPreference(testing::_)).Times(1)
|
|
.WillOnce(testing::Return(outcome));
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnEnableMFASuccess()).Times(0);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnEnableMFAFail(testing::_)).Times(1);
|
|
m_mockController->EnableMFAAsync(AWSClientAuthUnitTest::TEST_TOKEN);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, ForgotPassword_Success)
|
|
{
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, ForgotPassword(testing::_)).Times(1);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnForgotPasswordSuccess()).Times(1);
|
|
m_mockController->ForgotPasswordAsync(AWSClientAuthUnitTest::TEST_USERNAME);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, ForgotPassword_Fail_ForgotPasswordError)
|
|
{
|
|
Aws::Client::AWSError<Aws::CognitoIdentityProvider::CognitoIdentityProviderErrors> error;
|
|
error.SetExceptionName(AWSClientAuthUnitTest::TEST_EXCEPTION);
|
|
Aws::CognitoIdentityProvider::Model::ForgotPasswordOutcome outcome(error);
|
|
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, ForgotPassword(testing::_)).Times(1)
|
|
.WillOnce(testing::Return(outcome));
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnForgotPasswordSuccess()).Times(0);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnForgotPasswordFail(testing::_)).Times(1);
|
|
m_mockController->ForgotPasswordAsync(AWSClientAuthUnitTest::TEST_USERNAME);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, ConfirmForgotPassword_Success)
|
|
{
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, ConfirmForgotPassword(testing::_)).Times(1);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnConfirmForgotPasswordSuccess()).Times(1);
|
|
m_mockController->ConfirmForgotPasswordAsync(AWSClientAuthUnitTest::TEST_USERNAME, AWSClientAuthUnitTest::TEST_CODE, AWSClientAuthUnitTest::TEST_NEW_PASSWORD);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, ConfirmForgotPassword_Fail_ConfirmForgotPasswordError)
|
|
{
|
|
Aws::Client::AWSError<Aws::CognitoIdentityProvider::CognitoIdentityProviderErrors> error;
|
|
error.SetExceptionName("TestException");
|
|
Aws::CognitoIdentityProvider::Model::ConfirmForgotPasswordOutcome outcome(error);
|
|
|
|
EXPECT_CALL(*m_cognitoIdentityProviderClientMock, ConfirmForgotPassword(testing::_)).Times(1)
|
|
.WillOnce(testing::Return(outcome));
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnConfirmForgotPasswordSuccess()).Times(0);
|
|
EXPECT_CALL(m_awsCognitoUserManagementNotificationsBusMock, OnConfirmForgotPasswordFail(testing::_)).Times(1);
|
|
m_mockController->ConfirmForgotPasswordAsync(AWSClientAuthUnitTest::TEST_USERNAME, AWSClientAuthUnitTest::TEST_CODE, AWSClientAuthUnitTest::TEST_NEW_PASSWORD);
|
|
}
|
|
|
|
TEST_F(AWSCognitoUserManagementControllerTest, Initialize_Fail_InvalidPath)
|
|
{
|
|
AZ_TEST_START_TRACE_SUPPRESSION;
|
|
ASSERT_FALSE(m_mockController->Initialize(""));
|
|
AZ_TEST_STOP_TRACE_SUPPRESSION(1);
|
|
}
|