opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-13 08:57:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

[autofix.ci] apply automated fixes

Browse Source
This commit is contained in:
autofix-ci[bot] 2025-12-22 06:00:51 +00:00 committed by GitHub
parent ae0a9e7986
commit c8d7a4e10f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 37 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
api/tests/unit_tests/controllers/console/auth/test_email_register.py
View File

@ -45,12 +45,11 @@ class TestEmailRegisterSendEmailApi:
mock_get_account.return_value = mock_account mock_get_account.return_value = mock_account
feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True) feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True)
with patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")), patch( with (
"controllers.console.auth.email_register.dify_config", SimpleNamespace(BILLING_ENABLED=True) patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")),
), patch( patch("controllers.console.auth.email_register.dify_config", SimpleNamespace(BILLING_ENABLED=True)),
"controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD") patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
), patch( patch("controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags),
"controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags
): ):
with app.test_request_context( with app.test_request_context(
"/email-register/send-email", "/email-register/send-email",
@ -89,10 +88,10 @@ class TestEmailRegisterCheckApi:
mock_generate_token.return_value = (None, "new-token") mock_generate_token.return_value = (None, "new-token")
feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True) feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True)
with patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")), patch( with (
"controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD") patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")),
), patch( patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
"controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags patch("controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags),
): ):
with app.test_request_context( with app.test_request_context(
"/email-register/validity", "/email-register/validity",
@ -143,10 +142,10 @@ class TestEmailRegisterResetApi:
mock_get_account.return_value = None mock_get_account.return_value = None
feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True) feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True)
with patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")), patch( with (
"controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD") patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")),
), patch( patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
"controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags patch("controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags),
): ):
with app.test_request_context( with app.test_request_context(
"/email-register", "/email-register",

26
api/tests/unit_tests/controllers/console/auth/test_forgot_password.py
View File

@ -42,11 +42,14 @@ class TestForgotPasswordSendEmailApi:
wraps_features = SimpleNamespace(enable_email_password_login=True, is_allow_register=True) wraps_features = SimpleNamespace(enable_email_password_login=True, is_allow_register=True)
controller_features = SimpleNamespace(is_allow_register=True) controller_features = SimpleNamespace(is_allow_register=True)
with patch("controllers.console.auth.forgot_password.db", SimpleNamespace(engine="engine")), patch( with (
"controllers.console.auth.forgot_password.FeatureService.get_system_features", patch("controllers.console.auth.forgot_password.db", SimpleNamespace(engine="engine")),
return_value=controller_features, patch(
), patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")), patch( "controllers.console.auth.forgot_password.FeatureService.get_system_features",
"controllers.console.wraps.FeatureService.get_system_features", return_value=wraps_features return_value=controller_features,
),
patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
patch("controllers.console.wraps.FeatureService.get_system_features", return_value=wraps_features),
): ):
with app.test_request_context( with app.test_request_context(
"/forgot-password", "/forgot-password",
@ -89,8 +92,9 @@ class TestForgotPasswordCheckApi:
mock_generate_token.return_value = (None, "new-token") mock_generate_token.return_value = (None, "new-token")
wraps_features = SimpleNamespace(enable_email_password_login=True) wraps_features = SimpleNamespace(enable_email_password_login=True)
with patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")), patch( with (
"controllers.console.wraps.FeatureService.get_system_features", return_value=wraps_features patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
patch("controllers.console.wraps.FeatureService.get_system_features", return_value=wraps_features),
): ):
with app.test_request_context( with app.test_request_context(
"/forgot-password/validity", "/forgot-password/validity",
@ -134,10 +138,10 @@ class TestForgotPasswordResetApi:
mock_session_cls.return_value.__enter__.return_value = mock_session mock_session_cls.return_value.__enter__.return_value = mock_session
wraps_features = SimpleNamespace(enable_email_password_login=True) wraps_features = SimpleNamespace(enable_email_password_login=True)
with patch("controllers.console.auth.forgot_password.db", SimpleNamespace(engine="engine")), patch( with (
"controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD") patch("controllers.console.auth.forgot_password.db", SimpleNamespace(engine="engine")),
), patch( patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
"controllers.console.wraps.FeatureService.get_system_features", return_value=wraps_features patch("controllers.console.wraps.FeatureService.get_system_features", return_value=wraps_features),
): ):
with app.test_request_context( with app.test_request_context(
"/forgot-password/resets", "/forgot-password/resets",

22
api/tests/unit_tests/controllers/console/test_setup.py
View File

@ -17,19 +17,15 @@ class TestSetupApi:
mock_console_ns = SimpleNamespace(payload=payload) mock_console_ns = SimpleNamespace(payload=payload)
with patch("controllers.console.setup.console_ns", mock_console_ns), patch( with (
"controllers.console.setup.get_setup_status", return_value=False patch("controllers.console.setup.console_ns", mock_console_ns),
), patch( patch("controllers.console.setup.get_setup_status", return_value=False),
"controllers.console.setup.TenantService.get_tenant_count", return_value=0 patch("controllers.console.setup.TenantService.get_tenant_count", return_value=0),
), patch( patch("controllers.console.setup.get_init_validate_status", return_value=True),
"controllers.console.setup.get_init_validate_status", return_value=True patch("controllers.console.setup.extract_remote_ip", return_value="127.0.0.1"),
), patch( patch("controllers.console.setup.request", object()),
"controllers.console.setup.extract_remote_ip", return_value="127.0.0.1" patch("controllers.console.setup.RegisterService.setup") as mock_register,
), patch( ):
"controllers.console.setup.request", object()
), patch(
"controllers.console.setup.RegisterService.setup"
) as mock_register:
response, status = setup_api.post() response, status = setup_api.post()
assert response == {"result": "success"} assert response == {"result": "success"}