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

"""

 (c) 2015 - Copyright Red Hat Inc

 Authors:

   Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>

"""

from __future__ import unicode_literals, absolute_import

import json

import unittest

import shutil

import sys

import tempfile

import os

import pygit2

from mock import patch

sys.path.insert(

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

)

import pagure.lib.query

import tests

from pagure.lib.repo import PagureRepo

class PagureFlaskNoMasterBranchtests(tests.SimplePagureTest):

    """ Tests for flask application when the git repo has no master branch.

    """

    def set_up_git_repo(self):

        """ Set up the git repo to play with. """

        # Create a git repo to play with

        gitrepo = os.path.join(self.path, "repos", "test.git")

        repo = pygit2.init_repository(gitrepo, bare=True)

        newpath = tempfile.mkdtemp(prefix="pagure-other-test")

        repopath = os.path.join(newpath, "test")

        clone_repo = pygit2.clone_repository(gitrepo, repopath)

        # Create a file in that git repo

        with open(os.path.join(repopath, "sources"), "w") as stream:

            stream.write("foo\n bar")

        clone_repo.index.add("sources")

        clone_repo.index.write()

        # Commits the files added

        tree = clone_repo.index.write_tree()

        author = pygit2.Signature("Alice Author", "alice@authors.tld")

        committer = pygit2.Signature("Cecil Committer", "cecil@committers.tld")

        clone_repo.create_commit(

            "refs/heads/feature",  # the name of the reference to update

            author,

            committer,

            "Add sources file for testing",

            # binary string representing the tree object ID

            tree,

            # list of binary strings representing parents of the new commit

            [],

        )

        feature_branch = clone_repo.lookup_branch("feature")

        first_commit = feature_branch.peel().hex

        # Second commit

        with open(os.path.join(repopath, ".gitignore"), "w") as stream:

            stream.write("*~")

        clone_repo.index.add(".gitignore")

        clone_repo.index.write()

        tree = clone_repo.index.write_tree()

        author = pygit2.Signature("Alice Author", "alice@authors.tld")

        committer = pygit2.Signature("Cecil Committer", "cecil@committers.tld")

        clone_repo.create_commit(

            "refs/heads/feature",

            author,

            committer,

            "Add .gitignore file for testing",

            # binary string representing the tree object ID

            tree,

            # list of binary strings representing parents of the new commit

            [first_commit],

        )

        refname = "refs/heads/feature"

        ori_remote = clone_repo.remotes[0]

        PagureRepo.push(ori_remote, refname)

        shutil.rmtree(newpath)

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

    def test_view_repo(self, send_email):

        """ Test the view_repo endpoint when the git repo has no master

        branch.

        """

        send_email.return_value = True

        tests.create_projects(self.session)

        # Non-existant git repo

        output = self.app.get("/test")

        self.assertEqual(output.status_code, 404)

        self.set_up_git_repo()

        # With git repo

        output = self.app.get("/test")

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn('<section class="no-readme">', output_text)</section>

        self.assertIn(

            "The test project's README file is empty or unavailable.",

            output_text,

        )

        self.assertEqual(

            output_text.count('

            0,

        )

    '''

    @patch('pagure.lib.notify.send_email')

    def test_view_repo_branch(self, send_email):

        """ Test the view_repo_branch endpoint when the git repo has no

        master branch.

        """

        send_email.return_value = True

        tests.create_projects(self.session)

        # Non-existant git repo

        output = self.app.get('/test/branch/master')

        self.assertEqual(output.status_code, 404)

        self.set_up_git_repo()

        # With git repo

        output = self.app.get('/test/branch/feature')

        output_text = output.get_data(as_text=True)

        print(output_text)

        self.assertEqual(output.status_code, 200)

        self.assertIn('<section class="no-readme">', output_text)</section>

        self.assertIn(

            "The test project's README file is empty or unavailable.",

            output_text)

    '''

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

    def test_view_commits(self, send_email):

        """ Test the view_commits endpoint when the git repo has no

        master branch.

        """

        send_email.return_value = True

        tests.create_projects(self.session)

        # Non-existant git repo

        output = self.app.get("/test/commits")

        self.assertEqual(output.status_code, 404)

        self.set_up_git_repo()

        # With git repo

        output = self.app.get("/test/commits")

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            '
\n\n\n          
', output_text

        )

        self.assertNotIn('
', output_text)

        output = self.app.get("/test/commits/feature")

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            'Commits  2',

            output_text,

        )

        self.assertIn("Add sources file for testing", output_text)

        self.assertIn("Add .gitignore file for testing", output_text)

        self.assertNotIn(

            '
\n\n\n          
', output_text

        )

        self.assertEqual(output_text.count('class="list-group-item "'), 2)

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

    def test_view_file(self, send_email):

        """ Test the view_file endpoint when the git repo has no

        master branch.

        """

        send_email.return_value = True

        tests.create_projects(self.session)

        # Non-existant git repo

        output = self.app.get("/test/blob/master/f/sources")

        self.assertEqual(output.status_code, 404)

        self.set_up_git_repo()

        # With git repo

        output = self.app.get("/test/blob/master/f/sources")

        self.assertEqual(output.status_code, 404)

        output = self.app.get("/test/blob/feature/f/sources")

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            """

                feature

              sources

        """,

            output_text,

        )

        self.assertIn(

            '
'

            'foo\n bar',

            output_text,

        )

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

    def test_view_raw_file(self, send_email):

        """ Test the view_raw_file endpoint when the git repo has no

        master branch.

        """

        send_email.return_value = True

        tests.create_projects(self.session)

        # Non-existant git repo

        output = self.app.get("/test/raw/master")

        self.assertEqual(output.status_code, 404)

        output = self.app.get("/test/raw/master/f/sources")

        self.assertEqual(output.status_code, 404)

        self.set_up_git_repo()

        # With git repo

        output = self.app.get("/test/raw/master")

        self.assertEqual(output.status_code, 404)

        output = self.app.get("/test/raw/master/f/sources")

        self.assertEqual(output.status_code, 404)

        output = self.app.get("/test/raw/feature")

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn("diff --git a/.gitignore b/.gitignore", output_text)

        self.assertIn("+*~", output_text)

        output = self.app.get("/test/raw/feature/f/sources")

        self.assertEqual(output.status_code, 200)

        self.assertEqual("foo\n bar", output.get_data(as_text=True))

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

    def test_view_tree(self, send_email):

        """ Test the view_tree endpoint when the git repo has no

        master branch.

        """

        send_email.return_value = True

        tests.create_projects(self.session)

        # Non-existant git repo

        output = self.app.get("/test/tree/")

        self.assertEqual(output.status_code, 404)

        output = self.app.get("/test/tree/master")

        self.assertEqual(output.status_code, 404)

        self.set_up_git_repo()

        # With git repo

        output = self.app.get("/test/tree/master")

        self.assertEqual(output.status_code, 200)

        self.assertIn(

            "No content found in this repository",

            output.get_data(as_text=True),

        )

        output = self.app.get("/test/tree/master/sources")

        self.assertEqual(output.status_code, 200)

        self.assertIn(

            "No content found in this repository",

            output.get_data(as_text=True),

        )

        output = self.app.get("/test/tree/feature")

        self.assertEqual(output.status_code, 200)

        self.assertIn(

            '',

            output.get_data(as_text=True),

        )

        output = self.app.get("/test/tree/feature/sources")

        self.assertEqual(output.status_code, 200)

        self.assertIn(

            "No content found in this repository",

            output.get_data(as_text=True),

        )

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

    def test_new_request_pull(self, send_email):

        """ Test the new_request_pull endpoint when the git repo has no

        master branch.

        """

        send_email.return_value = True

        tests.create_projects(self.session)

        # Non-existant git repo

        output = self.app.get("/test/diff/master..feature")

        # (used to be 302 but seeing a diff is allowed even logged out)

        self.assertEqual(output.status_code, 404)

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            output = self.app.get("/test/diff/master..feature")

            self.assertEqual(output.status_code, 404)

        self.set_up_git_repo()

        output = self.app.get("/test/diff/master..feature")

        # (used to be 302 but seeing a diff is allowed even logged out)

        self.assertEqual(output.status_code, 400)

        self.assertIn(

            "
Branch master could not be found in the target repo
",

            output.get_data(as_text=True),

        )

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            output = self.app.get("/test/diff/master..feature")

            self.assertEqual(output.status_code, 400)

            self.assertIn(

                "
Branch master could not be found in the target repo
",

                output.get_data(as_text=True),

            )

if __name__ == "__main__":

    unittest.main(verbosity=2)