ユニークな値が登録できないことを確認するテストケースの書き方

Question

ユニークな値を持つmodelを作成し、そのテストケースを書きたいと考えています。

// models.py
class Post(models.Model):
    title = models.CharField(max_length=100)
    slug = models.SlugField(blank=True, null=True, unique=True)

// tests.py
class TestPost(TestCase):
    def test_post(self):
        Post.objects.create(title="投稿1", slug="hoge")
        posts_count = Post.objects.all().count()
        self.assertEqual(posts_count, 1)
    // テスト通る

    def test_post_duplicate(self):
        Post.objects.create(title="投稿1", slug="hoge")
        Post.objects.create(title="投稿2", slug="hoge")
        posts_count = Post.objects.all().count()
        self.assertEqual(posts_count, 1)
　　// テスト通らない

やりたいこと

ユニークな値を持つオブジェクトで、わざと重複した値を登録し失敗するテストケースを書きたいと思うのですが、正直書き方がよく分かっていません。

上記のような場合、どのようなテストケースを書くと良いのでしょうか。

管理画面から登録できないことは手動で確認済みですが、テストケースを書きたいと思い質問しました。よろしくお願いします。

Answer 1

assertRaisesで例外をキャッチすると、別のエラー(TransactionManagementError)が発生する

// tests.py
from django.db.utils import IntegrityError

class TestPost(TestCase):
    def test_post(self):
        Post.objects.create(title="投稿1", slug="hoge")
        posts_count = Post.objects.all().count()
        self.assertEqual(posts_count, 1)
    // テスト通る

    def test_post_duplicate(self):
        Post.objects.create(title="投稿1", slug="hoge")
        with self.assertRaises(IntegrityError):
            Post.objects.create(title="投稿2", slug="hoge")
        posts_count = Post.objects.all().count()
        self.assertEqual(posts_count, 1)
    // テスト通らない
django.db.transaction.TransactionManagementError: An error occurred in the current transaction. You can't execute queries until the end of the 'atomic' block.

transaction.atomicの例外をキャッチする。

from django.db import transaction
from django.db.utils import IntegrityError


def test_post_duplicate(self):
        Post.objects.create(title="投稿1", slug="hoge")
        try:
            with transaction.atomic():
                Post.objects.create(title="投稿2", slug="hoge")
        except IntegrityError:
            pass
        posts_count = Post.objects.all().count()
        self.assertEqual(posts_count, 1)

参考URL：
https://codehero.jp/python/21458387/transactionmanagementerror-you-cant-execute-queries-until-the-end-of-the-atom

Answer 2

assertRaisesを使用するのが良いかと思います。
https://docs.python.org/ja/3/library/unittest.html#unittest.TestCase.assertRaises

// tests.py
from django.db.utils import IntegrityError

class TestPost(TestCase):
    def test_post(self):
        Post.objects.create(title="投稿1", slug="hoge")
        posts_count = Post.objects.all().count()
        self.assertEqual(posts_count, 1)
    // テスト通る

    def test_post_duplicate(self):
        Post.objects.create(title="投稿1", slug="hoge")
        with self.assertRaises(IntegrityError):
            Post.objects.create(title="投稿2", slug="hoge")
        posts_count = Post.objects.all().count()
        self.assertEqual(posts_count, 1)

Comments

[teosu]

さっそくの回答ありがとうございます。
assertRaisesにて例外をキャッチできました。
ですが、次にTransactionManagementErrorが発生しテストが再び通らなくなりました。
これはDjango1.5以降に意図的に例外をトリガーにすると発生するみたいです。
詳しいコードは別で記載します。
いずれにしろ、解決のきっかけを与えてくださりありがとうございました！