# -*- coding: utf-8 -*-
#
# Copyright (C) 2020 CERN.
# Copyright (C) 2020 Northwestern University.
#
# Invenio-Drafts-Resources is free software; you can redistribute it and/or
# modify it under the terms of the MIT License; see LICENSE file for more
# details.

"""Invenio Drafts Resources module to create REST APIs"""

import json
import time

import pytest
from mock_module.api import Draft, Record
from sqlalchemy.orm.exc import NoResultFound

from invenio_drafts_resources.resources import DraftActionResource, \
    DraftActionResourceConfig, DraftResource, DraftResourceConfig, \
    DraftVersionResource, DraftVersionResourceConfig

HEADERS = {"content-type": "application/json", "accept": "application/json"}


def _assert_single_item_response(response):
    """Assert the fields present on a single item response."""
    response_fields = response.json.keys()
    fields_to_check = ['id', 'conceptid', 'metadata',
                       'created', 'updated', 'links']

    for field in fields_to_check:
        assert field in response_fields

#
# Operations tests
#


def test_create_draft(client, input_data, es_clear):
    """Test draft creation of a non-existing record."""
    response = client.post(
        "/mocks", data=json.dumps(input_data), headers=HEADERS)

    assert response.status_code == 201
    _assert_single_item_response(response)


def test_read_draft(client, input_data, es_clear):
    response = client.post(
        "/mocks", data=json.dumps(input_data), headers=HEADERS)

    assert response.status_code == 201

    recid = response.json['id']
    response = client.get(
        "/mocks/{}/draft".format(recid), headers=HEADERS)

    assert response.status_code == 200

    _assert_single_item_response(response)


def test_update_draft(client, input_data, es_clear):
    response = client.post(
        "/mocks", data=json.dumps(input_data), headers=HEADERS)

    assert response.status_code == 201
    assert response.json['metadata']['title'] == \
        input_data['metadata']['title']

    recid = response.json['id']

    orig_title = input_data['metadata']['title']
    edited_title = "Edited title"
    input_data['metadata']['title'] = edited_title

    # Update draft content
    update_response = client.put(
        "/mocks/{}/draft".format(recid),
        data=json.dumps(input_data),
        headers=HEADERS
    )

    assert update_response.status_code == 200
    assert update_response.json["metadata"]['title'] == edited_title
    assert update_response.json["id"] == recid

    # Check the updates where saved
    update_response = client.get(
        "/mocks/{}/draft".format(recid), headers=HEADERS)

    assert update_response.status_code == 200
    assert update_response.json["metadata"]['title'] == edited_title
    assert update_response.json["id"] == recid


def test_delete_draft(client, input_data, es_clear):
    response = client.post(
        "/mocks", data=json.dumps(input_data), headers=HEADERS)

    assert response.status_code == 201

    recid = response.json['id']

    update_response = client.delete(
        "/mocks/{}/draft".format(recid), headers=HEADERS)

    assert update_response.status_code == 204

    # Check draft deletion
    update_response = client.get(
        "/mocks/{}/draft".format(recid), headers=HEADERS)
    assert update_response.status_code == 404


def _create_and_publish(client, input_data):
    """Create a draft and publish it."""
    # Create the draft
    response = client.post(
        "/mocks", data=json.dumps(input_data), headers=HEADERS)

    assert response.status_code == 201
    recid = response.json['id']

    # Publish it
    response = client.post(
        "/mocks/{}/draft/actions/publish".format(recid), headers=HEADERS)

    assert response.status_code == 202
    _assert_single_item_response(response)

    return recid


def test_publish_draft(client, input_data, es_clear):
    """Test draft publication of a non-existing record.

    It has to first create said draft.
    """
    recid = _create_and_publish(client, input_data)

    # Check draft does not exists anymore
    response = client.get(
        "/mocks/{}/draft".format(recid), headers=HEADERS)

    assert response.status_code == 404

    # Check record exists
    response = client.get("/mocks/{}".format(recid), headers=HEADERS)

    assert response.status_code == 200

    _assert_single_item_response(response)


def test_search_records_and_drafts(client, input_data, es_clear):
    """Tests the search over the records index.

    Note: The three use cases are set in the same test so there is the
          possibility of failure. Meaning that if search is not done
          correctly more than one record/draft will be returned.
    """
    # Create a draft
    response = client.post(
        "/mocks", data=json.dumps(input_data), headers=HEADERS)
    assert response.status_code == 201
    recid = response.json['id']

    Draft.index.refresh()

    response = client.get("/mocks?status=draft", headers=HEADERS)
    assert response.status_code == 200
    assert response.json['hits']['total'] == 1
    assert response.json['hits']['hits'][0]['id'] == recid

    # Create a record
    recid = _create_and_publish(client, input_data)
    Record.index.refresh()

    response = client.get("/mocks?status=published", headers=HEADERS)
    assert response.status_code == 200
    assert response.json['hits']['total'] == 1
    assert response.json['hits']['hits'][0]['id'] == recid

    # Default to record search
    response = client.get("/mocks", headers=HEADERS)

    assert response.status_code == 200
    assert response.json['hits']['total'] == 1
    assert response.json['hits']['hits'][0]['id'] == recid


def test_action_not_configured(client, es_clear):
    """Tests a non configured action call."""
    # NOTE: recid can be dummy since it won't reach pass the resource view
    response = client.post(
        "/mocks/1234-abcd/draft/actions/non-configured", headers=HEADERS
    )
    assert response.status_code == 404


def test_command_not_implemented(client, es_clear):
    """Tests a configured action without implemented function."""
    # NOTE: recid can be dummy since it won't reach pass the resource view
    response = client.post(
        "/mocks/1234-abcd/draft/actions/command", headers=HEADERS
    )
    assert response.status_code == 500


#
# Flow tests (Note that operations are tested above
# therefore these tests do not assert their output)
#

def test_create_publish_new_revision(client, input_data, es_clear):
    """Test draft creation of an existing record and publish it."""
    recid = _create_and_publish(client, input_data)

    # Create new draft of said record
    orig_title = input_data["metadata"]["title"]
    input_data["metadata"]["title"] = "Edited title"
    response = client.post(
        "/mocks/{}/draft".format(recid),
        headers=HEADERS
    )

    assert response.status_code == 201
    assert response.json['revision_id'] == 4
    _assert_single_item_response(response)

    # Update that new draft
    response = client.put(
        "/mocks/{}/draft".format(recid),
        data=json.dumps(input_data),
        headers=HEADERS
    )

    assert response.status_code == 200

    # Check the actual record was not modified
    response = client.get(
        "/mocks/{}".format(recid), headers=HEADERS)

    assert response.status_code == 200
    _assert_single_item_response(response)
    assert response.json['metadata']['title'] == orig_title

    # Publish it to check the increment in reversion
    response = client.post(
        "/mocks/{}/draft/actions/publish".format(recid), headers=HEADERS)

    assert response.status_code == 202
    _assert_single_item_response(response)

    assert response.json['id'] == recid
    assert response.json['revision_id'] == 2
    assert response.json['metadata']['title'] == \
        input_data["metadata"]["title"]

    # Check it was actually edited
    response = client.get(
        "/mocks/{}".format(recid), headers=HEADERS)

    assert response.json['metadata']['title'] == \
        input_data["metadata"]["title"]


def test_mutiple_edit(client, input_data, es_clear):
    """Test the revision_id when editing record multiple times.

    This tests the `edit` service method.
    """
    # Needs `app` context because of invenio_access/permissions.py#166
    recid = _create_and_publish(client, input_data)

    # Create new draft of said record
    response = client.post(
        "/mocks/{}/draft".format(recid), headers=HEADERS)

    assert response.status_code == 201
    assert response.json['revision_id'] == 4

    # Request a second edit. Get the same draft (revision_id)
    response = client.post(
        "/mocks/{}/draft".format(recid), headers=HEADERS)

    assert response.status_code == 201
    assert response.json['revision_id'] == 4

    # Publish it to check the increment in version_id
    response = client.post(
        "/mocks/{}/draft/actions/publish".format(recid), headers=HEADERS)

    assert response.status_code == 202

    # Edit again
    response = client.post(
        "/mocks/{}/draft".format(recid), headers=HEADERS)

    assert response.status_code == 201
    assert response.json['revision_id'] == 7


def test_create_publish_new_version(client, input_data):
    """Creates a new version of a record.

    Publishes the draft to obtain 2 versions of a record.
    """
    recid = _create_and_publish(client, input_data)

    # Create new draft of said record
    response = client.post(
        "/mocks/{}/versions".format(recid), headers=HEADERS)

    assert response.status_code == 201
    _assert_single_item_response(response)
    assert response.json['revision_id'] == 1
    recid_2 = response.json['id']

    # Publish it to check the increment in version
    response = client.post(
        "/mocks/{}/draft/actions/publish".format(recid_2), headers=HEADERS)

    assert response.status_code == 202
    _assert_single_item_response(response)

    assert response.json['id'] == recid_2 != recid
    assert response.json['revision_id'] == 1
