eigent/backend/tests/app/model/test_chat_auth_source.py
Tong Chen 6300702844
Some checks failed
CodeQL Advanced / Analyze (actions) (push) Has been cancelled
CodeQL Advanced / Analyze (javascript-typescript) (push) Has been cancelled
CodeQL Advanced / Analyze (python) (push) Has been cancelled
Pre-commit / pre-commit (push) Has been cancelled
Test / Run Web + Local Brain Smoke (push) Has been cancelled
Test / Run Frontend Guardrails (push) Has been cancelled
Test / Run Python Tests (push) Has been cancelled
release: v1.0.1 (#1710)
Co-authored-by: Douglas <douglas.ym.lai@gmail.com>
Co-authored-by: Douglas Lai <115660088+Douglasymlai@users.noreply.github.com>
Co-authored-by: Tao Sun <168447269+fengju0213@users.noreply.github.com>
Co-authored-by: Weijie Bai <happy.regina.bai@gmail.com>
2026-06-27 18:14:29 +08:00

34 lines
1.3 KiB
Python

# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
import pytest
from pydantic import ValidationError
from app.model.chat import Chat
def test_chat_defaults_auth_source_to_none(sample_chat_data):
"""Legacy api_key path: no marker, no behavior change."""
chat = Chat(**sample_chat_data)
assert chat.auth_source is None
def test_chat_accepts_codex_subscription_marker(sample_chat_data):
chat = Chat(**{**sample_chat_data, "auth_source": "codex_subscription"})
assert chat.auth_source == "codex_subscription"
def test_chat_rejects_unknown_auth_source(sample_chat_data):
with pytest.raises(ValidationError):
Chat(**{**sample_chat_data, "auth_source": "bogus_source"})