Mercurial > public > sg101
view forums/tests/test_receivers.py @ 1089:64fb0d6933a4
Hide podcast logo on small screens.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Sun, 08 May 2016 21:03:09 -0500 |
parents | 8b48ec450497 |
children |
line wrap: on
line source
"""Tests for the forum app's signal handlers.""" from django.contrib.auth.models import User from django.test import TestCase from mock import patch import custom_search.receivers from forums.models import Forum from forums.models import Post from forums.models import Topic class ReceiverTestCase(TestCase): fixtures = ['forums.json'] def setUp(self): self.user = User.objects.create_user('user', 'user@example.com', 'pw') # Don't let our custom search signal handler class catch any of the # signals we are throwing here. custom_search.receivers.signal_processor.teardown() def tearDown(self): custom_search.receivers.signal_processor.setup() @patch('forums.latest.notify_topic_delete') def test_signal_handlers(self, mock_latest): forum = Forum.objects.get(pk=2) topic = Topic(forum=forum, name='Test Topic', user=self.user) topic.save() forum = Forum.objects.get(pk=2) self.assertEqual(1, forum.topic_count) post = Post(topic=topic, user=self.user, body='test') post.save() topic = Topic.objects.get(pk=topic.pk) self.assertEqual(topic.post_count, 1) forum = Forum.objects.get(pk=2) self.assertEqual(forum.post_count, 1) # To prevent cascading deletes topic.last_post_pre_delete() topic.save() forum.last_post_pre_delete() forum.save() post.delete() topic = Topic.objects.get(pk=topic.pk) self.assertEqual(topic.post_count, 0) forum = Forum.objects.get(pk=2) self.assertEqual(forum.post_count, 0) topic.delete() forum = Forum.objects.get(pk=2) self.assertEqual(0, forum.topic_count) mock_latest.assert_called_once_with(topic)