python-blog/articles/tests/test_articles.py

import pytest
from django.test import Client
from django.urls import reverse
from model_bakery import baker

from articles.models import Article, Page, User


@pytest.mark.django_db
def test_can_access_list(
    client: Client, published_article: Article, published_page: Page
):
    res = client.get(reverse("articles-list"))
    assert res.status_code == 200
    content = res.content.decode("utf-8")
    for art in [published_article, published_page]:
        assert art.title in content
    assert published_article.get_abstract() in content
    assert published_page.get_formatted_content() not in content


@pytest.mark.django_db
def test_abstract_shown_on_list(client: Client, author: User):
    abstract = "Some abstract"
    after = "Some content after abstract"
    baker.make(
        Article,
        status=Article.PUBLISHED,
        author=author,
        content=f"{abstract}\n<!--more-->\n{after}",
    )  # type: Article
    res = client.get(reverse("articles-list"))
    content = res.content.decode("utf-8")
    assert abstract in content
    assert after not in content