mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-15 21:27:23 -04:00
Mike Bayer
84 lines
3.4 KiB
Python
84 lines
3.4 KiB
Python
from testbase import PersistTest
|
|
import testbase
|
|
from sqlalchemy import *
|
|
|
|
class QuoteTest(PersistTest):
|
|
def setUpAll(self):
|
|
# TODO: figure out which databases/which identifiers allow special characters to be used,
|
|
# such as: spaces, quote characters, punctuation characters, set up tests for those as
|
|
# well.
|
|
global table1, table2, table3
|
|
metadata = BoundMetaData(testbase.db)
|
|
table1 = Table('WorstCase1', metadata,
|
|
Column('lowercase', Integer, primary_key=True),
|
|
Column('UPPERCASE', Integer),
|
|
Column('MixedCase', Integer),
|
|
Column('ASC', Integer, key='a123'))
|
|
table2 = Table('WorstCase2', metadata,
|
|
Column('desc', Integer, primary_key=True, key='d123'),
|
|
Column('Union', Integer, key='u123'),
|
|
Column('MixedCase', Integer))
|
|
table1.create()
|
|
table2.create()
|
|
|
|
def tearDown(self):
|
|
table1.delete().execute()
|
|
table2.delete().execute()
|
|
|
|
def tearDownAll(self):
|
|
table1.drop()
|
|
table2.drop()
|
|
|
|
def testbasic(self):
|
|
table1.insert().execute({'lowercase':1,'UPPERCASE':2,'MixedCase':3,'a123':4},
|
|
{'lowercase':2,'UPPERCASE':2,'MixedCase':3,'a123':4},
|
|
{'lowercase':4,'UPPERCASE':3,'MixedCase':2,'a123':1})
|
|
table2.insert().execute({'d123':1,'u123':2,'MixedCase':3},
|
|
{'d123':2,'u123':2,'MixedCase':3},
|
|
{'d123':4,'u123':3,'MixedCase':2})
|
|
|
|
res1 = select([table1.c.lowercase, table1.c.UPPERCASE, table1.c.MixedCase, table1.c.a123]).execute().fetchall()
|
|
print res1
|
|
assert(res1==[(1,2,3,4),(2,2,3,4),(4,3,2,1)])
|
|
|
|
res2 = select([table2.c.d123, table2.c.u123, table2.c.MixedCase]).execute().fetchall()
|
|
print res2
|
|
assert(res2==[(1,2,3),(2,2,3),(4,3,2)])
|
|
|
|
def testreflect(self):
|
|
meta2 = BoundMetaData(testbase.db)
|
|
t2 = Table('WorstCase2', meta2, autoload=True, quote=True)
|
|
assert t2.c.has_key('MixedCase')
|
|
|
|
def testlabels(self):
|
|
table1.insert().execute({'lowercase':1,'UPPERCASE':2,'MixedCase':3,'a123':4},
|
|
{'lowercase':2,'UPPERCASE':2,'MixedCase':3,'a123':4},
|
|
{'lowercase':4,'UPPERCASE':3,'MixedCase':2,'a123':1})
|
|
table2.insert().execute({'d123':1,'u123':2,'MixedCase':3},
|
|
{'d123':2,'u123':2,'MixedCase':3},
|
|
{'d123':4,'u123':3,'MixedCase':2})
|
|
|
|
res1 = select([table1.c.lowercase, table1.c.UPPERCASE, table1.c.MixedCase, table1.c.a123], use_labels=True).execute().fetchall()
|
|
print res1
|
|
assert(res1==[(1,2,3,4),(2,2,3,4),(4,3,2,1)])
|
|
|
|
res2 = select([table2.c.d123, table2.c.u123, table2.c.MixedCase], use_labels=True).execute().fetchall()
|
|
print res2
|
|
assert(res2==[(1,2,3),(2,2,3),(4,3,2)])
|
|
|
|
def testcascade(self):
|
|
lcmetadata = MetaData(case_sensitive=False)
|
|
t1 = Table('SomeTable', lcmetadata,
|
|
Column('UcCol', Integer),
|
|
Column('normalcol', String))
|
|
t2 = Table('othertable', lcmetadata,
|
|
Column('UcCol', Integer),
|
|
Column('normalcol', String, ForeignKey('SomeTable.normalcol')))
|
|
assert lcmetadata.case_sensitive is False
|
|
assert t1.c.UcCol.case_sensitive is False
|
|
assert t2.c.normalcol.case_sensitive is False
|
|
|
|
|
|
if __name__ == "__main__":
|
|
testbase.main()
|