?HAMA

Edit File Size: 6.14 KB

/opt/hc_python/lib64/python3.12/site-packages/alembic/testing/suite/test_autogen_comments.py

Text Base64

from sqlalchemy import Column
from sqlalchemy import Float
from sqlalchemy import MetaData
from sqlalchemy import String
from sqlalchemy import Table

from ._autogen_fixtures import AutogenFixtureTest
from ...testing import eq_
from ...testing import mock
from ...testing import TestBase

class AutogenerateCommentsTest(AutogenFixtureTest, TestBase):
    __backend__ = True

__requires__ = ("comments",)

def test_existing_table_comment_no_change(self):
        m1 = MetaData()
        m2 = MetaData()

Table(
            "some_table",
            m1,
            Column("test", String(10), primary_key=True),
            comment="this is some table",
        )

Table(
            "some_table",
            m2,
            Column("test", String(10), primary_key=True),
            comment="this is some table",
        )

diffs = self._fixture(m1, m2)

eq_(diffs, [])

def test_add_table_comment(self):
        m1 = MetaData()
        m2 = MetaData()

Table("some_table", m1, Column("test", String(10), primary_key=True))

Table(
            "some_table",
            m2,
            Column("test", String(10), primary_key=True),
            comment="this is some table",
        )

diffs = self._fixture(m1, m2)

eq_(diffs[0][0], "add_table_comment")
        eq_(diffs[0][1].comment, "this is some table")
        eq_(diffs[0][2], None)

def test_remove_table_comment(self):
        m1 = MetaData()
        m2 = MetaData()

Table(
            "some_table",
            m1,
            Column("test", String(10), primary_key=True),
            comment="this is some table",
        )

Table("some_table", m2, Column("test", String(10), primary_key=True))

diffs = self._fixture(m1, m2)

eq_(diffs[0][0], "remove_table_comment")
        eq_(diffs[0][1].comment, None)

def test_alter_table_comment(self):
        m1 = MetaData()
        m2 = MetaData()

Table(
            "some_table",
            m1,
            Column("test", String(10), primary_key=True),
            comment="this is some table",
        )

Table(
            "some_table",
            m2,
            Column("test", String(10), primary_key=True),
            comment="this is also some table",
        )

diffs = self._fixture(m1, m2)

eq_(diffs[0][0], "add_table_comment")
        eq_(diffs[0][1].comment, "this is also some table")
        eq_(diffs[0][2], "this is some table")

def test_existing_column_comment_no_change(self):
        m1 = MetaData()
        m2 = MetaData()

Table(
            "some_table",
            m1,
            Column("test", String(10), primary_key=True),
            Column("amount", Float, comment="the amount"),
        )

Table(
            "some_table",
            m2,
            Column("test", String(10), primary_key=True),
            Column("amount", Float, comment="the amount"),
        )

diffs = self._fixture(m1, m2)

eq_(diffs, [])

def test_add_column_comment(self):
        m1 = MetaData()
        m2 = MetaData()

Table(
            "some_table",
            m1,
            Column("test", String(10), primary_key=True),
            Column("amount", Float),
        )

Table(
            "some_table",
            m2,
            Column("test", String(10), primary_key=True),
            Column("amount", Float, comment="the amount"),
        )

diffs = self._fixture(m1, m2)
        eq_(
            diffs,
            [
                [
                    (
                        "modify_comment",
                        None,
                        "some_table",
                        "amount",
                        {
                            "existing_nullable": True,
                            "existing_type": mock.ANY,
                            "existing_server_default": False,
                        },
                        None,
                        "the amount",
                    )
                ]
            ],
        )

def test_remove_column_comment(self):
        m1 = MetaData()
        m2 = MetaData()

Table(
            "some_table",
            m1,
            Column("test", String(10), primary_key=True),
            Column("amount", Float, comment="the amount"),
        )

Table(
            "some_table",
            m2,
            Column("test", String(10), primary_key=True),
            Column("amount", Float),
        )

diffs = self._fixture(m1, m2)
        eq_(
            diffs,
            [
                [
                    (
                        "modify_comment",
                        None,
                        "some_table",
                        "amount",
                        {
                            "existing_nullable": True,
                            "existing_type": mock.ANY,
                            "existing_server_default": False,
                        },
                        "the amount",
                        None,
                    )
                ]
            ],
        )

def test_alter_column_comment(self):
        m1 = MetaData()
        m2 = MetaData()

Table(
            "some_table",
            m1,
            Column("test", String(10), primary_key=True),
            Column("amount", Float, comment="the amount"),
        )

Table(
            "some_table",
            m2,
            Column("test", String(10), primary_key=True),
            Column("amount", Float, comment="the adjusted amount"),
        )

diffs = self._fixture(m1, m2)

eq_(
            diffs,
            [
                [
                    (
                        "modify_comment",
                        None,
                        "some_table",
                        "amount",
                        {
                            "existing_nullable": True,
                            "existing_type": mock.ANY,
                            "existing_server_default": False,
                        },
                        "the amount",
                        "the adjusted amount",
                    )
                ]
            ],
        )

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2025-04-04 08:02:08	Edit Download recursive recursive
test_autogen_comments.py	6.14 KB	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive
test_autogen_computed.py	4.03 KB	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive
test_autogen_diffs.py	8.20 KB	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive
test_autogen_fks.py	32.16 KB	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive
test_autogen_identity.py	5.69 KB	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive
test_environment.py	11.60 KB	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive
test_op.py	1.31 KB	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive
_autogen_fixtures.py	9.65 KB	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive
__init__.py	288 B	lrw-r--r--	2025-04-04 08:02:08	Edit Download recursive recursive

Directory Contents