pass
class SQLiteIdentifierPreparer(compiler.IdentifierPreparer):
- reserved_words = compiler.IdentifierPreparer.reserved_words.union([
- 'abort', 'add', 'after', 'alter', 'attach', 'autoincrement', 'before',
- 'begin','by', 'cascade', 'commit', 'conflict', 'database', 'deferred',
- 'delete', 'detach', 'drop', 'each', 'escape', 'exclusive', 'explain',
- 'fail', 'glob', 'if', 'ignore', 'immediate', 'index', 'insert',
- 'instead', 'key', 'match', 'of', 'plan', 'pragma', 'query', 'raise',
- 'reindex', 'rename', 'replace', 'restrict', 'rollback', 'row', 'temp',
- 'temporary', 'transaction', 'trigger', 'update', 'vacuum', 'values',
- 'view', 'virtual'])
+ reserved_words = util.Set([
+ 'add', 'after', 'all', 'alter', 'analyze', 'and', 'as', 'asc',
+ 'attach', 'autoincrement', 'before', 'begin', 'between', 'by',
+ 'cascade', 'case', 'cast', 'check', 'collate', 'column', 'commit',
+ 'conflict', 'constraint', 'create', 'cross', 'current_date',
+ 'current_time', 'current_timestamp', 'database', 'default',
+ 'deferrable', 'deferred', 'delete', 'desc', 'detach', 'distinct',
+ 'drop', 'each', 'else', 'end', 'escape', 'except', 'exclusive',
+ 'explain', 'false', 'fail', 'for', 'foreign', 'from', 'full', 'glob',
+ 'group', 'having', 'if', 'ignore', 'immediate', 'in', 'index',
+ 'initially', 'inner', 'insert', 'instead', 'intersect', 'into', 'is',
+ 'isnull', 'join', 'key', 'left', 'like', 'limit', 'match', 'natural',
+ 'not', 'notnull', 'null', 'of', 'offset', 'on', 'or', 'order', 'outer',
+ 'plan', 'pragma', 'primary', 'query', 'raise', 'references',
+ 'reindex', 'rename', 'replace', 'restrict', 'right', 'rollback',
+ 'row', 'select', 'set', 'table', 'temp', 'temporary', 'then', 'to',
+ 'transaction', 'trigger', 'true', 'union', 'unique', 'update', 'using',
+ 'vacuum', 'values', 'view', 'virtual', 'when', 'where',
+ ])
+
def __init__(self, dialect):
super(SQLiteIdentifierPreparer, self).__init__(dialect, omit_schema=True)
def suite():
modules_to_test = (
- 'dialect.mysql',
- 'dialect.postgres',
- 'dialect.oracle',
- 'dialect.mssql',
'dialect.access',
- 'dialect.informix',
'dialect.firebird',
+ 'dialect.informix',
+ 'dialect.mssql',
+ 'dialect.mysql',
+ 'dialect.oracle',
+ 'dialect.postgres',
+ 'dialect.sqlite',
)
alltests = unittest.TestSuite()
for name in modules_to_test:
--- /dev/null
+"""SQLite-specific tests."""
+
+import testbase
+import datetime
+from sqlalchemy import *
+from sqlalchemy.databases import sqlite
+from testlib import *
+
+
+class TestTypes(AssertMixin):
+ @testing.supported('sqlite')
+ def test_date(self):
+ meta = MetaData(testbase.db)
+ t = Table('testdate', meta,
+ Column('id', Integer, primary_key=True),
+ Column('adate', Date),
+ Column('adatetime', DateTime))
+ meta.create_all()
+ try:
+ d1 = datetime.date(2007, 10, 30)
+ d2 = datetime.datetime(2007, 10, 30)
+
+ t.insert().execute(adate=str(d1), adatetime=str(d2))
+
+ self.assert_(t.select().execute().fetchall()[0] ==
+ (1, datetime.date(2007, 10, 30),
+ datetime.datetime(2007, 10, 30)))
+
+ finally:
+ meta.drop_all()
+
+
+class DialectTest(AssertMixin):
+ @testing.supported('sqlite')
+ def test_extra_reserved_words(self):
+ """Tests reserved words in identifiers.
+
+ 'true', 'false', and 'column' are undocumented reserved words
+ when used as column identifiers (as of 3.5.1). Covering them here
+ to ensure they remain in place if the dialect's reserved_words set
+ is updated in the future.
+ """
+
+ meta = MetaData(testbase.db)
+ t = Table('reserved', meta,
+ Column('safe', Integer),
+ Column('true', Integer),
+ Column('false', Integer),
+ Column('column', Integer))
+
+ try:
+ meta.create_all()
+ t.insert().execute(safe=1)
+ list(t.select().execute())
+ finally:
+ meta.drop_all()
+
+ @testing.supported('sqlite')
+ def test_quoted_identifiers(self):
+ """Tests autoload of tables created with quoted column names."""
+
+ # This is quirky in sqlite.
+ testbase.db.execute("""CREATE TABLE "django_content_type" (
+ "id" integer NOT NULL PRIMARY KEY,
+ "django_stuff" text NULL
+ )
+ """)
+ testbase.db.execute("""
+ CREATE TABLE "django_admin_log" (
+ "id" integer NOT NULL PRIMARY KEY,
+ "action_time" datetime NOT NULL,
+ "content_type_id" integer NULL REFERENCES "django_content_type" ("id"),
+ "object_id" text NULL,
+ "change_message" text NOT NULL
+ )
+ """)
+ try:
+ meta = MetaData(testbase.db)
+ table1 = Table("django_admin_log", meta, autoload=True)
+ table2 = Table("django_content_type", meta, autoload=True)
+ j = table1.join(table2)
+ assert j.onclause == table1.c.content_type_id==table2.c.id
+ finally:
+ testbase.db.execute("drop table django_admin_log")
+ testbase.db.execute("drop table django_content_type")
+
+
+if __name__ == "__main__":
+ testbase.main()
finally:
testbase.db.execute("drop table book")
- @testing.supported('sqlite')
- def test_goofy_sqlite(self):
- """test autoload of table where quotes were used with all the colnames. quirky in sqlite."""
- testbase.db.execute("""CREATE TABLE "django_content_type" (
- "id" integer NOT NULL PRIMARY KEY,
- "django_stuff" text NULL
- )
- """)
- testbase.db.execute("""
- CREATE TABLE "django_admin_log" (
- "id" integer NOT NULL PRIMARY KEY,
- "action_time" datetime NOT NULL,
- "content_type_id" integer NULL REFERENCES "django_content_type" ("id"),
- "object_id" text NULL,
- "change_message" text NOT NULL
- )
- """)
- try:
- meta = MetaData(testbase.db)
- table1 = Table("django_admin_log", meta, autoload=True)
- table2 = Table("django_content_type", meta, autoload=True)
- j = table1.join(table2)
- assert j.onclause == table1.c.content_type_id==table2.c.id
- finally:
- testbase.db.execute("drop table django_admin_log")
- testbase.db.execute("drop table django_content_type")
-
@testing.exclude('mysql', '<', (4, 1, 1))
def test_composite_fk(self):
"""test reflection of composite foreign keys"""
l = map(list, users_with_date.select().execute().fetchall())
self.assert_(l == insert_data,
'DateTest mismatch: got:%s expected:%s' % (l, insert_data))
-
- @testing.supported('sqlite')
- def test_sqlite_date(self):
- meta = MetaData(testbase.db)
- t = Table('testdate', meta,
- Column('id', Integer, primary_key=True),
- Column('adate', Date),
- Column('adatetime', DateTime))
- t.create(checkfirst=True)
- try:
- d1 = datetime.date(2007, 10, 30)
- d2 = datetime.datetime(2007, 10, 30)
-
- t.insert().execute(adate=str(d1), adatetime=str(d2))
-
- self.assert_(t.select().execute().fetchall()[0] ==
- (1, datetime.date(2007, 10, 30),
- datetime.datetime(2007, 10, 30)))
-
- finally:
- t.drop(checkfirst=True)
def testtextdate(self):
x = testbase.db.text(
projects / thirdparty / sqlalchemy / sqlalchemy.git / commitdiff
