#485 add possibility to pass comments to sqlalchemy

tests/test_model_definition/test_models_are_pickable.py

@@ -0,0 +1,60 @@
+import pickle
+from typing import Optional
+
+import databases
+import pytest
+import sqlalchemy
+
+import ormar
+from tests.settings import DATABASE_URL
+
+database = databases.Database(DATABASE_URL, force_rollback=True)
+metadata = sqlalchemy.MetaData()
+
+
+class User(ormar.Model):
+    class Meta:
+        tablename = "users"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100)
+    properties = ormar.JSON(nullable=True)
+
+
+class Post(ormar.Model):
+    class Meta:
+        tablename = "posts"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100)
+    created_by: Optional[User] = ormar.ForeignKey(User)
+
+
+@pytest.fixture(autouse=True, scope="module")
+def create_test_database():
+    engine = sqlalchemy.create_engine(DATABASE_URL)
+    metadata.drop_all(engine)
+    metadata.create_all(engine)
+    yield
+    metadata.drop_all(engine)
+
+
+@pytest.mark.asyncio
+async def test_dumping_and_loading_model_works():
+    user = await User(name="Test", properties={"aa": "bb"}).save()
+    post = Post(name="Test post")
+    await user.posts.add(post)
+    pickled_value = pickle.dumps(user)
+    python_value = pickle.loads(pickled_value)
+    assert isinstance(python_value, User)
+    assert python_value.name == "Test"
+    assert python_value.properties == {"aa": "bb"}
+    assert python_value.posts[0].name == "Test post"
+    await python_value.load()
+    await python_value.update(name="Test2")
+    check = await User.objects.get()
+    assert check.name == "Test2"

tests/test_model_definition/test_setting_comments_in_db.py

@@ -0,0 +1,36 @@
+import databases
+import pytest
+import sqlalchemy
+
+import ormar
+from ormar.models import Model
+from tests.settings import DATABASE_URL
+
+metadata = sqlalchemy.MetaData()
+
+database = databases.Database(DATABASE_URL, force_rollback=True)
+
+
+class Comment(Model):
+    class Meta(ormar.ModelMeta):
+        tablename = "comments"
+        metadata = metadata
+        database = database
+
+    test: int = ormar.Integer(primary_key=True, comment="primary key of comments")
+    test_string: str = ormar.String(max_length=250, comment="test that it works")
+
+
+@pytest.fixture(autouse=True, scope="module")
+def create_test_database():
+    engine = sqlalchemy.create_engine(DATABASE_URL)
+    metadata.create_all(engine)
+    yield
+    metadata.drop_all(engine)
+
+
+@pytest.mark.asyncio
+async def test_comments_are_set_in_db():
+    columns = Comment.Meta.table.c
+    for c in columns:
+        assert c.comment == Comment.Meta.model_fields[c.name].comment