# -*- coding: utf-8 -*-
# Copyright (C) 2014-2016 Andrey Antukh <niwi@niwi.nz>
# Copyright (C) 2014-2016 Jesús Espino <jespinog@gmail.com>
# Copyright (C) 2014-2016 David Barragán <bameda@dbarragan.com>
# Copyright (C) 2014-2016 Alejandro Alonso <alejandro.alonso@kaleidos.net>
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import pytest

from unittest import mock

from django.core.urlresolvers import reverse

from .. import factories as f
from taiga.base.utils import json


pytestmark = pytest.mark.django_db


def test_invalid_project_export(client):
    user = f.UserFactory.create()
    client.login(user)

    url = reverse("exporter-detail", args=[1000000])

    response = client.get(url, content_type="application/json")
    assert response.status_code == 404


def test_valid_project_export_with_celery_disabled(client, settings):
    settings.CELERY_ENABLED = False

    user = f.UserFactory.create()
    project = f.ProjectFactory.create(owner=user)
    f.MembershipFactory(project=project, user=user, is_admin=True)
    client.login(user)

    url = reverse("exporter-detail", args=[project.pk])

    response = client.get(url, content_type="application/json")
    assert response.status_code == 200
    response_data = response.data
    assert "url" in response_data
    assert response_data["url"].endswith(".json")


def test_valid_project_export_with_celery_disabled_and_gzip(client, settings):
    settings.CELERY_ENABLED = False

    user = f.UserFactory.create()
    project = f.ProjectFactory.create(owner=user)
    f.MembershipFactory(project=project, user=user, is_admin=True)
    client.login(user)

    url = reverse("exporter-detail", args=[project.pk])

    response = client.get(url+"?dump_format=gzip", content_type="application/json")
    assert response.status_code == 200
    response_data = response.data
    assert "url" in response_data
    assert response_data["url"].endswith(".gz")


def test_valid_project_export_with_celery_enabled(client, settings):
    settings.CELERY_ENABLED = True

    user = f.UserFactory.create()
    project = f.ProjectFactory.create(owner=user)
    f.MembershipFactory(project=project, user=user, is_admin=True)
    client.login(user)

    url = reverse("exporter-detail", args=[project.pk])

    #delete_project_dump task should have been launched
    with mock.patch('taiga.export_import.tasks.delete_project_dump') as delete_project_dump_mock:
        response = client.get(url, content_type="application/json")
        assert response.status_code == 202
        response_data = response.data
        assert "export_id" in response_data

        args = (project.id, project.slug, response_data["export_id"], None)
        kwargs = {"countdown": settings.EXPORTS_TTL}
        delete_project_dump_mock.apply_async.assert_called_once_with(args, **kwargs)


def test_valid_project_export_with_celery_enabled_and_gzip(client, settings):
    settings.CELERY_ENABLED = True

    user = f.UserFactory.create()
    project = f.ProjectFactory.create(owner=user)
    f.MembershipFactory(project=project, user=user, is_admin=True)
    client.login(user)

    url = reverse("exporter-detail", args=[project.pk])

    #delete_project_dump task should have been launched
    with mock.patch('taiga.export_import.tasks.delete_project_dump') as delete_project_dump_mock:
        response = client.get(url+"?dump_format=gzip", content_type="application/json")
        assert response.status_code == 202
        response_data = response.data
        assert "export_id" in response_data

        args = (project.id, project.slug, response_data["export_id"], "gzip")
        kwargs = {"countdown": settings.EXPORTS_TTL}
        delete_project_dump_mock.apply_async.assert_called_once_with(args, **kwargs)


def test_valid_project_with_throttling(client, settings):
    settings.CELERY_ENABLED = False
    settings.REST_FRAMEWORK["DEFAULT_THROTTLE_RATES"]["import-dump-mode"] = "1/minute"

    user = f.UserFactory.create()
    project = f.ProjectFactory.create(owner=user)
    f.MembershipFactory(project=project, user=user, is_admin=True)
    client.login(user)

    url = reverse("exporter-detail", args=[project.pk])

    response = client.get(url, content_type="application/json")
    assert response.status_code == 200
    response = client.get(url, content_type="application/json")
    assert response.status_code == 429