mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 17:30:23 +08:00
30 lines
797 B
Python
30 lines
797 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
|
||
|
|
from src.services.request.candidate import RequestCandidateService
|
||
|
|
|
||
|
|
|
||
|
|
def test_create_candidate_persists_snapshot_fields() -> None:
|
||
|
|
db = MagicMock()
|
||
|
|
|
||
|
|
candidate = RequestCandidateService.create_candidate(
|
||
|
|
db=db,
|
||
|
|
request_id="req-1",
|
||
|
|
candidate_index=0,
|
||
|
|
retry_index=1,
|
||
|
|
user_id="user-1",
|
||
|
|
api_key_id="key-1",
|
||
|
|
username="alice",
|
||
|
|
api_key_name="Primary Key",
|
||
|
|
provider_id="provider-1",
|
||
|
|
endpoint_id="endpoint-1",
|
||
|
|
key_id="pool-key-1",
|
||
|
|
status="available",
|
||
|
|
)
|
||
|
|
|
||
|
|
assert candidate.username == "alice"
|
||
|
|
assert candidate.api_key_name == "Primary Key"
|
||
|
|
db.add.assert_called_once_with(candidate)
|
||
|
|
db.flush.assert_called_once()
|