view forums/tests/test_receivers.py @ 1146:439eb97a6ca8

Add friends logos to V3 base template.
author Brian Neal <bgneal@gmail.com>
date Tue, 15 Nov 2016 19:44:29 -0600
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)