#########################################################################################
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.                    #
# SPDX-License-Identifier: MIT-0                                                        #
#                                                                                       #
# Permission is hereby granted, free of charge, to any person obtaining a copy of this  #
# software and associated documentation files (the "Software"), to deal in the Software #
# without restriction, including without limitation the rights to use, copy, modify,    #
# merge, publish, distribute, sublicense, and/or sell copies of the Software, and to    #
# permit persons to whom the Software is furnished to do so.                            #
#                                                                                       #
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,   #
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A         #
# PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT    #
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION     #
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE        #
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                                #
#########################################################################################

import unittest
import json

import os
import sys
from pathlib import Path
import logging


def init():
    global logger
    LOGLEVEL = os.environ.get('LOGLEVEL', 'INFO').upper()
    logging.basicConfig(format='%(asctime)s | %(levelname)s | %(message)s', level=LOGLEVEL)
    logger = logging.getLogger('integration_unit_tests')

    # This is to get around the relative path import issue.
    # Absolute paths are being used in this file after setting the root directory
    file = Path(__file__).resolve()
    package_root_directory = file.parents[1]
    sys.path.append(str(package_root_directory))
    sys.path.append(str(package_root_directory) + '/credential_manager/lambdas')
    logging.debug(sys.path)

    os.environ['solution_identifier'] = '"AwsSolution/SO0097/CMF-ON-AWS"'


logger = None
init()


class CredentialManagerTestBase(unittest.TestCase):

    def setUp(self):
        os.environ['region'] = 'us-east-1'
        os.environ['AWS_ACCESS_KEY_ID'] = 'testing'
        os.environ['AWS_SECRET_ACCESS_KEY'] = 'testing'
        os.environ['AWS_SECURITY_TOKEN'] = 'testing'
        os.environ['AWS_SESSION_TOKEN'] = 'testing'
        os.environ['AWS_DEFAULT_REGION'] = 'us-east-1'

        self.init_event_vars()

    def init_event_vars(self):
        self.create_os_secret_event = {
            'body': json.dumps({
                'user': 'testUser',
                'password': 'testPassword',
                'secretName': 'secretName',
                'osType': 'osType',
                'secretType': 'OS',
                'description': 'description',
                'isSSHKey': 'MySSHKey',
            })
        }
        self.create_os_secret_event_2 = {
            'body': json.dumps({
                'secretName': 'secretName',
                'osType': 'osTypeUpdated',
                'secretType': 'OS',
            })
        }
        self.create_os_secret_event_updated = {
            'body': json.dumps({
                'user': 'testUserUpdated',
                'password': 'testPassword',
                'secretName': 'secretName',
                'osType': 'osType',
                'secretType': 'OS',
                'isSSHKey': 'MySSHKey',
            })
        }
        self.create_key_value_secret_event = {
            'body': json.dumps({
                'secretName': 'secretName',
                'secretKey': 'secretKey',
                'secretValue': 'secretValue',
                'secretType': 'keyValue',
                'description': 'description',
            })
        }
        self.create_key_value_secret_event_updated = {
            'body': json.dumps({
                'secretName': 'secretName',
                'secretKey': 'secretKeyUpdated',
                'secretValue': 'secretValue',
                'secretType': 'keyValue',
                'description': 'description',
            })
        }
        self.create_plain_text_secret_event = {
            'body': json.dumps({
                'secretName': 'secretName',
                'secretString': 'secretString',
                'secretType': 'plainText',
                'description': 'description',
            })
        }
        self.create_plain_text_secret_event_updated = {
            'body': json.dumps({
                'secretName': 'secretName',
                'secretString': 'secretStringUpdated',
                'secretType': 'plainText',
                'description': 'description',
            })
        }
        self.get_secret_event = {
            'queryStringParameters': {
                'Name': 'secretName',
            }
        }

        self.lambda_get_event = {
            'httpMethod': 'GET',
        }

        self.lambda_put_event = {
            'httpMethod': 'PUT',
            'body': json.dumps({
                'secretType': 'OS',
            }),
        }

        self.lambda_delete_event = {
            'httpMethod': 'DELETE',
            'body': json.dumps({
                'secretType': 'OS',
            }),
        }

        self.lambda_post_OS_event = {
            'httpMethod': 'POST',
            'body': json.dumps({
                'secretType': 'OS',
            }),
        }

        self.lambda_post_keyValue_event = {
            'httpMethod': 'POST',
            'body': json.dumps({
                'secretType': 'keyValue',
            }),
        }

        self.lambda_post_plainText_event = {
            'httpMethod': 'POST',
            'body': json.dumps({
                'secretType': 'plainText',
            }),
        }

    def tearDown(self):
        pass