def test_sqlcreds_connection_from_sqlalchemy(sql_creds):
"""
Simple test to ensure that the generated creds can connect to the database by
interpretting the connection from a SQLAlchemy engine
The sql_creds fixture necessarily uses username and password (no Windows auth)
"""
# Create an actual SQL Alchemy connection
# n.b. Keys are case sensitive
conn_str = (
f"DRIVER={sql_creds.driver};"
f"Server={sql_creds.server};"
f"Database={sql_creds.database};"
f"UID={sql_creds.username};"
f"PWD={sql_creds.password};"
)
params = urllib.parse.quote_plus(conn_str)
mssql_engine = create_engine("mssql+pyodbc:///?odbc_connect=%s" % params)
# Re-interpret using SqlCreds
test_engine = SqlCreds.from_engine(mssql_engine).engine
# Check the SqlCreds version works
df = pd.read_sql(con=test_engine, sql="SELECT TOP 1 * FROM sys.objects")
assert df.shape[0] == 1
def test_sql_creds_from_sqlalchemy_non_default_port():
"""
Tests that the SqlCreds object can be created from a SqlAlchemy engine
* With Username and Password
* Non-Default Port specified (9999)
"""
params = urllib.parse.quote_plus(
"Driver={ODBC Driver 99 for SQL Server};Server=tcp:test_server,9999;Database=test_database;"
"UID=test_user;PWD=test_password"
)
mssql_engine = create_engine("mssql+pyodbc:///?odbc_connect=%s" % params)
creds = SqlCreds.from_engine(mssql_engine)
assert creds.server == "test_server"
assert creds.database == "test_database"
assert creds.username == "test_user"
assert creds.password == "test_password"
assert creds.port == 9999
assert creds.with_krb_auth is False
assert isinstance(creds.engine, engine.Connectable)
assert str(creds.engine.url) == (
"mssql+pyodbc:///?odbc_connect="
+ "Driver={ODBC Driver 99 for SQL Server};Server=tcp:test_server,9999;Database=test_database;"
+ "UID=test_user;PWD=test_password"
)
def setup(docker_db):
# start docker container
docker_db.start()
time.sleep(20) # wait for container to start up
# create database
docker_db.create_database("benchmark_db")
# create creds
creds = SqlCreds.from_engine(
docker_db.create_engine(db_name="benchmark_db"))
return creds
def test_sql_creds_from_sqlalchemy_windows_auth_blank_port():
"""
Tests that the SqlCreds object can be created from a SqlAlchemy engine - without Username, Password or Port
"""
params = quote_plus(
"Driver={ODBC Driver 99 for SQL Server};Server=tcp:test_server;Database=test_database"
)
mssql_engine = create_engine("mssql+pyodbc:///?odbc_connect=%s" % params)
creds = SqlCreds.from_engine(mssql_engine)
assert creds.server == "test_server"
assert creds.database == "test_database"
assert creds.username == ""
assert creds.password == ""
assert creds.port is None
assert creds.with_krb_auth is True
assert isinstance(creds.engine, engine.Connectable)
assert str(creds.engine.url) == _quote_engine_url(
"Driver={ODBC Driver 99 for SQL Server};Server=tcp:test_server;Database=test_database"
)
def test_sql_creds_from_sqlalchemy_blank_port():
"""
Tests that the SqlCreds object can be created from a SqlAlchemy engine with no Port specified
"""
params = quote_plus(
"Driver={ODBC Driver 99 for SQL Server};Server=tcp:test_server;Database=test_database;"
"UID=test_user;PWD=test_password"
)
mssql_engine = create_engine("mssql+pyodbc:///?odbc_connect=%s" % params)
creds = SqlCreds.from_engine(mssql_engine)
assert creds.server == "test_server"
assert creds.database == "test_database"
assert creds.username == "test_user"
assert creds.password == "test_password"
assert creds.port is None
assert creds.with_krb_auth is False
assert isinstance(creds.engine, engine.Connectable)
assert str(creds.engine.url) == _quote_engine_url(
"Driver={ODBC Driver 99 for SQL Server};Server=tcp:test_server;Database=test_database;"
+ "UID=test_user;PWD=test_password"
)
def test_sql_creds_from_sqlalchemy_windows_auth():
"""
Tests that the SqlCreds object can be created from a SqlAlchemy engine
* Without Username and Password
* Use default port (1433)
"""
params = urllib.parse.quote_plus(
"Driver={ODBC Driver 99 for SQL Server};Server=tcp:test_server,1433;Database=test_database"
)
mssql_engine = create_engine("mssql+pyodbc:///?odbc_connect=%s" % params)
creds = SqlCreds.from_engine(mssql_engine)
assert creds.server == "test_server"
assert creds.database == "test_database"
assert creds.username == ""
assert creds.password == ""
assert creds.port == 1433
assert creds.with_krb_auth is True
assert isinstance(creds.engine, engine.Connectable)
assert str(creds.engine.url) == (
"mssql+pyodbc:///?odbc_connect="
+ "Driver={ODBC Driver 99 for SQL Server};Server=tcp:test_server,1433;Database=test_database"
)