wehub-resource-sync
534bb94eea
Test Migrations / Migrations (SQLite) (push) Has been cancelled
Build Dev Image / build-dev-image (push) Has been cancelled
Check i18n Keys / Check i18n Key Consistency (push) Has been cancelled
Lint / Ruff Lint & Format (push) Has been cancelled
Lint / Frontend Lint (push) Has been cancelled
Test Migrations / Migrations (PostgreSQL) (push) Has been cancelled
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, Mock
|
|
|
|
import pytest
|
|
|
|
from langbot.pkg.api.http.service.apikey import ApiKeyService
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize('api_key', [None, 123, b'lbk_bytes', '', 'plain_key', ' LBK_bad', 'sk-lbk_fake'])
|
|
async def test_verify_api_key_rejects_non_lbk_keys_without_db_query(api_key):
|
|
persistence_mgr = SimpleNamespace(execute_async=AsyncMock())
|
|
instance_config = SimpleNamespace(data={'api': {'global_api_key': ''}})
|
|
service = ApiKeyService(SimpleNamespace(persistence_mgr=persistence_mgr, instance_config=instance_config))
|
|
|
|
result = await service.verify_api_key(api_key)
|
|
|
|
assert result is False
|
|
persistence_mgr.execute_async.assert_not_awaited()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
('db_row', 'expected'),
|
|
[
|
|
(object(), True),
|
|
(None, False),
|
|
],
|
|
)
|
|
async def test_verify_api_key_keeps_db_validation_for_lbk_keys(db_row, expected):
|
|
query_result = Mock()
|
|
query_result.first.return_value = db_row
|
|
persistence_mgr = SimpleNamespace(execute_async=AsyncMock(return_value=query_result))
|
|
instance_config = SimpleNamespace(data={'api': {'global_api_key': ''}})
|
|
service = ApiKeyService(SimpleNamespace(persistence_mgr=persistence_mgr, instance_config=instance_config))
|
|
|
|
result = await service.verify_api_key('lbk_valid_format')
|
|
|
|
assert result is expected
|
|
persistence_mgr.execute_async.assert_awaited_once()
|