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

"""

 (c) 2018 - Copyright Red Hat Inc

 Authors:

   Clement Verna <cverna@tutanota.com></cverna@tutanota.com>

"""

from __future__ import unicode_literals, absolute_import

import unittest

import sys

import os

import time

import mock

import pygit2

sys.path.insert(

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

)

import pagure.lib.git

import pagure.lib.query

import tests

from tests.test_pagure_lib_git_get_tags_objects import add_repo_tag

class PagureFlaskUiArchivesTest(tests.Modeltests):

    """ Tests checking the archiving mechanism. """

    def setUp(self):

        """ Set up the environnment, ran before every tests. """

        super(PagureFlaskUiArchivesTest, self).setUp()

        tests.create_projects(self.session)

        tests.create_projects_git(os.path.join(self.path, "repos"), bare=True)

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

        # test has both commits and tags

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

        tests.add_readme_git_repo(repopath)

        repo = pygit2.Repository(repopath)

        add_repo_tag(self.path, repo, ["v1.0", "v1.1"], "test.git")

        # test2 has only commits

        tests.add_readme_git_repo(

            os.path.join(self.path, "repos", "test2.git")

        )

        # somenamespace/test3 has neither commits nor tags

        # Create the archive folder:

        self.archive_path = os.path.join(self.path, "archives")

        os.mkdir(self.archive_path)

    def test_project_no_conf(self):

        """ Test getting the archive when pagure isn't configured. """

        output = self.app.get(

            "/somenamespace/test3/archive/tag1/test3-tag1.zip",

            follow_redirects=True,

        )

        self.assertEqual(output.status_code, 404)

        self.assertIn(

            "This pagure instance isn't configured to support "

            "this feature",

            output.get_data(as_text=True),

        )

        self.assertEqual(os.listdir(self.archive_path), [])

    def test_project_invalid_conf(self):

        """ Test getting the archive when pagure is wrongly configured. """

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "invalid")},

        ):

            output = self.app.get(

                "/somenamespace/test3/archive/tag1/test3-tag1.zip",

                follow_redirects=True,

            )

            self.assertEqual(output.status_code, 500)

            self.assertIn(

                "Incorrect configuration, please contact your admin",

                output.get_data(as_text=True),

            )

        self.assertEqual(os.listdir(self.archive_path), [])

    def test_project_invalid_format(self):

        """ Test getting the archive when the format provided is invalid. """

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/somenamespace/test3/archive/tag1/test3-tag1.unzip",

                follow_redirects=True,

            )

            self.assertEqual(output.status_code, 404)

        self.assertEqual(os.listdir(self.archive_path), [])

    def test_project_no_commit(self):

        """ Test getting the archive of an empty project. """

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/somenamespace/test3/archive/tag1/test3-tag1.zip",

                follow_redirects=True,

            )

            self.assertEqual(output.status_code, 404)

            self.assertIn(

                "
Invalid commit provided
", output.get_data(as_text=True)

            )

        self.assertEqual(os.listdir(self.archive_path), [])

    def test_project_no_tag(self):

        """ Test getting the archive of a non-empty project but without

        tags. """

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/test2/archive/tag1/test2-tag1.zip", follow_redirects=True

            )

            self.assertEqual(output.status_code, 404)

            self.assertIn(

                "
Invalid commit provided
", output.get_data(as_text=True)

            )

        self.assertEqual(os.listdir(self.archive_path), [])

    def test_project_no_tag(self):

        """ Test getting the archive of an empty project. """

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/test2/archive/tag1/test2-tag1.zip", follow_redirects=True

            )

            self.assertEqual(output.status_code, 404)

            self.assertIn(

                "
Invalid commit provided
", output.get_data(as_text=True)

            )

        self.assertEqual(os.listdir(self.archive_path), [])

    def test_project_w_tag_zip(self):

        """ Test getting the archive from a tag. """

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/test/archive/v1.0/test-v1.0.zip", follow_redirects=True

            )

            self.assertEqual(output.status_code, 200)

        self.assertEqual(os.listdir(self.archive_path), ["test"])

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test")), ["tags"]

        )

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test", "tags")),

            ["v1.0"],

        )

        self.assertEqual(

            len(

                os.listdir(

                    os.path.join(self.archive_path, "test", "tags", "v1.0")

                )

            ),

            1,

        )

        files = os.listdir(

            os.path.join(self.archive_path, "test", "tags", "v1.0")

        )

        self.assertEqual(

            os.listdir(

                os.path.join(

                    self.archive_path, "test", "tags", "v1.0", files[0]

                )

            ),

            ["test-v1.0.zip"],

        )

    def test_project_w_tag_tar(self):

        """ Test getting the archive from a tag. """

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/test/archive/v1.0/test-v1.0.tar", follow_redirects=True

            )

            self.assertEqual(output.status_code, 200)

        self.assertEqual(os.listdir(self.archive_path), ["test"])

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test")), ["tags"]

        )

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test", "tags")),

            ["v1.0"],

        )

        self.assertEqual(

            len(

                os.listdir(

                    os.path.join(self.archive_path, "test", "tags", "v1.0")

                )

            ),

            1,

        )

        files = os.listdir(

            os.path.join(self.archive_path, "test", "tags", "v1.0")

        )

        self.assertEqual(

            os.listdir(

                os.path.join(

                    self.archive_path, "test", "tags", "v1.0", files[0]

                )

            ),

            ["test-v1.0.tar"],

        )

    def test_project_w_tag_tar_gz(self):

        """ Test getting the archive from a tag. """

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/test/archive/v1.0/test-v1.0.tar.gz", follow_redirects=True

            )

            self.assertEqual(output.status_code, 200)

        self.assertEqual(os.listdir(self.archive_path), ["test"])

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test")), ["tags"]

        )

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test", "tags")),

            ["v1.0"],

        )

        self.assertEqual(

            len(

                os.listdir(

                    os.path.join(self.archive_path, "test", "tags", "v1.0")

                )

            ),

            1,

        )

        files = os.listdir(

            os.path.join(self.archive_path, "test", "tags", "v1.0")

        )

        self.assertEqual(

            os.listdir(

                os.path.join(

                    self.archive_path, "test", "tags", "v1.0", files[0]

                )

            ),

            ["test-v1.0.tar.gz"],

        )

    def test_project_w_commit_tar_gz(self):

        """ Test getting the archive from a commit. """

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

        repo = pygit2.Repository(repopath)

        commit = repo.head.target.hex

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/test/archive/%s/test-v1.0.tar.gz" % commit,

                follow_redirects=True,

            )

            self.assertEqual(output.status_code, 200)

        self.assertEqual(os.listdir(self.archive_path), ["test"])

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test")), [commit]

        )

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test", commit)),

            ["test-v1.0.tar.gz"],

        )

    def test_project_w_commit_tar_gz_twice(self):

        """ Test getting the archive from a commit twice, so we hit the

        disk cache. """

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

        repo = pygit2.Repository(repopath)

        commit = repo.head.target.hex

        with mock.patch.dict(

            "pagure.config.config",

            {"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},

        ):

            output = self.app.get(

                "/test/archive/%s/test-v1.0.tar.gz" % commit,

                follow_redirects=True,

            )

            self.assertEqual(output.status_code, 200)

            output = self.app.get(

                "/test/archive/%s/test-v1.0.tar.gz" % commit,

                follow_redirects=True,

            )

            self.assertEqual(output.status_code, 200)

        self.assertEqual(os.listdir(self.archive_path), ["test"])

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test")), [commit]

        )

        self.assertEqual(

            os.listdir(os.path.join(self.archive_path, "test", commit)),

            ["test-v1.0.tar.gz"],

        )

if __name__ == "__main__":

    unittest.main(verbosity=2)