import pytest
from fatcat_openapi_client import *
from fixtures import *


def test_multiple_edits_same_group(api):

    c1 = CreatorEntity(display_name="test updates")

    # create
    eg = quick_eg(api)
    c1 = api.get_creator(api.create_creator(eg.editgroup_id, c1).ident)
    api.accept_editgroup(eg.editgroup_id)

    # try multiple edits in the same group
    eg = quick_eg(api)
    c2 = CreatorEntity(display_name="left")
    c3 = CreatorEntity(display_name="right")
    edit = api.update_creator(eg.editgroup_id, c1.ident, c2)
    # should fail with existing
    with pytest.raises(fatcat_openapi_client.rest.ApiException):
        api.update_creator(eg.editgroup_id, c1.ident, c3)
    # ... but succeed after deleting
    api.delete_creator_edit(edit.editgroup_id, edit.edit_id)
    api.update_creator(eg.editgroup_id, c1.ident, c3)
    api.accept_editgroup(eg.editgroup_id)
    res = api.get_creator(c1.ident)
    assert res.display_name == "right"
    eg = api.get_editgroup(eg.editgroup_id)
    assert len(eg.edits.creators) == 1

    # cleanup
    eg = quick_eg(api)
    api.delete_creator(eg.editgroup_id, c1.ident)
    api.accept_editgroup(eg.editgroup_id)


def test_edit_after_accept(api):

    c1 = CreatorEntity(display_name="test updates")

    # create
    eg = quick_eg(api)
    c1 = api.get_creator(api.create_creator(eg.editgroup_id, c1).ident)
    api.accept_editgroup(eg.editgroup_id)

    # should be unable to create an edit on an old editgroup
    c2 = CreatorEntity(display_name="left")
    try:
        api.create_creator(eg.editgroup_id, c2)
        assert False
    except fatcat_openapi_client.rest.ApiException as e:
        assert 400 <= e.status < 500
        # TODO: need better message
        # assert "accepted" in e.body

    # cleanup
    eg = quick_eg(api)
    api.delete_creator(eg.editgroup_id, c1.ident)
    api.accept_editgroup(eg.editgroup_id)


def test_edit_deletion(api):

    c1 = CreatorEntity(display_name="test edit updates")

    # create
    eg = quick_eg(api)
    c1 = api.get_creator(api.create_creator(eg.editgroup_id, c1).ident)
    api.accept_editgroup(eg.editgroup_id)

    # try multiple edits in the same group
    c2 = CreatorEntity(display_name="update one")
    eg = quick_eg(api)
    eg = api.get_editgroup(eg.editgroup_id)
    assert len(eg.edits.creators) == 0
    edit = api.update_creator(eg.editgroup_id, c1.ident, c2)
    eg = api.get_editgroup(eg.editgroup_id)
    assert len(eg.edits.creators) == 1
    api.delete_creator_edit(edit.editgroup_id, edit.edit_id)
    eg = api.get_editgroup(eg.editgroup_id)
    assert len(eg.edits.creators) == 0

    api.accept_editgroup(eg.editgroup_id)
    res = api.get_creator(c1.ident)
    assert res.display_name == "test edit updates"
    eg = api.get_editgroup(eg.editgroup_id)
    assert len(eg.edits.creators) == 0

    # cleanup
    eg = quick_eg(api)
    api.delete_creator(eg.editgroup_id, c1.ident)
    api.accept_editgroup(eg.editgroup_id)


def test_empty_editgroup(api):
    eg = quick_eg(api)
    api.accept_editgroup(eg.editgroup_id)


def test_delete_accepted_edit(api):

    c1 = CreatorEntity(display_name="test edit updates")

    # create
    eg = quick_eg(api)
    edit = api.create_creator(eg.editgroup_id, c1)
    api.accept_editgroup(eg.editgroup_id)

    # try to delete
    try:
        api.delete_creator_edit(edit.editgroup_id, edit.edit_id)
        assert False
    except fatcat_openapi_client.rest.ApiException as e:
        assert 400 <= e.status < 500
        assert "accepted" in e.body


def test_wip_revision(api):

    c1 = CreatorEntity(display_name="test edit nugget")

    # fetch revision before accepting
    eg = quick_eg(api)
    c1 = api.get_creator(api.create_creator(eg.editgroup_id, c1).ident)
    rev = api.get_creator_revision(c1.revision)
    assert "nugget" in rev.display_name
    assert rev.state is None
    assert rev.ident is None
    assert rev.revision == c1.revision

    # fetch revision after accepting
    api.accept_editgroup(eg.editgroup_id)
    rev = api.get_creator_revision(c1.revision)
    assert "nugget" in rev.display_name
    assert rev.state is None
    assert rev.ident is None
    assert rev.revision == c1.revision


def test_edit_get_all(api_dummy_entities):

    ade = api_dummy_entities
    api = ade["api"]
    eg = api.get_editgroup(ade["editgroup"].editgroup_id)

    assert api.get_creator_edit(eg.edits.creators[0].edit_id)
    assert api.get_container_edit(eg.edits.containers[0].edit_id)
    assert api.get_file_edit(eg.edits.files[0].edit_id)
    assert api.get_fileset_edit(eg.edits.filesets[0].edit_id)
    assert api.get_webcapture_edit(eg.edits.webcaptures[0].edit_id)
    assert api.get_release_edit(eg.edits.releases[0].edit_id)
    assert api.get_work_edit(eg.edits.works[0].edit_id)


def test_edit_delete_all(api_dummy_entities):

    ade = api_dummy_entities
    api = ade["api"]
    eg = api.get_editgroup(ade["editgroup"].editgroup_id)

    api.delete_creator_edit(eg.editgroup_id, eg.edits.creators[0].edit_id)
    api.delete_container_edit(eg.editgroup_id, eg.edits.containers[0].edit_id)
    api.delete_file_edit(eg.editgroup_id, eg.edits.files[0].edit_id)
    api.delete_fileset_edit(eg.editgroup_id, eg.edits.filesets[0].edit_id)
    api.delete_webcapture_edit(eg.editgroup_id, eg.edits.webcaptures[0].edit_id)
    api.delete_release_edit(eg.editgroup_id, eg.edits.releases[0].edit_id)
    api.delete_work_edit(eg.editgroup_id, eg.edits.works[0].edit_id)

    # verify deletion
    eg = api.get_editgroup(ade["editgroup"].editgroup_id)
    assert len(eg.edits.creators) == 0
    assert len(eg.edits.containers) == 0
    assert len(eg.edits.files) == 0
    assert len(eg.edits.filesets) == 0
    assert len(eg.edits.webcaptures) == 0
    assert len(eg.edits.releases) == 0
    assert len(eg.edits.works) == 0