<reponame>rossumai/rossumctl
import re
from functools import partial
from traceback import print_tb

import pytest
from more_itertools import ilen

from rossum.workspace import create_command, list_command, delete_command, change_command
from tests.conftest import (
    TOKEN,
    match_uploaded_json,
    ORGANIZATIONS_URL,
    WORKSPACES_URL,
    DOCUMENTS_URL,
    QUEUES_URL,
    ANNOTATIONS_URL,
)

ORGANIZATION_ID = "1"
ORGANIZATION_URL = f"{ORGANIZATIONS_URL}/{ORGANIZATION_ID}"


@pytest.mark.usefixtures("mock_login_request", "mock_organization_urls", "rossum_credentials")
class TestCreate:
    def test_success(self, requests_mock, cli_runner):
        name = "TestName"
        new_id = "2"

        requests_mock.post(
            WORKSPACES_URL,
            additional_matcher=partial(
                match_uploaded_json, {"name": name, "organization": ORGANIZATION_URL}
            ),
            request_headers={"Authorization": f"Token {TOKEN}"},
            status_code=201,
            json={"id": new_id},
        )
        result = cli_runner.invoke(create_command, [name])
        assert not result.exit_code, print_tb(result.exc_info[2])
        assert f"{new_id}\n" == result.output


@pytest.mark.usefixtures("mock_login_request", "rossum_credentials")
class TestList:
    def test_success(self, requests_mock, cli_runner):
        workspace_id = 1
        name = "<EMAIL>"
        queue_id = 1

        queue_url = f"{QUEUES_URL}/{queue_id}"

        requests_mock.get(
            WORKSPACES_URL,
            json={
                "pagination": {"total": 1, "next": None},
                "results": [{"id": workspace_id, "queues": [queue_url], "name": name}],
            },
        )
        requests_mock.get(
            QUEUES_URL,
            json={
                "pagination": {"total": 1, "next": None},
                "results": [{"id": queue_id, "url": queue_url}],
            },
        )

        result = cli_runner.invoke(list_command)
        assert not result.exit_code, print_tb(result.exc_info[2])
        expected_table = f"""\
  id  name                queues
----  ----------------  --------
   {workspace_id}  {name}         {queue_id}
"""
        assert result.output == expected_table


@pytest.mark.usefixtures("mock_login_request", "rossum_credentials")
class TestDelete:
    def test_success(self, requests_mock, cli_runner):
        workspace_id = "1"
        queue_id = "1"
        workspace_url = f"{WORKSPACES_URL}/{workspace_id}"
        n_documents = 2

        requests_mock.get(
            workspace_url,
            request_headers={"Authorization": f"Token {TOKEN}"},
            json={"id": workspace_id, "url": workspace_url},
        )

        requests_mock.get(
            f"{QUEUES_URL}?workspace={workspace_id}",
            complete_qs=True,
            request_headers={"Authorization": f"Token {TOKEN}"},
            json={"pagination": {"next": None, "total": 1}, "results": [{"id": queue_id}]},
        )

        requests_mock.get(
            f"{ANNOTATIONS_URL}?queue={queue_id}&page_size=50&sideload=documents",
            complete_qs=True,
            request_headers={"Authorization": f"Token {TOKEN}"},
            json={
                "pagination": {"next": None, "total": 1},
                "results": [
                    {"id": i, "url": rf"{ANNOTATIONS_URL}/{i}", "document": rf"{DOCUMENTS_URL}/{i}"}
                    for i in range(n_documents)
                ],
                "documents": [
                    {"id": i, "url": rf"{DOCUMENTS_URL}/{i}"} for i in range(n_documents)
                ],
            },
        )

        requests_mock.delete(
            workspace_url, request_headers={"Authorization": f"Token {TOKEN}"}, status_code=204
        )

        requests_mock.delete(
            re.compile(rf"{DOCUMENTS_URL}/\d+"),
            request_headers={"Authorization": f"Token {TOKEN}"},
            status_code=204,
        )
        result = cli_runner.invoke(delete_command, [workspace_id, "--yes"])
        assert not result.exit_code, print_tb(result.exc_info[2])
        assert not result.output
        assert (
            ilen(r for r in requests_mock.request_history if r.method == "DELETE")
            == n_documents + 1
        )


@pytest.mark.usefixtures("mock_login_request", "rossum_credentials")
class TestChange:
    def test_success(self, requests_mock, cli_runner):
        name = "TestName"
        workspace_id = "1"

        requests_mock.patch(
            f"{WORKSPACES_URL}/{workspace_id}",
            additional_matcher=partial(match_uploaded_json, {"name": name}),
            request_headers={"Authorization": f"Token {TOKEN}"},
            status_code=200,
        )
        result = cli_runner.invoke(change_command, [workspace_id, "-n", name])
        assert not result.exit_code, print_tb(result.exc_info[2])
        assert not result.output

import re
from functools import partial
from traceback import print_tb

import pytest
from more_itertools import ilen

from rossum.workspace import create_command, list_command, delete_command, change_command
from tests.conftest import (
    TOKEN,
    match_uploaded_json,
    ORGANIZATIONS_URL,
    WORKSPACES_URL,
    DOCUMENTS_URL,
    QUEUES_URL,
    ANNOTATIONS_URL,
)

ORGANIZATION_ID = "1"
ORGANIZATION_URL = f"{ORGANIZATIONS_URL}/{ORGANIZATION_ID}"


@pytest.mark.usefixtures("mock_login_request", "mock_organization_urls", "rossum_credentials")
class TestCreate:
    def test_success(self, requests_mock, cli_runner):
        name = "TestName"
        new_id = "2"

        requests_mock.post(
            WORKSPACES_URL,
            additional_matcher=partial(
                match_uploaded_json, {"name": name, "organization": ORGANIZATION_URL}
            ),
            request_headers={"Authorization": f"Token {TOKEN}"},
            status_code=201,
            json={"id": new_id},
        )
        result = cli_runner.invoke(create_command, [name])
        assert not result.exit_code, print_tb(result.exc_info[2])
        assert f"{new_id}\n" == result.output


@pytest.mark.usefixtures("mock_login_request", "rossum_credentials")
class TestList:
    def test_success(self, requests_mock, cli_runner):
        workspace_id = 1
        name = "<EMAIL>"
        queue_id = 1

        queue_url = f"{QUEUES_URL}/{queue_id}"

        requests_mock.get(
            WORKSPACES_URL,
            json={
                "pagination": {"total": 1, "next": None},
                "results": [{"id": workspace_id, "queues": [queue_url], "name": name}],
            },
        )
        requests_mock.get(
            QUEUES_URL,
            json={
                "pagination": {"total": 1, "next": None},
                "results": [{"id": queue_id, "url": queue_url}],
            },
        )

        result = cli_runner.invoke(list_command)
        assert not result.exit_code, print_tb(result.exc_info[2])
        expected_table = f"""\
  id  name                queues
----  ----------------  --------
   {workspace_id}  {name}         {queue_id}
"""
        assert result.output == expected_table


@pytest.mark.usefixtures("mock_login_request", "rossum_credentials")
class TestDelete:
    def test_success(self, requests_mock, cli_runner):
        workspace_id = "1"
        queue_id = "1"
        workspace_url = f"{WORKSPACES_URL}/{workspace_id}"
        n_documents = 2

        requests_mock.get(
            workspace_url,
            request_headers={"Authorization": f"Token {TOKEN}"},
            json={"id": workspace_id, "url": workspace_url},
        )

        requests_mock.get(
            f"{QUEUES_URL}?workspace={workspace_id}",
            complete_qs=True,
            request_headers={"Authorization": f"Token {TOKEN}"},
            json={"pagination": {"next": None, "total": 1}, "results": [{"id": queue_id}]},
        )

        requests_mock.get(
            f"{ANNOTATIONS_URL}?queue={queue_id}&page_size=50&sideload=documents",
            complete_qs=True,
            request_headers={"Authorization": f"Token {TOKEN}"},
            json={
                "pagination": {"next": None, "total": 1},
                "results": [
                    {"id": i, "url": rf"{ANNOTATIONS_URL}/{i}", "document": rf"{DOCUMENTS_URL}/{i}"}
                    for i in range(n_documents)
                ],
                "documents": [
                    {"id": i, "url": rf"{DOCUMENTS_URL}/{i}"} for i in range(n_documents)
                ],
            },
        )

        requests_mock.delete(
            workspace_url, request_headers={"Authorization": f"Token {TOKEN}"}, status_code=204
        )

        requests_mock.delete(
            re.compile(rf"{DOCUMENTS_URL}/\d+"),
            request_headers={"Authorization": f"Token {TOKEN}"},
            status_code=204,
        )
        result = cli_runner.invoke(delete_command, [workspace_id, "--yes"])
        assert not result.exit_code, print_tb(result.exc_info[2])
        assert not result.output
        assert (
            ilen(r for r in requests_mock.request_history if r.method == "DELETE")
            == n_documents + 1
        )


@pytest.mark.usefixtures("mock_login_request", "rossum_credentials")
class TestChange:
    def test_success(self, requests_mock, cli_runner):
        name = "TestName"
        workspace_id = "1"

        requests_mock.patch(
            f"{WORKSPACES_URL}/{workspace_id}",
            additional_matcher=partial(match_uploaded_json, {"name": name}),
            request_headers={"Authorization": f"Token {TOKEN}"},
            status_code=200,
        )
        result = cli_runner.invoke(change_command, [workspace_id, "-n", name])
        assert not result.exit_code, print_tb(result.exc_info[2])
        assert not result.output
\   id  name                queues ----  ----------------  --------    {workspace_id}  {name}         {queue_id}