"""Script for executing the lambda locally against AWS Dev."""

import os
from time import sleep

from src import app as index

os.environ['SNS_GENERAL_STATUS_TOPIC_ARN'] = 'arn:aws:sns:us-east-1:103233932089:dev-asset_transcoder_general_status'
os.environ['MEDIAINFO_PATH'] = '/usr/local/bin/mediainfo'

success_event = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': '461f5444_a522_4ff1_b51f_17b699df621b.wav',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

success_flac_event = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': '51b5fccf_cc6e_48c7_9cbf_29c179f14b6d.flac',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

success_wav_48k_event = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': '95913417_f2a0_41d4_bb8c_f0c817780c4c.wav',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

success_wav_24b_event = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': '6058eaba_1036_4ae2_8b15_9def76e954a0.wav',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

success_m4a_event = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': 'b5a0ab9c_6f60_4a0d_a430_ec6cef52631e.m4a',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

not_found_event = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': '00392262_1dbe_4a14_bf1d_b649f9984e00.wav',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

invalid_event = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': '461f5444_a522_4ff1_b51f_17b699df621b.wav',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

bad_audio = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': 'dd4f9973_388c_45bf_b2d3_2a8e4a501c53.wav',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

bad_channels = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': 'e8d982d1_28b7_4ae3_a893_8f561b5c8d86.wav',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

bad_m4a = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': '9c539f1e_9adf_4692_8755_c00010f6a105.m4a',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}

bad_wav = {
    'bucket': 'dev-orcd-asset-transcoder-input',
    'key': '8d82ab0d_f83d_41ba_816c_662e8988f17a.wav',
    'file_type': 'jpg',
    'config': {
        'output_bucket_name': 'dev-orcd-podcast-output-assets',
        'preview_bucket_name': 'dev-orcd-podcast-output-assets',
        'elastic_transcoder_id': '1581101525157-xjcv85'
    }
}


def test_event(lambda_module, event):
    """Method for testing the handler function with a test event."""
    try:
        lambda_module.handler(event, None)
    except Exception:
        pass


if __name__ == '__main__':
    for event in [success_event, success_flac_event, success_m4a_event, success_wav_48k_event, success_wav_24b_event,
                  not_found_event, invalid_event, bad_audio, bad_channels, bad_m4a, bad_wav]:
        test_event(index, event)
        sleep(1)