import integration_scripts

from integration_scripts import s3_backoff_utils as s3_utils


def check_logging():
    integration_scripts.logger.debug('logging works.')


def check_mysql():
    result = integration_scripts.db_utils.get_ar_mysql_results(
        sql='SELECT * FROM artist_info limit 1')

    integration_scripts.logger.debug('>----------| MySQL |------------<')
    integration_scripts.logger.debug(result.next())


def check_snowflake():
    result = integration_scripts.db_utils.get_snowflake_results(
        sql='SELECT * FROM TEST_INGESTION LIMIT 1'
    )

    integration_scripts.logger.debug('>----------| Snowflake |------------<')
    integration_scripts.logger.debug(result[0])


def check_s3():
    key = 'tmp/test.txt'
    msg = 'test string'
    bucket = 'dev-lvona'

    s3_utils.put_string_to_s3(key, msg, bucket)
    integration_scripts.logger.debug('>----------| AWS S3 |------------<')
    integration_scripts.logger.debug(
        'String \'{}\' written to file {} in bucket {}'.format(
            key, msg, bucket))


def check_xlsx():
    pass


def check_xlsx_s3():
    pass


def main():
    check_logging()
    check_mysql()
    check_snowflake()
    check_s3()
    check_xlsx()
    check_xlsx_s3()


if __name__ == '__main__':
    main()