def test_custom_type_compare(self):
class MyType(TypeDecorator):
impl = Integer
def compare_against_backend(self, dialect, conn_type):
return isinstance(conn_type, Integer)
ac = ops.AlterColumnOp("sometable", "somecol")
autogenerate.compare._compare_type(
self.autogen_context,
ac,
None,
"sometable",
"somecol",
Column("somecol", INTEGER()),
Column("somecol", MyType()),
)
assert not ac.has_changes()
ac = ops.AlterColumnOp("sometable", "somecol")
autogenerate.compare._compare_type(
self.autogen_context,
ac,
None,
"sometable",
"somecol",
Column("somecol", String()),
Column("somecol", MyType()),
)
diff = ac.to_diff_tuple()
eq_(diff[0][0:4], ('modify_type', None, 'sometable', 'somecol'))
def test_no_needless_pass(self):
writer1 = autogenerate.Rewriter()
@writer1.rewrites(ops.AlterColumnOp)
def rewrite_alter_column(context, revision, op):
return []
directives = [
ops.MigrationScript(
util.rev_id(),
ops.UpgradeOps(
ops=[
ops.ModifyTableOps(
"t1",
ops=[
ops.AlterColumnOp(
"foo",
"bar",
modify_nullable=False,
existing_type=sa.Integer(),
),
ops.AlterColumnOp(
"foo",
"bar",
modify_nullable=False,
existing_type=sa.Integer(),
),
],
),
ops.ModifyTableOps(
"t1",
ops=[
ops.AlterColumnOp(
"foo",
"bar",
modify_nullable=False,
existing_type=sa.Integer(),
)
],
),
]
),
ops.DowngradeOps(ops=[]),
)
]
ctx, rev = mock.Mock(), mock.Mock()
writer1(ctx, rev, directives)
eq_(
autogenerate.render_python_code(directives[0].upgrade_ops),
"# ### commands auto generated by Alembic - please adjust! ###\n"
" pass\n"
" # ### end Alembic commands ###",
)
def _compare_default_roundtrip(self,
type_,
orig_default,
alternate=None,
diff_expected=None):
diff_expected = diff_expected \
if diff_expected is not None \
else alternate is not None
if alternate is None:
alternate = orig_default
t1 = Table("test", self.metadata,
Column("somecol", type_, server_default=orig_default))
t2 = Table("test", MetaData(),
Column("somecol", type_, server_default=alternate))
t1.create(self.bind)
insp = Inspector.from_engine(self.bind)
cols = insp.get_columns(t1.name)
insp_col = Column("somecol",
cols[0]['type'],
server_default=text(cols[0]['default']))
op = ops.AlterColumnOp("test", "somecol")
_compare_server_default(self.autogen_context, op, None, "test",
"somecol", insp_col, t2.c.somecol)
diffs = op.to_diff_tuple()
eq_(bool(diffs), diff_expected)
def test_alter_column_modify_programmatic_default(self):
# test issue #736
# when autogenerate.compare creates the operation object
# programmatically, the server_default of the op has the full
# DefaultClause present. make sure the usual renderer works.
context = op_fixture("mysql")
m1 = MetaData()
autogen_context = api.AutogenContext(context, m1)
operation = ops.AlterColumnOp("t", "c")
for fn in (
compare._compare_nullable,
compare._compare_type,
compare._compare_server_default,
):
fn(
autogen_context,
operation,
None,
"t",
"c",
Column("c", Float(), nullable=False, server_default=text("0")),
Column("c", Float(), nullable=True, default=0),
)
op.invoke(operation)
context.assert_("ALTER TABLE t MODIFY c FLOAT NULL DEFAULT 0")
def test_skip_null_type_comparison_local(self):
ac = ops.AlterColumnOp("sometable", "somecol")
autogenerate.compare._compare_type(
self.autogen_context, ac,
None, "sometable", "somecol",
Column("somecol", Integer()),
Column("somecol", NULLTYPE),
)
diff = ac.to_diff_tuple()
assert not diff
def add_column_nullable(context, revision, op):
if op.column.nullable:
return op
else:
op.column.nullable = True
return [
op,
ops.AlterColumnOp(
op.table_name,
op.column.name,
modify_nullable=False,
existing_type=op.column.type,
)
]
def test_affinity_typedec(self):
class MyType(TypeDecorator):
impl = CHAR
def load_dialect_impl(self, dialect):
if dialect.name == 'sqlite':
return dialect.type_descriptor(Integer())
else:
return dialect.type_descriptor(CHAR(32))
uo = ops.AlterColumnOp('sometable', 'somecol')
autogenerate.compare._compare_type(
self.autogen_context, uo, None, "sometable", "somecol",
Column("somecol", Integer, nullable=True),
Column("somecol", MyType()))
assert not uo.has_changes()