feat(post): add nullable description + attachment_count (migration 0007)
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,31 @@
|
|||||||
|
"""fc2d-iv: post.description + post.attachment_count
|
||||||
|
|
||||||
|
Revision ID: 0007
|
||||||
|
Revises: 0006
|
||||||
|
Create Date: 2026-05-18
|
||||||
|
|
||||||
|
"""
|
||||||
|
from typing import Sequence, Union
|
||||||
|
|
||||||
|
import sqlalchemy as sa
|
||||||
|
from alembic import op
|
||||||
|
|
||||||
|
revision: str = "0007"
|
||||||
|
down_revision: Union[str, None] = "0006"
|
||||||
|
branch_labels: Union[str, Sequence[str], None] = None
|
||||||
|
depends_on: Union[str, Sequence[str], None] = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade() -> None:
|
||||||
|
op.add_column(
|
||||||
|
"post", sa.Column("description", sa.Text(), nullable=True)
|
||||||
|
)
|
||||||
|
op.add_column(
|
||||||
|
"post",
|
||||||
|
sa.Column("attachment_count", sa.Integer(), nullable=True),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade() -> None:
|
||||||
|
op.drop_column("post", "attachment_count")
|
||||||
|
op.drop_column("post", "description")
|
||||||
@@ -28,6 +28,9 @@ class Post(Base):
|
|||||||
|
|
||||||
raw_metadata: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
raw_metadata: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
||||||
|
|
||||||
|
description: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||||
|
attachment_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
||||||
|
|
||||||
downloaded_at: Mapped[datetime] = mapped_column(
|
downloaded_at: Mapped[datetime] = mapped_column(
|
||||||
DateTime(timezone=True), nullable=False, server_default=func.now()
|
DateTime(timezone=True), nullable=False, server_default=func.now()
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -0,0 +1,48 @@
|
|||||||
|
"""FC-2d-iv: post.description + post.attachment_count round-trip."""
|
||||||
|
|
||||||
|
from datetime import UTC, datetime
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from backend.app.models import Artist, Post, Source
|
||||||
|
|
||||||
|
pytestmark = pytest.mark.integration
|
||||||
|
|
||||||
|
|
||||||
|
async def _post(db, **post_kwargs):
|
||||||
|
artist = Artist(name="Nadia", slug="nadia")
|
||||||
|
db.add(artist)
|
||||||
|
await db.flush()
|
||||||
|
src = Source(artist_id=artist.id, platform="web", url="http://x")
|
||||||
|
db.add(src)
|
||||||
|
await db.flush()
|
||||||
|
post = Post(
|
||||||
|
source_id=src.id, external_post_id="p1",
|
||||||
|
post_date=datetime(2026, 3, 1, tzinfo=UTC),
|
||||||
|
**post_kwargs,
|
||||||
|
)
|
||||||
|
db.add(post)
|
||||||
|
await db.flush()
|
||||||
|
return post.id
|
||||||
|
|
||||||
|
|
||||||
|
def test_post_has_new_columns():
|
||||||
|
cols = {c.name for c in Post.__table__.columns}
|
||||||
|
assert "description" in cols
|
||||||
|
assert "attachment_count" in cols
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_description_and_attachment_count_round_trip(db):
|
||||||
|
pid = await _post(db, description="<p>hi</p>", attachment_count=3)
|
||||||
|
row = await db.get(Post, pid)
|
||||||
|
assert row.description == "<p>hi</p>"
|
||||||
|
assert row.attachment_count == 3
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_new_fields_default_null(db):
|
||||||
|
pid = await _post(db)
|
||||||
|
row = await db.get(Post, pid)
|
||||||
|
assert row.description is None
|
||||||
|
assert row.attachment_count is None
|
||||||
Reference in New Issue
Block a user