chengma/cmhub
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add client release force update flag

Browse Source
This commit is contained in:
QiuSW
2026-07-08 15:38:59 +08:00
parent 30d9df34eb
commit 6646c64a76
14 changed files with 130 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/api/tests.py
+52 -2
View File
@@ -9,6 +9,7 @@ from unittest.mock import patch
import requests
from cryptography.fernet import Fernet
from django.conf import settings
from django.contrib import admin
from django.contrib.auth import get_user_model
from django.core.cache import cache
from django.test import TestCase, override_settings
@@ -423,6 +424,7 @@ class ClientLatestReleaseApiTests(TestCase):
file_name: str = "",
sha256: str = "a" * 64,
release_notes: str = "首版 Windows 客户端",
force_update: bool = False,
) -> DownloadRelease:
return DownloadRelease.objects.create(
platform=platform,
@@ -432,6 +434,7 @@ class ClientLatestReleaseApiTests(TestCase):
file=file_name,
sha256=sha256,
release_notes=release_notes,
force_update=force_update,
)
def test_latest_release_is_public_without_api_key_and_returns_current_release(self):
@@ -449,9 +452,17 @@ class ClientLatestReleaseApiTests(TestCase):
self.assertEqual(response.data["platform"], "windows")
self.assertEqual(
set(response.data["release"]),
{"version", "download_url", "sha256", "release_notes", "published_at"},
{
"version",
"download_url",
"sha256",
"release_notes",
"force_update",
"published_at",
},
)
self.assertEqual(response.data["release"]["version"], "1.2.3")
self.assertFalse(response.data["release"]["force_update"])
self.assertEqual(
response.data["release"]["download_url"],
"https://download.example.com/cmhub-1.2.3.exe",
@@ -466,6 +477,29 @@ class ClientLatestReleaseApiTests(TestCase):
timezone.localtime(release.updated_at).isoformat(),
)
def test_latest_release_returns_force_update_true(self):
self.create_release(
version="0.1.1",
external_url="https://download.example.com/cmhub-0.1.1.zip",
release_notes="优化了ai模块的生图的功能",
force_update=True,
)
response = self.client.get(f"{self.url}?platform=windows")
self.assertEqual(response.status_code, 200)
self.assertEqual(response.data["platform"], "windows")
self.assertEqual(response.data["release"]["version"], "0.1.1")
self.assertEqual(
response.data["release"]["download_url"],
"https://download.example.com/cmhub-0.1.1.zip",
)
self.assertEqual(
response.data["release"]["release_notes"],
"优化了ai模块的生图的功能",
)
self.assertTrue(response.data["release"]["force_update"])
def test_latest_release_ignores_web_session_and_does_not_return_user_data(self):
user = get_user_model().objects.create_user(
username="release-session-user",
@@ -538,6 +572,7 @@ class ClientLatestReleaseApiTests(TestCase):
self.assertEqual(response.status_code, 200)
self.assertIsNone(response.data["release"])
self.assertEqual(response.data["message"], "暂未发布")
self.assertNotIn("force_update", response.data)
def test_latest_release_rejects_invalid_platform(self):
response = self.client.get(f"{self.url}?platform=android")
@@ -572,7 +607,14 @@ class ClientLatestReleaseApiTests(TestCase):
self.assertEqual(set(response.data), {"platform", "release"})
self.assertEqual(
set(response.data["release"]),
{"version", "download_url", "sha256", "release_notes", "published_at"},
{
"version",
"download_url",
"sha256",
"release_notes",
"force_update",
"published_at",
},
)
response_body = json.dumps(response.data, ensure_ascii=False)
for forbidden in (
@@ -590,6 +632,14 @@ class ClientLatestReleaseApiTests(TestCase):
):
self.assertNotIn(forbidden, response_body)
def test_download_release_admin_exposes_force_update(self):
registered_admin = admin.site._registry[DownloadRelease]
self.assertIn("force_update", registered_admin.list_display)
self.assertIn("force_update", registered_admin.list_filter)
version_fields = registered_admin.fieldsets[0][1]["fields"]
self.assertIn("force_update", version_fields)
@override_settings(
PAYMENT_CALLBACK_MODE="mock",