mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-06 01:32:36 +00:00
60f3df793f
- New lib/crewai-core/ package: version, paths, constants, lock_store, user_data,
printer, telemetry. Pure leaf — depends only on appdirs/portalocker/rich/otel.
- crewai now depends on crewai-core; old crewai.utilities.{version,paths,printer,
lock_store} and the user-data block of events/listeners/tracing/utils.py become
one-shot DeprecationWarning shims that re-export from crewai_core.
- crewai-cli drops its hard dep on crewai and depends only on crewai-core. CLI
imports for telemetry/version/printer/constants now point at crewai_core.
- tools/main.py lazy-imports project_utils + get_user_id; the publish/login
subcommands print a friendly "requires crewai" error if it's missing.
- crewai-cli is now genuinely standalone: 'crewai --help', 'version', 'login',
'config', 'traces', 'create', 'template' all work without crewai installed.
- 351 CLI tests + 9 crewai-core smoke tests + crewai's full mypy (471 files) clean.
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
"""Tests for lock_store.
|
|
|
|
We verify our own logic: the _redis_available guard and which portalocker
|
|
backend is selected. We trust portalocker to handle actual locking mechanics.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
import crewai_core.lock_store as lock_store
|
|
from crewai_core.lock_store import lock
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def no_redis_url(monkeypatch):
|
|
monkeypatch.setattr(lock_store, "_REDIS_URL", None)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _redis_available
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_redis_not_available_without_url():
|
|
assert lock_store._redis_available() is False
|
|
|
|
|
|
def test_redis_not_available_when_package_missing(monkeypatch):
|
|
monkeypatch.setattr(lock_store, "_REDIS_URL", "redis://localhost:6379")
|
|
monkeypatch.setitem(sys.modules, "redis", None) # None → ImportError on import
|
|
assert lock_store._redis_available() is False
|
|
|
|
|
|
def test_redis_available_with_url_and_package(monkeypatch):
|
|
monkeypatch.setattr(lock_store, "_REDIS_URL", "redis://localhost:6379")
|
|
monkeypatch.setitem(sys.modules, "redis", mock.MagicMock())
|
|
assert lock_store._redis_available() is True
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# lock strategy selection
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_uses_file_lock_when_redis_unavailable():
|
|
with mock.patch("portalocker.Lock") as mock_lock:
|
|
with lock("file_test"):
|
|
pass
|
|
|
|
mock_lock.assert_called_once()
|
|
assert "crewai:" in mock_lock.call_args.args[0]
|
|
|
|
|
|
def test_uses_redis_lock_when_redis_available(monkeypatch):
|
|
fake_conn = mock.MagicMock()
|
|
monkeypatch.setattr(lock_store, "_redis_available", mock.Mock(return_value=True))
|
|
monkeypatch.setattr(lock_store, "_redis_connection", mock.Mock(return_value=fake_conn))
|
|
|
|
with mock.patch("portalocker.RedisLock") as mock_redis_lock:
|
|
with lock("redis_test"):
|
|
pass
|
|
|
|
mock_redis_lock.assert_called_once()
|
|
kwargs = mock_redis_lock.call_args.kwargs
|
|
assert kwargs["channel"].startswith("crewai:")
|
|
assert kwargs["connection"] is fake_conn
|