cli/integration/test_integration_hub_service.py

"""Integration tests for the hub_service module."""

from unittest.mock import create_autospec, patch

import pytest
import requests
from openbb_cli.controllers.hub_service import upload_routine
from openbb_core.app.model.hub.hub_session import HubSession

# pylint: disable=unused-argument, redefined-outer-name, unused-variable


@pytest.fixture
def auth_header():
    """Return a fake auth header."""
    return "Bearer fake_token"


@pytest.fixture
def hub_session_mock():
    """Return a mock HubSession."""
    mock = create_autospec(HubSession, instance=True)
    mock.username = "TestUser"
    return mock


# Fixture for routine data
@pytest.fixture
def routine_data():
    """Return a dictionary with routine data."""
    return {
        "name": "Test Routine",
        "description": "A test routine",
        "routine": "print('Hello World')",
        "override": False,
        "tags": "test",
        "public": True,
    }


@pytest.mark.integration
def test_upload_routine_timeout(auth_header, routine_data):
    """Test upload_routine with a timeout exception."""
    with patch(
        "requests.post", side_effect=requests.exceptions.Timeout
    ) as mocked_post:  # noqa: F841

        response = upload_routine(auth_header, **routine_data)

        assert response is None


@pytest.mark.integration
def test_upload_routine_connection_error(auth_header, routine_data):
    """Test upload_routine with a connection error."""
    with patch(
        "requests.post", side_effect=requests.exceptions.ConnectionError
    ) as mocked_post:  # noqa: F841

        response = upload_routine(auth_header, **routine_data)

        assert response is None