mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-06-04 00:54:17 +08:00
37903a1cae
* Integration test batch 1 * Mark test * Update cli/integration/test_integration_obbject_registry.py Co-authored-by: Henrique Joaquim <henriquecjoaquim@gmail.com> * example of command tests * Add more test cases --------- Co-authored-by: Henrique Joaquim <henriquecjoaquim@gmail.com> Co-authored-by: Danglewood <85772166+deeleeramone@users.noreply.github.com>
63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
"""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
|