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.
134 lines
3.6 KiB
C++
134 lines
3.6 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 <aws/core/auth/AWSCredentialsProvider.h>
|
|
|
|
#include <Credential/AWSCredentialBus.h>
|
|
#include <TestFramework/AWSCoreFixture.h>
|
|
|
|
using namespace AWSCore;
|
|
|
|
class TestCredentialHandlerOne
|
|
: AWSCredentialRequestBus::Handler
|
|
{
|
|
public:
|
|
void ActivateHandler()
|
|
{
|
|
m_handlerCounter = 0;
|
|
m_credentialsProvider = std::make_shared<Aws::Auth::AnonymousAWSCredentialsProvider>();
|
|
AWSCredentialRequestBus::Handler::BusConnect();
|
|
}
|
|
|
|
void DeactivateHandler()
|
|
{
|
|
AWSCredentialRequestBus::Handler::BusDisconnect();
|
|
m_credentialsProvider.reset();
|
|
}
|
|
|
|
int GetCredentialHandlerOrder() const override
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
std::shared_ptr<Aws::Auth::AWSCredentialsProvider> GetCredentialsProvider() override
|
|
{
|
|
++m_handlerCounter;
|
|
return m_credentialsProvider;
|
|
}
|
|
|
|
std::shared_ptr<Aws::Auth::AWSCredentialsProvider> m_credentialsProvider;
|
|
AZStd::atomic<int> m_handlerCounter;
|
|
};
|
|
|
|
class TestCredentialHandlerTwo
|
|
: AWSCredentialRequestBus::Handler
|
|
{
|
|
public:
|
|
void ActivateHandler()
|
|
{
|
|
m_handlerCounter = 0;
|
|
m_credentialsProvider = std::make_shared<Aws::Auth::AnonymousAWSCredentialsProvider>();
|
|
AWSCredentialRequestBus::Handler::BusConnect();
|
|
}
|
|
|
|
void DeactivateHandler()
|
|
{
|
|
AWSCredentialRequestBus::Handler::BusDisconnect();
|
|
m_credentialsProvider.reset();
|
|
}
|
|
|
|
int GetCredentialHandlerOrder() const override
|
|
{
|
|
return 2;
|
|
}
|
|
|
|
std::shared_ptr<Aws::Auth::AWSCredentialsProvider> GetCredentialsProvider() override
|
|
{
|
|
++m_handlerCounter;
|
|
return m_credentialsProvider;
|
|
}
|
|
|
|
std::shared_ptr<Aws::Auth::AWSCredentialsProvider> m_credentialsProvider;
|
|
AZStd::atomic<int> m_handlerCounter;
|
|
};
|
|
|
|
class AWSCredentialBusTest
|
|
: public AWSCoreFixture
|
|
{
|
|
public:
|
|
AWSCredentialBusTest()
|
|
{
|
|
m_handlerOne = AZStd::make_unique<TestCredentialHandlerOne>();
|
|
m_handlerTwo = AZStd::make_unique<TestCredentialHandlerTwo>();
|
|
}
|
|
|
|
void SetUp() override
|
|
{
|
|
AWSCoreFixture::SetUpFixture();
|
|
|
|
m_handlerOne->ActivateHandler();
|
|
m_handlerTwo->ActivateHandler();
|
|
}
|
|
|
|
void TearDown() override
|
|
{
|
|
m_handlerOne->DeactivateHandler();
|
|
m_handlerTwo->DeactivateHandler();
|
|
|
|
AWSCoreFixture::TearDownFixture();
|
|
}
|
|
|
|
AZStd::unique_ptr<TestCredentialHandlerOne> m_handlerOne;
|
|
AZStd::unique_ptr<TestCredentialHandlerTwo> m_handlerTwo;
|
|
};
|
|
|
|
TEST_F(AWSCredentialBusTest, GetCredentialsProvider_CallFromMultithread_GetExpectedCredentialsProviderAndNumberOfCalls)
|
|
{
|
|
constexpr int testThreadNumber = 10;
|
|
AZStd::atomic<int> actualEbusCalls = 0;
|
|
AZStd::vector<AZStd::thread> testThreadPool;
|
|
for (int index = 0; index < testThreadNumber; ++index)
|
|
{
|
|
testThreadPool.emplace_back(AZStd::thread([&]() {
|
|
AWSCredentialResult result;
|
|
AWSCredentialRequestBus::BroadcastResult(result, &AWSCredentialRequests::GetCredentialsProvider);
|
|
ASSERT_TRUE(result.result);
|
|
EXPECT_TRUE(result.result == m_handlerOne->m_credentialsProvider);
|
|
actualEbusCalls++;
|
|
}));
|
|
}
|
|
|
|
for (auto& testThread : testThreadPool)
|
|
{
|
|
testThread.join();
|
|
}
|
|
EXPECT_TRUE(m_handlerOne->m_handlerCounter == testThreadNumber);
|
|
EXPECT_TRUE(m_handlerTwo->m_handlerCounter == testThreadNumber);
|
|
EXPECT_TRUE(actualEbusCalls == testThreadNumber);
|
|
}
|