o3de/Gems/AWSCore/Code/Tests/Framework/AWSApiClientJobConfigTest.cpp

/*
 * 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 <aws/s3/S3Client.h>

#include <Credential/AWSCredentialBus.h>
#include <Framework/AWSApiClientJobConfig.h>
#include <TestFramework/AWSCoreFixture.h>

using namespace AWSCore;

class AWSApiClientJobConfigTest
    : public AWSCoreFixture
    , public AWSCredentialRequestBus::Handler
{
public:
    AWSApiClientJobConfigTest()
        : m_credentialHandlerCounter(0)
    {
    }

    void SetUp() override
    {
        AWSCoreFixture::SetUpFixture();

        m_credentialHandlerCounter = 0;
    }

    // AWSCredentialRequestBus interface implementation
    int GetCredentialHandlerOrder() const override { return -1; }
    std::shared_ptr<Aws::Auth::AWSCredentialsProvider> GetCredentialsProvider() override
    {
        m_credentialHandlerCounter++;
        return m_credentialsHandler;
    }

    std::shared_ptr<Aws::Auth::AWSCredentialsProvider> m_credentialsHandler;
    int m_credentialHandlerCounter;
};

TEST_F(AWSApiClientJobConfigTest, GetClient_HaveHandlerReturnCredentialsProvider_CreateClientWithExpectedCredential)
{
    AWSCredentialRequestBus::Handler::BusConnect();
    m_credentialsHandler = std::make_shared<Aws::Auth::AnonymousAWSCredentialsProvider>();
    AwsApiClientJobConfig<Aws::S3::S3Client> testClientJobConfig;
    auto actualClient = testClientJobConfig.GetClient();
    EXPECT_TRUE(m_credentialHandlerCounter == 1);
    EXPECT_TRUE(actualClient);
    AWSCredentialRequestBus::Handler::BusDisconnect();
}

TEST_F(AWSApiClientJobConfigTest, GetClient_NoHandlerReturnCredentialsProvider_CreateClientWithDefaultCredential)
{
    AwsApiClientJobConfig<Aws::S3::S3Client> testClientJobConfig;
    auto actualClient = testClientJobConfig.GetClient();
    EXPECT_TRUE(m_credentialHandlerCounter == 0);
    EXPECT_TRUE(actualClient);
}