#
# 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.
#
# $Revision: #1 $

import unittest
from unittest import mock
from resource_manager_common import constant

from resource_types import Custom_ResourceGroupConfiguration
from cgf_utils import custom_resource_response


class UnitTest_CloudGemFramework_ProjectResourceHandler_ResourceGroupConfiguration(unittest.TestCase):

    event = {}

    context = {}

    def setUp(self):
        self.event = {
            'ResourceProperties': {
                'ConfigurationBucket': 'TestBucket',
                'ConfigurationKey': 'TestKey',
                'ResourceGroupName': 'TestResourceGroup'
            },
            'StackId': 'arn:aws:cloudformation:TestRegion:TestAccount:stack/TestStack/TestUUID'
        }


    def test_handler(self):

        expected_data = {
            'ConfigurationBucket': 'TestBucket',
            'ConfigurationKey': 'TestKey/resource-group/TestResourceGroup',
            'TemplateURL': 'https://s3.amazonaws.com/TestBucket/TestKey/resource-group/TestResourceGroup/'+constant.RESOURCE_GROUP_TEMPLATE_FILENAME
        }

        expected_physical_id = 'CloudCanvas:LambdaConfiguration:TestStack:TestResourceGroup'
                
        with mock.patch.object(custom_resource_response, 'success_response') as mock_custom_resource_response_succeed:
            Custom_ResourceGroupConfiguration.handler(self.event, self.context)
            mock_custom_resource_response_succeed.assert_called_with(expected_data, expected_physical_id)