lolafect/tests/test_integration/test_connections.py

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 external resources. 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)