import pytest

from apps.blog.models import ArticleIndexPage, ArticlePage
from apps.blog.tests.factories import AuthorFactory


@pytest.mark.django_db
def test_homepage_render(client, home_page):
    resp = client.get("/")
    assert resp.status_code == 200


@pytest.mark.django_db
def test_article_index_pagination_and_tag_filter(client, home_page):
    index = ArticleIndexPage(title="Articles", slug="articles")
    home_page.add_child(instance=index)
    author = AuthorFactory()
    for n in range(14):
        article = ArticlePage(
            title=f"A{n}",
            slug=f"a{n}",
            author=author,
            summary="summary",
            body=[("rich_text", "<p>body</p>")],
        )
        index.add_child(instance=article)
        article.save_revision().publish()

    resp = client.get("/articles/?page=2")
    assert resp.status_code == 200
    assert resp.context["articles"].number == 2


@pytest.mark.django_db
def test_article_page_related_context(client, home_page):
    index = ArticleIndexPage(title="Articles", slug="articles")
    home_page.add_child(instance=index)
    author = AuthorFactory()
    main = ArticlePage(
        title="Main",
        slug="main",
        author=author,
        summary="summary",
        body=[("rich_text", "<p>body</p>")],
    )
    index.add_child(instance=main)
    main.save_revision().publish()

    related = ArticlePage(
        title="Related",
        slug="related",
        author=author,
        summary="summary",
        body=[("rich_text", "<p>body</p>")],
    )
    index.add_child(instance=related)
    related.save_revision().publish()

    resp = client.get("/articles/main/")
    assert resp.status_code == 200
    assert "related_articles" in resp.context