/*
 * 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 <AzCore/RTTI/BehaviorContext.h>

#include <ScriptCanvas/AWSScriptBehaviorDynamoDB.h>
#include <TestFramework/AWSCoreFixture.h>

using namespace AWSCore;

class AWSScriptBehaviorDynamoDBNotificationBusHandlerMock
    : public AWSScriptBehaviorDynamoDBNotificationBusHandler
{
public:
    AZ_CLASS_ALLOCATOR(AWSScriptBehaviorDynamoDBNotificationBusHandlerMock, AZ::SystemAllocator)
    AWSScriptBehaviorDynamoDBNotificationBusHandlerMock()
    {
        AWSScriptBehaviorDynamoDBNotificationBus::Handler::BusConnect();
    }

    ~AWSScriptBehaviorDynamoDBNotificationBusHandlerMock() override
    {
        AWSScriptBehaviorDynamoDBNotificationBus::Handler::BusDisconnect();
    }

    MOCK_METHOD1(OnGetItemSuccess, void(const DynamoDBAttributeValueMap&));
    MOCK_METHOD1(OnGetItemError, void(const AZStd::string&));
};

using AWSScriptBehaviorDynamoDBTest = AWSCoreFixture;

TEST_F(AWSScriptBehaviorDynamoDBTest, GetItemRaw_CallWithEmptyTableName_InvokeOnError)
{
    AWSScriptBehaviorDynamoDBNotificationBusHandlerMock dynamodbHandlerMock;
    EXPECT_CALL(dynamodbHandlerMock, OnGetItemError(::testing::_)).Times(1);
    DynamoDBAttributeValueMap dummyMap;
    AWSScriptBehaviorDynamoDB::GetItemRaw("", dummyMap, "dummyRegion");
}

TEST_F(AWSScriptBehaviorDynamoDBTest, GetItemRaw_CallWithEmptyKeyMap_InvokeOnError)
{
    AWSScriptBehaviorDynamoDBNotificationBusHandlerMock dynamodbHandlerMock;
    EXPECT_CALL(dynamodbHandlerMock, OnGetItemError(::testing::_)).Times(1);
    DynamoDBAttributeValueMap dummyMap;
    AWSScriptBehaviorDynamoDB::GetItemRaw("dummyTable", dummyMap, "dummyRegion");
}

TEST_F(AWSScriptBehaviorDynamoDBTest, GetItemRaw_CallWithInvalidKeyMap_InvokeOnError)
{
    AWSScriptBehaviorDynamoDBNotificationBusHandlerMock dynamodbHandlerMock;
    EXPECT_CALL(dynamodbHandlerMock, OnGetItemError(::testing::_)).Times(1);
    DynamoDBAttributeValueMap dummyMap;
    dummyMap.emplace("dummyKey", "{invalidJsonFormat}");
    AWSScriptBehaviorDynamoDB::GetItemRaw("dummyTable", dummyMap, "dummyRegion");
}

TEST_F(AWSScriptBehaviorDynamoDBTest, GetItemRaw_CallWithEmptyRegionName_InvokeOnError)
{
    AWSScriptBehaviorDynamoDBNotificationBusHandlerMock dynamodbHandlerMock;
    EXPECT_CALL(dynamodbHandlerMock, OnGetItemError(::testing::_)).Times(1);
    DynamoDBAttributeValueMap dummyMap;
    dummyMap.emplace("dummyKey", "{}");
    AWSScriptBehaviorDynamoDB::GetItemRaw("dummyTable", dummyMap, "");
}

TEST_F(AWSScriptBehaviorDynamoDBTest, GetItem_NoTableNameInResourceMappingFound_InvokeOnError)
{
    AWSScriptBehaviorDynamoDBNotificationBusHandlerMock dynamodbHandlerMock;
    EXPECT_CALL(dynamodbHandlerMock, OnGetItemError(::testing::_)).Times(1);
    DynamoDBAttributeValueMap dummyMap;
    AWSScriptBehaviorDynamoDB::GetItem("dummyTable", dummyMap);
}

TEST_F(AWSScriptBehaviorDynamoDBTest, OnSuccessOnError_Call_GetExpectedNumOfInvoke)
{
    AWSScriptBehaviorDynamoDBNotificationBusHandlerMock dynamodbHandlerMock;
    EXPECT_CALL(dynamodbHandlerMock, OnGetItemSuccess(::testing::_)).Times(1);
    EXPECT_CALL(dynamodbHandlerMock, OnGetItemError(::testing::_)).Times(1);
    DynamoDBAttributeValueMap dummyMap;
    AWSScriptBehaviorDynamoDBNotificationBus::Broadcast(&AWSScriptBehaviorDynamoDBNotificationBus::Events::OnGetItemSuccess, dummyMap);
    AWSScriptBehaviorDynamoDBNotificationBus::Broadcast(&AWSScriptBehaviorDynamoDBNotificationBus::Events::OnGetItemError, "dummy success message");
}