from lolafect.lolaconfig import build_lolaconfig
from lolafect.connections import connect_to_trino, close_trino_connection

#__          __     _____  _   _ _____ _   _  _____ _
#\ \        / /\   |  __ \| \ | |_   _| \ | |/ ____| |
# \ \  /\  / /  \  | |__) |  \| | | | |  \| | |  __| |
#  \ \/  \/ / /\ \ |  _  /| . ` | | | | . ` | | |_ | |
#   \  /\  / ____ \| | \ \| |\  |_| |_| |\  | |__| |_|
#    \/  \/_/    \_\_|  \_\_| \_|_____|_| \_|\_____(_)
# This testing suite requires:
# - The calling shell to have permission in AWS
# - The calling shell to be within the Mercadão network
# - Do not use this tests as part of CI/CD pipelines since they are not idempotent and
#   rely on many external. Instead, use them manually to check yourself that things are
#   working properly.
TEST_LOLACONFIG = build_lolaconfig(flow_name="testing-suite")


def test_that_trino_connect_and_disconnect_works_properly():

    connection = connect_to_trino.run(trino_credentials=TEST_LOLACONFIG.TRINO_CREDENTIALS)

    connection.cursor().execute("SELECT 1")

    close_trino_connection.run(trino_connection=connection)