test_helpers

Complexity

Total Complexity

3

Size/Duplication

Total Lines	39
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	tcms.core.tests.test_helpers.TestAddComments.setUpTestData()	0	7	1
A	tcms.core.tests.test_helpers.TestAddComments.test_add_comment_to_case()	0	17	2

# -*- coding: utf-8 -*-

from django.contrib.auth.models import User
from django.contrib.contenttypes.models import ContentType
from django_comments.models import Comment

from tcms.tests import BasePlanCase
from tcms.core.helpers.comments import add_comment


class TestAddComments(BasePlanCase):
    """Test comments.add_comment"""

    @classmethod
    def setUpTestData(cls):
        super(TestAddComments, cls).setUpTestData()

        cls.reviewer = User.objects.create_user(
            username='reviewer',
            email='reviewer@example.com')

    def test_add_comment_to_case(self):
        cases = [self.case_1, self.case_2]
        add_comment(cases, 'new comment', self.reviewer)

        for case in cases:
            case_ct = ContentType.objects.get_for_model(case.__class__)
            comments = Comment.objects.filter(content_type=case_ct,
                                              object_pk=case.pk)
            self.assertEqual(1, len(comments))

            comment = comments[0]
            self.assertEqual('new comment', comment.comment)
            self.assertEqual(self.reviewer, comment.user)
            self.assertEqual(self.reviewer.username, comment.user_name)
            self.assertEqual(self.reviewer.email, comment.user_email)
            self.assertTrue(comment.is_public)
            self.assertFalse(comment.is_removed)