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

"""

 (c) 2016 - Copyright Red Hat Inc

 Authors:

   Adam Williamson <awilliam@redhat.com></awilliam@redhat.com>

"""

from __future__ import unicode_literals, absolute_import

import unittest

import sys

import os

import mock

import munch

import six

sys.path.insert(

    0, os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")

)

import pagure.lib.query  # pylint: disable=wrong-import-position

import pagure.lib.model  # pylint: disable=wrong-import-position

import pagure.lib.notify  # pylint: disable=wrong-import-position

import tests  # pylint: disable=wrong-import-position

class PagureLibNotifyEmailtests(tests.Modeltests):

    """ Some tests for the various email construction functions. In

    their own class so they can have some shared fixtures.

    """

    def setUp(self):

        """ Override setUp to add more fixtures used for many tests. """

        super(PagureLibNotifyEmailtests, self).setUp()

        tests.create_projects(self.session)

        # we don't want to send any mails while setting up

        patcher = mock.patch("pagure.lib.notify.send_email")

        patcher.start()

        self.user1 = pagure.lib.query.get_user(self.session, "pingou")

        self.user2 = pagure.lib.query.get_user(self.session, "foo")

        self.project1 = pagure.lib.query._get_project(self.session, "test")

        self.project2 = pagure.lib.query._get_project(self.session, "test2")

        self.project3 = pagure.lib.query._get_project(

            self.session, "test3", namespace="somenamespace"

        )

        # Create a forked repo, should be project #4

        # Not using fork_project as it tries to do a git clone

        item = pagure.lib.model.Project(

            user_id=2,  # foo

            name="test",

            description="test project #1",

            is_fork=True,

            parent_id=1,

            hook_token="aaabbbyyy",

        )

        self.session.add(item)

        self.session.commit()

        self.forkedproject = pagure.lib.query._get_project(

            self.session, "test", user="foo"

        )

        # Report an issue on project #1

        self.issue1 = pagure.lib.query.new_issue(

            session=self.session,

            repo=self.project1,

            title="issue",

            content="a bug report",

            user="pingou",

        )

        # Add a comment on the issue

        pagure.lib.query.add_issue_comment(

            self.session, self.issue1, comment="Test comment", user="pingou"

        )

        self.comment1 = pagure.lib.query.get_issue_comment(

            self.session, self.issue1.uid, 1

        )

        # Report an issue on project #3 (namespaced)

        self.issue2 = pagure.lib.query.new_issue(

            session=self.session,

            repo=self.project3,

            title="namespaced project issue",

            content="a bug report on a namespaced project",

            user="pingou",

        )

        # report an issue on foo's fork of project #1

        self.issue3 = pagure.lib.query.new_issue(

            session=self.session,

            repo=self.forkedproject,

            title="forked project issue",

            content="a bug report on a forked project",

            user="pingou",

        )

        patcher.stop()

    @mock.patch("pagure.lib.notify.send_email")

    def test_notify_new_comment(self, fakemail):

        """Simple test for notification about new comment."""

        exptext = """

pingou added a new comment to an issue you are following:

``

Test comment

``

To reply, visit the link below

http://localhost.localdomain/test/issue/1

"""

        pagure.lib.notify.notify_new_comment(self.comment1)

        (_, args, kwargs) = fakemail.mock_calls[0]

        # Mail text should be as expected.

        self.assertEqual(args[0], exptext)

        self.assertTrue(isinstance(args[0], six.text_type))

        # Mail subject should be as expected.

        self.assertEqual(args[1], "Issue #1: issue")

        # Mail should be sent to user #1.

        self.assertEqual(args[2], self.user1.default_email)

        # Mail ID should be comment #1's mail ID...

        self.assertEqual(kwargs["mail_id"], self.comment1.mail_id)

        # In reply to issue #1's mail ID.

        self.assertEqual(kwargs["in_reply_to"], self.issue1.mail_id)

        # Project name should be...project (full) name.

        self.assertEqual(kwargs["project_name"], self.project1.fullname)

        # Mail should be from user1 (who wrote the comment).

        self.assertEqual(kwargs["user_from"], self.user1.fullname)

    @mock.patch("pagure.lib.notify.send_email")

    def test_user_notified_new_comment(self, fakemail):

        """Check that users that are @mentionned are notified."""

        self.comment1.comment = "Hey @foo. Let's do it!"

        g = munch.Munch()

        g.session = self.session

        with mock.patch("flask.g", g):

            pagure.lib.notify.notify_new_comment(self.comment1)

        (_, args, kwargs) = fakemail.mock_calls[0]

        # Mail should be sent to both users

        self.assertIn(

            args[2],

            ["bar@pingou.com,foo@bar.com", "foo@bar.com,bar@pingou.com"],

        )

        # Mail ID should be comment #1's mail ID...

        self.assertEqual(kwargs["mail_id"], self.comment1.mail_id)

        # In reply to issue #1's mail ID.

        self.assertEqual(kwargs["in_reply_to"], self.issue1.mail_id)

        # Project name should be...project (full) name.

        self.assertEqual(kwargs["project_name"], self.project1.fullname)

        # Mail should be from user1 (who wrote the comment).

        self.assertEqual(kwargs["user_from"], self.user1.fullname)

    @mock.patch("pagure.lib.notify.send_email")

    def test_user_notified_new_comment_start_row(self, fakemail):

        """Check that users that are @mentionned are notified."""

        self.comment1.comment = "@foo, okidoki"

        g = munch.Munch()

        g.session = self.session

        with mock.patch("flask.g", g):

            pagure.lib.notify.notify_new_comment(self.comment1)

        (_, args, kwargs) = fakemail.mock_calls[0]

        # Mail should be sent to both users

        self.assertIn(

            args[2],

            ["bar@pingou.com,foo@bar.com", "foo@bar.com,bar@pingou.com"],

        )

        # Mail ID should be comment #1's mail ID...

        self.assertEqual(kwargs["mail_id"], self.comment1.mail_id)

        # In reply to issue #1's mail ID.

        self.assertEqual(kwargs["in_reply_to"], self.issue1.mail_id)

        # Project name should be...project (full) name.

        self.assertEqual(kwargs["project_name"], self.project1.fullname)

        # Mail should be from user1 (who wrote the comment).

        self.assertEqual(kwargs["user_from"], self.user1.fullname)

    @mock.patch("pagure.lib.notify.send_email")

    def test_user_notified_new_comment_with_email(self, fakemail):

        """Ensures that @mention doesn't over-reach."""

        self.comment1.comment = "So apparently bar@foo.com exists"

        g = munch.Munch()

        g.fas_user = tests.FakeUser(username="pingou")

        g.authenticated = True

        g.session = self.session

        with mock.patch("flask.g", g):

            pagure.lib.notify.notify_new_comment(self.comment1)

        (_, args, kwargs) = fakemail.mock_calls[0]

        # Mail should be sent to both users

        self.assertEqual(args[2], "bar@pingou.com")

        # Mail ID should be comment #1's mail ID...

        self.assertEqual(kwargs["mail_id"], self.comment1.mail_id)

        # In reply to issue #1's mail ID.

        self.assertEqual(kwargs["in_reply_to"], self.issue1.mail_id)

        # Project name should be...project (full) name.

        self.assertEqual(kwargs["project_name"], self.project1.fullname)

        # Mail should be from user1 (who wrote the comment).

        self.assertEqual(kwargs["user_from"], self.user1.fullname)

    @mock.patch("pagure.lib.notify.send_email")

    def test_user_notified_new_comment_with_email_with_number(self, fakemail):

        """Ensures that @mention doesn't over-reach."""

        self.comment1.comment = "So apparently bar123@foo.com exists"

        g = munch.Munch()

        g.fas_user = tests.FakeUser(username="pingou")

        g.authenticated = True

        g.session = self.session

        with mock.patch("flask.g", g):

            pagure.lib.notify.notify_new_comment(self.comment1)

        (_, args, kwargs) = fakemail.mock_calls[0]

        # Mail should be sent to both users

        self.assertEqual(args[2], "bar@pingou.com")

        # Mail ID should be comment #1's mail ID...

        self.assertEqual(kwargs["mail_id"], self.comment1.mail_id)

        # In reply to issue #1's mail ID.

        self.assertEqual(kwargs["in_reply_to"], self.issue1.mail_id)

        # Project name should be...project (full) name.

        self.assertEqual(kwargs["project_name"], self.project1.fullname)

        # Mail should be from user1 (who wrote the comment).

        self.assertEqual(kwargs["user_from"], self.user1.fullname)

    @mock.patch("pagure.lib.notify.send_email")

    def test_notify_new_issue_namespaced(

        self, fakemail

    ):  # pylint: disable=invalid-name

        """Test for notifying of a new issue, namespaced project."""

        exptext = """

pingou reported a new issue against the project: `test3` that you are following:

``

a bug report on a namespaced project

``

To reply, visit the link below

http://localhost.localdomain/somenamespace/test3/issue/1

"""

        pagure.lib.notify.notify_new_issue(self.issue2)

        (_, args, kwargs) = fakemail.mock_calls[0]

        # Mail text should be as expected.

        self.assertEqual(args[0], exptext)

        self.assertTrue(isinstance(args[0], six.text_type))

        # Mail subject should be as expected.

        self.assertEqual(args[1], "Issue #1: namespaced project issue")

        # Mail should be sent to user #1.

        self.assertEqual(args[2], self.user1.default_email)

        # Mail ID should be issue's mail ID.

        self.assertEqual(kwargs["mail_id"], self.issue2.mail_id)

        # Project name should be...project (full) name.

        self.assertEqual(kwargs["project_name"], self.project3.fullname)

        # Mail should be from user1 (who submitted the issue).

        self.assertEqual(kwargs["user_from"], self.user1.fullname)

    @mock.patch("pagure.lib.notify.send_email")

    def test_notify_assigned_issue_forked(

        self, fakemail

    ):  # pylint: disable=invalid-name

        """Test for notifying re-assignment of issue on forked project.

        'foo' reassigns issue on his fork of 'test' to 'pingou'.

        """

        exptext = """

The issue: `forked project issue` of project: `test` has been assigned to `pingou` by foo.

http://localhost.localdomain/fork/foo/test/issue/1

"""

        pagure.lib.notify.notify_assigned_issue(

            self.issue3, self.user1, self.user2

        )

        (_, args, kwargs) = fakemail.mock_calls[0]

        # Mail text should be as expected.

        self.assertEqual(args[0], exptext)

        self.assertTrue(isinstance(args[0], six.text_type))

        # Mail subject should be as expected.

        self.assertEqual(args[1], "Issue #1: forked project issue")

        # Mail should be sent to user #1.

        # NOTE: Not sent to user #2...

        self.assertEqual(args[2], self.user1.default_email)

        # Mail ID should contain issue's mail ID and '/assigned/'

        self.assertIn(

            "{0}/assigned/".format(self.issue3.mail_id), kwargs["mail_id"]

        )

        # Project name should be...project (full) name.

        self.assertEqual(kwargs["project_name"], self.forkedproject.fullname)

        # Mail should be from user1 (who submitted the issue).

        self.assertEqual(kwargs["user_from"], self.user2.fullname)

    @mock.patch("pagure.lib.notify.send_email")

    # for non-ASCII testing, we mock these return values

    @mock.patch("pagure.lib.git.get_author", return_value="Cecil Cõmmîttër")

    @mock.patch(

        "pagure.lib.git.get_commit_subject", return_value="We love Motörhead"

    )

    def test_notify_new_commits(

        self, _, __, fakemail

    ):  # pylint: disable=invalid-name

        """Test for notification on new commits, especially when

        non-ASCII text is involved.

        """

        exptext = """

The following commits were pushed to the repo test on branch

master, which you are following:

abcdefg    Cecil Cõmmîttër    We love Motörhead

To view more about the commits, visit:

http://localhost.localdomain/test/commits/master

"""

        # first arg (abspath) doesn't matter and we can use a commit

        # ID that doesn't actually exist, as we are mocking

        # the get_author and get_commit_subject calls anyway

        pagure.lib.notify.notify_new_commits(

            "/", self.project1, "master", ["abcdefg"]

        )

        (_, args, kwargs) = fakemail.mock_calls[0]

        # Mail text should be as expected.

        self.assertEqual(args[0], exptext)

        self.assertTrue(isinstance(args[0], six.text_type))

        # Mail subject should be as expected.

        self.assertEqual(args[1], 'New Commits To "test" (master)')

        # Mail doesn't actually get sent to anyone by default

        self.assertEqual(args[2], "")

        # Project name should be...project (full) name.

        self.assertEqual(kwargs["project_name"], self.project1.fullname)

# Add more tests to verify that correct mails are sent to correct people here

if __name__ == "__main__":

    unittest.main(verbosity=2)