import pytest
import requests_mock

from trailblazer.services.slurm.api_service.slurm_api_client.slurm_api_client import SlurmAPIClient


@pytest.fixture
def client() -> SlurmAPIClient:
    return SlurmAPIClient(
        base_url="https://slurm.example.com",
        access_token="access_token",
        user_name="user_name",
    )


@pytest.fixture
def mock_request():
    with requests_mock.Mocker() as mock:
        yield mock


@pytest.fixture
def jobs_response() -> dict:
    return {
        "jobs": [
            {
                "name": "job 1",
                "priority": "high",
            },
            {
                "name": "job 2",
                "priority": "low",
            },
        ]
    }
