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

"""

 (c) 2015-2018 - Copyright Red Hat Inc

 Authors:

   Patrick Uiterwijk <puiterwijk@redhat.com></puiterwijk@redhat.com>

"""

from __future__ import unicode_literals, absolute_import

import datetime

import unittest

import shutil

import sys

import tempfile

import os

import six

import json

import pygit2

from mock import patch, MagicMock

sys.path.insert(

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

)

import pagure.lib.query

import tests

class PagureFlaskAppClonetests(tests.Modeltests):

    """ Tests for the clone bridging. """

    def setUp(self):

        super(PagureFlaskAppClonetests, self).setUp()

        tests.create_projects(self.session)

        tests.create_tokens(self.session)

        tests.create_tokens_acl(self.session)

        self.create_project_full("clonetest", {"create_readme": "y"})

    @patch.dict("pagure.config.config", {"ALLOW_HTTP_PULL_PUSH": False})

    def test_http_clone_disabled(self):

        """ Test that the HTTP clone endpoint gets correctly closed. """

        output = self.app.get(

            "/clonetest.git/info/refs?service=git-upload-pack"

        )

        self.assertEqual(output.status_code, 403)

        self.assertIn("not allowed", output.get_data(as_text=True))

    @patch.dict("pagure.config.config", {"ALLOW_HTTP_PULL_PUSH": True})

    def test_http_clone_invalid_service(self):

        """ Test that the HTTP endpoint refuses invalid services. """

        output = self.app.get("/clonetest.git/info/refs?service=myservice")

        self.assertEqual(output.status_code, 400)

        self.assertIn("Unknown service", output.get_data(as_text=True))

    @patch.dict("pagure.config.config", {"ALLOW_HTTP_PULL_PUSH": True})

    def test_http_clone_invalid_project(self):

        """ Test that the HTTP endpoint refuses invalid projects. """

        output = self.app.get(

            "/nosuchrepo.git/info/refs?service=git-upload-pack"

        )

        self.assertEqual(output.status_code, 404)

        self.assertIn("Project not found", output.get_data(as_text=True))

    @patch.dict("pagure.config.config", {"ALLOW_HTTP_PULL_PUSH": True})

    def test_http_clone_dumb(self):

        """ Test that the HTTP endpoint refuses dumb service request. """

        output = self.app.get("/clonetest.git/info/refs")

        self.assertEqual(output.status_code, 400)

        self.assertIn("Please switch", output.get_data(as_text=True))

    @patch.dict(

        "pagure.config.config",

        {

            "ALLOW_HTTP_PULL_PUSH": True,

            "ALLOW_HTTP_PUSH": False,

            "HTTP_REPO_ACCESS_GITOLITE": None,

        },

    )

    def test_http_push_disabled(self):

        """ Test that the HTTP push gets refused. """

        output = self.app.get(

            "/clonetest.git/info/refs?service=git-receive-pack"

        )

        self.assertEqual(output.status_code, 403)

        self.assertIn("pushing disabled", output.get_data(as_text=True))

        output = self.app.post("/clonetest.git/git-receive-pack")

        self.assertEqual(output.status_code, 403)

        self.assertIn("pushing disabled", output.get_data(as_text=True))

    @patch.dict(

        "pagure.config.config",

        {

            "ALLOW_HTTP_PULL_PUSH": True,

            "ALLOW_HTTP_PUSH": True,

            "HTTP_REPO_ACCESS_GITOLITE": None,

        },

    )

    def test_http_push_unauthed(self):

        """ Test that the HTTP push gets refused unauthed. """

        output = self.app.get(

            "/clonetest.git/info/refs?service=git-receive-pack"

        )

        self.assertEqual(output.status_code, 403)

        self.assertIn("Unauthenticated push", output.get_data(as_text=True))

    @patch.dict("pagure.config.config", {"ALLOW_HTTP_PULL_PUSH": True})

    def test_http_clone_private_project_unauthed(self):

        """ Test that the HTTP endpoint enforced project.private. """

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

        project.private = True

        self.session.add(project)

        self.session.commit()

        output = self.app.get(

            "/clonetest.git/info/refs?service=git-upload-pack"

        )

        self.assertEqual(output.status_code, 404)

        self.assertIn("Project not found", output.get_data(as_text=True))

    @patch.dict(

        "pagure.config.config",

        {

            "ALLOW_HTTP_PULL_PUSH": True,

            "ALLOW_HTTP_PUSH": False,

            "HTTP_REPO_ACCESS_GITOLITE": None,

        },

    )

    def test_http_clone(self):

        """ Test that HTTP cloning gives reasonable output. """

        # Unfortunately, actually testing a git clone would need the app to

        # run on a TCP port, which the test environment doesn't do.

        output = self.app.get(

            "/clonetest.git/info/refs?service=git-upload-pack"

        )

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn("# service=git-upload-pack", output_text)

        self.assertIn(" refs/heads/master\n0000", output_text)

        output = self.app.post(

            "/clonetest.git/git-upload-pack",

            headers={"Content-Type": "application/x-git-upload-pack-request"},

        )

        # Git 2.17 returns 415, older return 200

        # Either means we didn't fully crash when returning the response

        self.assertIn(output.status_code, (200, 415))

    @patch.dict(

        "pagure.config.config",

        {

            "ALLOW_HTTP_PULL_PUSH": True,

            "ALLOW_HTTP_PUSH": False,

            "HTTP_REPO_ACCESS_GITOLITE": None,

        },

    )

    def test_http_clone_private(self):

        """ Test that HTTP cloning gives reasonable output with project.private. """

        # Unfortunately, actually testing a git clone would need the app to

        # run on a TCP port, which the test environment doesn't do.

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

        project.private = True

        self.session.add(project)

        self.session.commit()

        output = self.app.get(

            "/clonetest.git/info/refs?service=git-upload-pack"

        )

        self.assertEqual(output.status_code, 404)

        self.assertIn("Project not found", output.get_data(as_text=True))

        output = self.app.get(

            "/clonetest.git/info/refs?service=git-upload-pack",

            environ_overrides={"REMOTE_USER": "pingou"},

        )

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn("# service=git-upload-pack", output_text)

        self.assertIn(" refs/heads/master\n0000", output_text)

    @patch.dict(

        "pagure.config.config",

        {

            "ALLOW_HTTP_PULL_PUSH": True,

            "ALLOW_HTTP_PUSH": True,

            "HTTP_REPO_ACCESS_GITOLITE": None,

        },

    )

    def test_http_push(self):

        """ Test that the HTTP push gets accepted. """

        output = self.app.get(

            "/clonetest.git/info/refs?service=git-receive-pack",

            environ_overrides={"REMOTE_USER": "pingou"},

        )

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn("# service=git-receive-pack", output_text)

        self.assertIn(" refs/heads/master\x00", output_text)