From d7b8b475f8a5c0ddf955157f89db3d44d0dc0d9a Mon Sep 17 00:00:00 2001 From: Mark Sandan Date: Mon, 23 May 2016 17:08:36 -0700 Subject: Add DDLCompiler.create_table_suffix() Allows custom dialects to add keywords after the CREATE TABLE section. Change-Id: I6fa66dfcf00ef95122f491a9115410df2746cf88 --- test/sql/test_compiler.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'test/sql') diff --git a/test/sql/test_compiler.py b/test/sql/test_compiler.py index ca3468710..8b4e5053b 100644 --- a/test/sql/test_compiler.py +++ b/test/sql/test_compiler.py @@ -3081,6 +3081,22 @@ class DDLTest(fixtures.TestBase, AssertsCompiledSQL): "PRIMARY KEY (b, a))" ) + def test_create_table_suffix(self): + class MyDialect(default.DefaultDialect): + class MyCompiler(compiler.DDLCompiler): + def create_table_suffix(self, table): + return 'SOME SUFFIX' + + ddl_compiler = MyCompiler + + m = MetaData() + t1 = Table('t1', m, Column('q', Integer)) + self.assert_compile( + schema.CreateTable(t1), + "CREATE TABLE t1 SOME SUFFIX (q INTEGER)", + dialect=MyDialect() + ) + def test_table_no_cols(self): m = MetaData() t1 = Table('t1', m) -- cgit v1.2.1