42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from articles.models import Article, User
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_publish_article(unpublished_article: Article):
|
||
|
|
assert unpublished_article.status == Article.DRAFT
|
||
|
|
assert unpublished_article.published_at is None
|
||
|
|
published_article = unpublished_article.publish()
|
||
|
|
assert published_article.status == Article.PUBLISHED
|
||
|
|
assert published_article.published_at is not None
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_unpublish_article(published_article: Article):
|
||
|
|
assert published_article.status == Article.PUBLISHED
|
||
|
|
assert published_article.published_at is not None
|
||
|
|
unpublished_article = published_article.unpublish()
|
||
|
|
assert unpublished_article.status == Article.DRAFT
|
||
|
|
assert unpublished_article.published_at is None
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_save_article_adds_missing_slug(author: User):
|
||
|
|
# Explicitly calling bulk_create with one article because it doesn't call save().
|
||
|
|
articles = Article.objects.bulk_create(
|
||
|
|
[Article(author=author, title="noice title", slug="", status=Article.DRAFT)]
|
||
|
|
)
|
||
|
|
article = articles[0]
|
||
|
|
assert article.slug == ""
|
||
|
|
article.save()
|
||
|
|
assert article.slug != ""
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_save_article_doesnt_change_existing_slug(published_article: Article):
|
||
|
|
original_slug = published_article.slug
|
||
|
|
published_article.title = "This is a brand new title"
|
||
|
|
published_article.save()
|
||
|
|
assert published_article.slug == original_slug
|