# Copyright (C) CVAT.ai Corporation
#
# SPDX-License-Identifier: MIT


import json
from types import SimpleNamespace

from cvat_sdk.api_client import ApiClient, Configuration
from cvat_sdk.api_client.api.tasks_api import TasksApi

from shared.utils.config import BASE_URL, make_api_client


def test_can_make_custom_request_with_call_api_method(admin_user):
    with make_api_client(admin_user) as api_client:
        _, response = api_client.call_api("/api/users/self", method="GET", _parse_response=False)

        assert json.loads(response.data)["username"] == admin_user


def test_can_make_custom_request_with_request_method(admin_user):
    with make_api_client(admin_user) as api_client:
        headers = api_client.get_common_headers()
        query_params = []
        api_client.update_params_for_auth(headers=headers, queries=query_params)
        assert not query_params

        response = api_client.request(
            "GET", BASE_URL + "/api/users/self", headers=headers, _parse_response=False
        )

        assert json.loads(response.data)["username"] == admin_user


def test_call_api_request_headers_override_defaults_and_skip_none(monkeypatch):
    api_client = ApiClient(Configuration(host=BASE_URL))
    api_client.set_default_header("X-Organization", "default-org")
    api_client.set_default_header("X-Remove-Me", "default-value")

    captured_headers = {}

    def fake_request(method, url, query_params=None, headers=None, **kwargs):
        captured_headers.update(headers)
        return SimpleNamespace(data=b"{}", status=200, msg="OK", headers={})

    monkeypatch.setattr(api_client, "request", fake_request)

    _, response = api_client.call_api(
        "/api/server/about",
        method="GET",
        header_params={
            "X-Organization": "request-org",
            "X-Remove-Me": None,
        },
        _parse_response=False,
    )

    assert response.status == 200
    assert captured_headers["X-Organization"] == "request-org"
    assert "X-Remove-Me" not in captured_headers


def test_endpoint_header_params_with_none_are_omitted(monkeypatch):
    api_client = ApiClient(Configuration(host=BASE_URL))
    api_client.set_default_header("X-Organization", "default-org")

    captured_request = {}

    def fake_request(method, url, query_params=None, headers=None, **kwargs):
        captured_request.update(
            method=method,
            url=url,
            query_params=query_params,
            headers=headers,
        )
        return SimpleNamespace(
            data=b'{"count":0,"next":null,"previous":null,"results":[]}',
            status=200,
            msg="OK",
            headers={},
        )

    monkeypatch.setattr(api_client, "request", fake_request)

    _, response = TasksApi(api_client).list_endpoint.call_with_http_info(
        org_id=123,
        x_organization=None,
        _parse_response=False,
    )

    assert response.status == 200
    assert "X-Organization" not in captured_request["headers"]
    assert dict(captured_request["query_params"]) == {"org_id": 123}