import pytest

from datadog_tools.connectors.swagger import Swagger


@pytest.fixture
def mock_docs_endpoint_exists(mocker):
    mocker.patch(
        "datadog_tools.connectors.swagger.requests.get",
        return_value=mocker.Mock(status_code=200),
    )


@pytest.fixture
def mock_docs_endpoint_not_exists(mocker):
    mocker.patch(
        "datadog_tools.connectors.swagger.requests.get",
        return_value=mocker.Mock(status_code=404),
    )


def test_spec_exists(mock_docs_endpoint_exists):
    swagger = Swagger(base_url="https://swagger.example.com")
    assert swagger.spec_exists("test-service") is True


def test_spec_not_exists(mock_docs_endpoint_not_exists):
    swagger = Swagger(base_url="https://swagger.example.com")
    assert swagger.spec_exists("non-existent-service") is False


def test_spec_url():
    swagger = Swagger(base_url="https://swagger.example.com")
    url = swagger.spec_url("test-service")
    assert url == "https://swagger.example.com/service/test-service"