From 05768140563e1bd3457ceffd195b12841ec0906a Mon Sep 17 00:00:00 2001 From: CaselIT Date: Sat, 9 Jan 2021 11:57:46 +0100 Subject: [PATCH] Support default nullable on identity columns. This fixes the test for I0516d506ff327cff35cda605e8897a27440e0373 Change-Id: Ief4546fcb7d4d266d0f1f73ed53069dc5cb34fde --- tests/test_autogen_render.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/tests/test_autogen_render.py b/tests/test_autogen_render.py index d2dcc349..5d264935 100644 --- a/tests/test_autogen_render.py +++ b/tests/test_autogen_render.py @@ -2123,7 +2123,7 @@ class AutogenRenderTest(TestBase): % persisted, ) - @config.requirements.identity_columns + @config.requirements.identity_columns_api @testing.combinations( ({}, "sa.Identity(always=False)"), (dict(always=None), "sa.Identity(always=None)"), @@ -2148,16 +2148,15 @@ class AutogenRenderTest(TestBase): ), ) def test_render_add_column_identity(self, kw, text): - op_obj = ops.AddColumnOp( - "foo", Column("x", Integer, sa.Identity(**kw)) - ) + col = Column("x", Integer, sa.Identity(**kw)) + op_obj = ops.AddColumnOp("foo", col) eq_ignore_whitespace( autogenerate.render_op_text(self.autogen_context, op_obj), "op.add_column('foo', sa.Column('x', sa.Integer(), " - "%s, nullable=True))" % text, + "%s, nullable=%r))" % (text, col.nullable), ) - @config.requirements.identity_columns + @config.requirements.identity_columns_api @testing.combinations( ({}, "sa.Identity(always=False)"), (dict(always=None), "sa.Identity(always=None)"), @@ -2195,7 +2194,7 @@ class AutogenRenderTest(TestBase): "server_default=%s)" % text, ) - @config.requirements.identity_columns + @config.requirements.identity_columns_api def test_render_alter_column_drop_identity(self): op_obj = ops.AlterColumnOp( "foo", -- 2.47.2