"""Test connecting to MySQL and PostgreSQL via SSH tunnel on localhost."""

import mysql.connector
import psycopg

HOST = "localhost"
USER = "REPLACE_WITH_GROUP_NAME"
PASSWORD = "REPLACE_WITH_GROUP_PASSWORD"

print("Connecting to MySQL...", end="", flush=True)
con = mysql.connector.connect(host=HOST, user=USER, password=PASSWORD, database=USER)
cur = con.cursor()
cur.execute("SELECT 'Success!'")
print(cur.fetchone()[0])
cur.close()
con.close()

print("Connecting to PostgreSQL...", end="", flush=True)
con = psycopg.connect(host=HOST, user=USER, password=PASSWORD, dbname=USER)
cur = con.cursor()
cur.execute("SELECT 'Success!'")
print(cur.fetchone()[0])
cur.close()
con.close()