From 4917e195c184b79d63ea60a31ae8ea59e499162c Mon Sep 17 00:00:00 2001 From: rmoretto Date: Sat, 5 Sep 2020 11:16:26 -0300 Subject: [PATCH] added tests to the Table.prefixes parameter render --- tests/test_autogen_render.py | 37 ++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) diff --git a/tests/test_autogen_render.py b/tests/test_autogen_render.py index d86e934d..7582f9e8 100644 --- a/tests/test_autogen_render.py +++ b/tests/test_autogen_render.py @@ -966,6 +966,43 @@ class AutogenRenderTest(TestBase): ")", ) + def test_render_table_w_prefixes(self): + m = MetaData() + t = Table( + "test", + m, + Column("id", Integer, primary_key=True), + prefixes=["TEST", "PREFIXES"], + ) + op_obj = ops.CreateTableOp.from_table(t) + eq_ignore_whitespace( + autogenerate.render_op_text(self.autogen_context, op_obj), + "op.create_table('test'," + "sa.Column('id', sa.Integer(), nullable=False)," + "sa.PrimaryKeyConstraint('id')," + "prefixes=['TEST', 'PREFIXES']" + ")", + ) + + def test_render_table_w_prefixes_schema(self): + m = MetaData(schema="foo") + t = Table( + "test", + m, + Column("id", Integer, primary_key=True), + prefixes=["TEST", "PREFIXES"], + ) + op_obj = ops.CreateTableOp.from_table(t) + eq_ignore_whitespace( + autogenerate.render_op_text(self.autogen_context, op_obj), + "op.create_table('test'," + "sa.Column('id', sa.Integer(), nullable=False)," + "sa.PrimaryKeyConstraint('id')," + "schema='foo'," + "prefixes=['TEST', 'PREFIXES']" + ")", + ) + def test_render_addtl_args(self): m = MetaData() t = Table( -- 2.47.2