mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-27 02:52:53 -04:00
Khairi Hafsham
772374735d
tested using pycodestyle version 2.2.0 Fixes: #3885 Change-Id: I5df43adc3aefe318f9eeab72a078247a548ec566 Pull-request: https://github.com/zzzeek/sqlalchemy/pull/343
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
from sqlalchemy import *
|
|
from sqlalchemy.testing import eq_
|
|
from sqlalchemy.testing import engines
|
|
from sqlalchemy.testing import fixtures
|
|
from sqlalchemy.testing.mock import Mock
|
|
|
|
|
|
def mock_dbapi():
|
|
return Mock(paramstyle='qmark',
|
|
connect=Mock(
|
|
return_value=Mock(
|
|
cursor=Mock(return_value=Mock(description=None,
|
|
rowcount=None)))))
|
|
|
|
|
|
class MxODBCTest(fixtures.TestBase):
|
|
|
|
def test_native_odbc_execute(self):
|
|
t1 = Table('t1', MetaData(), Column('c1', Integer))
|
|
dbapi = mock_dbapi()
|
|
|
|
engine = engines.testing_engine('mssql+mxodbc://localhost',
|
|
options={'module': dbapi,
|
|
'_initialize': False})
|
|
conn = engine.connect()
|
|
|
|
# crud: uses execute
|
|
conn.execute(t1.insert().values(c1='foo'))
|
|
conn.execute(t1.delete().where(t1.c.c1 == 'foo'))
|
|
conn.execute(t1.update().where(t1.c.c1 == 'foo').values(c1='bar'))
|
|
|
|
# select: uses executedirect
|
|
conn.execute(t1.select())
|
|
|
|
# manual flagging
|
|
conn.execution_options(native_odbc_execute=True).execute(t1.select())
|
|
conn.execution_options(native_odbc_execute=False)\
|
|
.execute(t1.insert().values(c1='foo'))
|
|
|
|
eq_([c[2] for c in
|
|
dbapi.connect.return_value.cursor.return_value.execute.mock_calls],
|
|
[{'direct': True}, {'direct': True}, {'direct': True},
|
|
{'direct': True}, {'direct': False}, {'direct': True}])
|