mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-09 10:20:00 -04:00
Mike Bayer
20cdc64588
become an externally usable package but still remains within the main sqlalchemy parent package. in this system, we use kind of an ugly hack to get the noseplugin imported outside of the "sqlalchemy" package, while still making it available within sqlalchemy for usage by third party libraries.
142 lines
5.6 KiB
Python
142 lines
5.6 KiB
Python
# coding: utf-8
|
|
"""verrrrry basic unicode column name testing"""
|
|
|
|
from sqlalchemy import *
|
|
from sqlalchemy.testing import fixtures, engines
|
|
from sqlalchemy import testing
|
|
from sqlalchemy.testing.engines import utf8_engine
|
|
from sqlalchemy.sql import column
|
|
from sqlalchemy.testing.schema import Table, Column
|
|
|
|
class UnicodeSchemaTest(fixtures.TestBase):
|
|
__requires__ = ('unicode_ddl',)
|
|
|
|
@classmethod
|
|
def setup_class(cls):
|
|
global unicode_bind, metadata, t1, t2, t3
|
|
|
|
unicode_bind = utf8_engine()
|
|
|
|
metadata = MetaData(unicode_bind)
|
|
t1 = Table('unitable1', metadata,
|
|
Column(u'méil', Integer, primary_key=True),
|
|
Column(u'\u6e2c\u8a66', Integer),
|
|
test_needs_fk=True,
|
|
)
|
|
t2 = Table(u'Unitéble2', metadata,
|
|
Column(u'méil', Integer, primary_key=True, key="a"),
|
|
Column(u'\u6e2c\u8a66', Integer, ForeignKey(u'unitable1.méil'),
|
|
key="b"
|
|
),
|
|
test_needs_fk=True,
|
|
)
|
|
|
|
# Few DBs support Unicode foreign keys
|
|
if testing.against('sqlite'):
|
|
t3 = Table(u'\u6e2c\u8a66', metadata,
|
|
Column(u'\u6e2c\u8a66_id', Integer, primary_key=True,
|
|
autoincrement=False),
|
|
Column(u'unitable1_\u6e2c\u8a66', Integer,
|
|
ForeignKey(u'unitable1.\u6e2c\u8a66')
|
|
),
|
|
Column(u'Unitéble2_b', Integer,
|
|
ForeignKey(u'Unitéble2.b')
|
|
),
|
|
Column(u'\u6e2c\u8a66_self', Integer,
|
|
ForeignKey(u'\u6e2c\u8a66.\u6e2c\u8a66_id')
|
|
),
|
|
test_needs_fk=True,
|
|
)
|
|
else:
|
|
t3 = Table(u'\u6e2c\u8a66', metadata,
|
|
Column(u'\u6e2c\u8a66_id', Integer, primary_key=True,
|
|
autoincrement=False),
|
|
Column(u'unitable1_\u6e2c\u8a66', Integer),
|
|
Column(u'Unitéble2_b', Integer),
|
|
Column(u'\u6e2c\u8a66_self', Integer),
|
|
test_needs_fk=True,
|
|
)
|
|
metadata.create_all()
|
|
|
|
@engines.close_first
|
|
def teardown(self):
|
|
if metadata.tables:
|
|
t3.delete().execute()
|
|
t2.delete().execute()
|
|
t1.delete().execute()
|
|
|
|
@classmethod
|
|
def teardown_class(cls):
|
|
global unicode_bind
|
|
metadata.drop_all()
|
|
del unicode_bind
|
|
|
|
def test_insert(self):
|
|
t1.insert().execute({u'méil':1, u'\u6e2c\u8a66':5})
|
|
t2.insert().execute({'a':1, 'b':1})
|
|
t3.insert().execute({u'\u6e2c\u8a66_id': 1,
|
|
u'unitable1_\u6e2c\u8a66': 5,
|
|
u'Unitéble2_b': 1,
|
|
u'\u6e2c\u8a66_self': 1})
|
|
|
|
assert t1.select().execute().fetchall() == [(1, 5)]
|
|
assert t2.select().execute().fetchall() == [(1, 1)]
|
|
assert t3.select().execute().fetchall() == [(1, 5, 1, 1)]
|
|
|
|
def test_reflect(self):
|
|
t1.insert().execute({u'méil':2, u'\u6e2c\u8a66':7})
|
|
t2.insert().execute({'a':2, 'b':2})
|
|
t3.insert().execute({u'\u6e2c\u8a66_id': 2,
|
|
u'unitable1_\u6e2c\u8a66': 7,
|
|
u'Unitéble2_b': 2,
|
|
u'\u6e2c\u8a66_self': 2})
|
|
|
|
meta = MetaData(unicode_bind)
|
|
tt1 = Table(t1.name, meta, autoload=True)
|
|
tt2 = Table(t2.name, meta, autoload=True)
|
|
tt3 = Table(t3.name, meta, autoload=True)
|
|
|
|
tt1.insert().execute({u'méil':1, u'\u6e2c\u8a66':5})
|
|
tt2.insert().execute({u'méil':1, u'\u6e2c\u8a66':1})
|
|
tt3.insert().execute({u'\u6e2c\u8a66_id': 1,
|
|
u'unitable1_\u6e2c\u8a66': 5,
|
|
u'Unitéble2_b': 1,
|
|
u'\u6e2c\u8a66_self': 1})
|
|
|
|
self.assert_(tt1.select(order_by=desc(u'méil')).execute().fetchall() ==
|
|
[(2, 7), (1, 5)])
|
|
self.assert_(tt2.select(order_by=desc(u'méil')).execute().fetchall() ==
|
|
[(2, 2), (1, 1)])
|
|
self.assert_(tt3.select(order_by=desc(u'\u6e2c\u8a66_id')).
|
|
execute().fetchall() ==
|
|
[(2, 7, 2, 2), (1, 5, 1, 1)])
|
|
meta.drop_all()
|
|
metadata.create_all()
|
|
|
|
class EscapesDefaultsTest(fixtures.TestBase):
|
|
def test_default_exec(self):
|
|
metadata = MetaData(testing.db)
|
|
t1 = Table('t1', metadata,
|
|
Column(u'special_col', Integer, Sequence('special_col'), primary_key=True),
|
|
Column('data', String(50)) # to appease SQLite without DEFAULT VALUES
|
|
)
|
|
metadata.create_all()
|
|
|
|
try:
|
|
engine = metadata.bind
|
|
|
|
# reset the identifier preparer, so that we can force it to cache
|
|
# a unicode identifier
|
|
engine.dialect.identifier_preparer = engine.dialect.preparer(engine.dialect)
|
|
select([column(u'special_col')]).select_from(t1).execute().close()
|
|
assert isinstance(engine.dialect.identifier_preparer.format_sequence(Sequence('special_col')), unicode)
|
|
|
|
# now execute, run the sequence. it should run in u"Special_col.nextid" or similar as
|
|
# a unicode object; cx_oracle asserts that this is None or a String (postgresql lets it pass thru).
|
|
# ensure that executioncontext._exec_default() is encoding.
|
|
t1.insert().execute(data='foo')
|
|
finally:
|
|
metadata.drop_all()
|
|
|
|
|