Implement more complete support for comments

Author: gordthompson

sqlalchemy_cockroachdb/base.py

@@ -569,6 +569,10 @@ class CockroachDBImpl(alembic.ddl.postgresql.PostgresqlImpl):
     def visit_column_type(*args, **kwargs):
         return alembic.ddl.postgresql.visit_column_type(*args, **kwargs)
 
+    @compiles(alembic.ddl.postgresql.ColumnComment, "cockroachdb")
+    def visit_column_comment(*args, **kwargs):
+        return alembic.ddl.postgresql.visit_column_comment(*args, **kwargs)
+
 
 # If sqlalchemy-migrate is installed, register there too.
 try:

test/test_suite_alembic.py

@@ -37,13 +37,15 @@ def test_existing_table_comment_no_change(self):
     def test_remove_column_comment(self):
         pass
 
+    @skip("cockroachdb")
+    def test_remove_table_comment(self):
+        pass
+
 
 class AutogenerateComputedTest(_AutogenerateComputedTest):
+    @skip("cockroachdb")
     def test_add_computed_column(self):
-        if not (
-            config.db.dialect.driver == "asyncpg" and not config.db.dialect._is_v231plus
-        ):
-            super().test_add_computed_column()
+        pass
 
     @skip("cockroachdb")
     def test_cant_change_computed_warning(self):