# -*- 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)