1
0
Fork 0
harlequin-odbc/tests/conftest.py

28 lines
999 B
Python
Raw Normal View History

from __future__ import annotations
from typing import Generator
import pyodbc
import pytest
from harlequin_odbc.adapter import (
HarlequinOdbcAdapter,
HarlequinOdbcConnection,
)
MASTER_DB_CONN = "Driver={ODBC Driver 18 for SQL Server};Server=tcp:localhost,1433;Database=master;Uid=sa;Pwd={for-testing};Encrypt=yes;TrustServerCertificate=yes;Connection Timeout=5;" # noqa: E501
TEST_DB_CONN = "Driver={ODBC Driver 18 for SQL Server};Server=tcp:localhost,1433;Database=test;Uid=sa;Pwd={for-testing};Encrypt=yes;TrustServerCertificate=yes;Connection Timeout=5;" # noqa: E501
@pytest.fixture
def connection() -> Generator[HarlequinOdbcConnection, None, None]:
master_conn = pyodbc.connect(MASTER_DB_CONN, autocommit=True)
cur = master_conn.cursor()
cur.execute("drop database if exists test;")
cur.execute("create database test;")
cur.close()
master_conn.close()
conn = HarlequinOdbcAdapter(conn_str=(TEST_DB_CONN,)).connect()
yield conn
conn.close()