LiuMangMang/SmartUp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: fix trailing whitespaces and add symmetric compact_model_mapping assertions

Browse Source
This commit is contained in:
SmartUp Developer
2026-07-02 19:48:47 +08:00
parent c20c9cfd26
commit d07ab62772
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/test_sync_upstream_models.py
+6 -4
View File
@@ -82,8 +82,8 @@ def test_sync_upstream_models_original_json_endpoint(db_session, monkeypatch):
def list_accounts(self): def list_accounts(self):
return [ return [
{ {
"id": "1", "id": "1",
"name": "Account 1", "name": "Account 1",
"credentials": { "credentials": {
"base_url": "old", "base_url": "old",
"access_token": "secret_access_token", # Sensitive credential "access_token": "secret_access_token", # Sensitive credential
@@ -210,8 +210,8 @@ def test_sync_upstream_models_streaming_endpoint(db_session, monkeypatch):
def list_accounts(self): def list_accounts(self):
return [ return [
{ {
"id": "1", "id": "1",
"name": "Account 1", "name": "Account 1",
"credentials": { "credentials": {
"base_url": "old", "base_url": "old",
"access_token": "secret_access_token", # Sensitive credential "access_token": "secret_access_token", # Sensitive credential
@@ -281,6 +281,7 @@ def test_sync_upstream_models_streaming_endpoint(db_session, monkeypatch):
assert "access_token" not in payload1 assert "access_token" not in payload1
assert "api_key" not in payload1 assert "api_key" not in payload1
assert payload1["openai_capabilities"] == "some_capabilities" assert payload1["openai_capabilities"] == "some_capabilities"
assert payload1["compact_model_mapping"] == "some_mapping"
# Second update to write back model_mapping # Second update to write back model_mapping
payload2 = acc1_updates[1][1]["credentials"] payload2 = acc1_updates[1][1]["credentials"]
@@ -288,6 +289,7 @@ def test_sync_upstream_models_streaming_endpoint(db_session, monkeypatch):
assert "access_token" not in payload2 assert "access_token" not in payload2
assert "api_key" not in payload2 assert "api_key" not in payload2
assert payload2["openai_capabilities"] == "some_capabilities" assert payload2["openai_capabilities"] == "some_capabilities"
assert payload2["compact_model_mapping"] == "some_mapping"
assert payload2["model_mapping"] == {"gpt-3.5": "gpt-3.5", "gpt-4": "gpt-4"} assert payload2["model_mapping"] == {"gpt-3.5": "gpt-3.5", "gpt-4": "gpt-4"}
finally: finally: