opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[autofix.ci] apply automated fixes

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
feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True)
with patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")), patch(
"controllers.console.auth.email_register.dify_config", SimpleNamespace(BILLING_ENABLED=True)
), patch(
"controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")
), patch(
"controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags
with (
patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")),
patch("controllers.console.auth.email_register.dify_config", SimpleNamespace(BILLING_ENABLED=True)),
patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
patch("controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags),
):
with app.test_request_context(
"/email-register/send-email",
@ -89,10 +88,10 @@ class TestEmailRegisterCheckApi:
mock_generate_token.return_value = (None, "new-token")
feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True)
with patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")), patch(
"controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")
), patch(
"controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags
with (
patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")),
patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
patch("controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags),
):
with app.test_request_context(
"/email-register/validity",
@ -143,10 +142,10 @@ class TestEmailRegisterResetApi:
mock_get_account.return_value = None
feature_flags = SimpleNamespace(enable_email_password_login=True, is_allow_register=True)
with patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")), patch(
"controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")
), patch(
"controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags
with (
patch("controllers.console.auth.email_register.db", SimpleNamespace(engine="engine")),
patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")),
patch("controllers.console.wraps.FeatureService.get_system_features", return_value=feature_flags),
):
with app.test_request_context(
"/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)
controller_features = SimpleNamespace(is_allow_register=True)
with patch("controllers.console.auth.forgot_password.db", SimpleNamespace(engine="engine")), patch(
"controllers.console.auth.forgot_password.FeatureService.get_system_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 (
patch("controllers.console.auth.forgot_password.db", SimpleNamespace(engine="engine")),
patch(
"controllers.console.auth.forgot_password.FeatureService.get_system_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(
"/forgot-password",
@ -89,8 +92,9 @@ class TestForgotPasswordCheckApi:
mock_generate_token.return_value = (None, "new-token")
wraps_features = SimpleNamespace(enable_email_password_login=True)
with patch("controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")), patch(
"controllers.console.wraps.FeatureService.get_system_features", return_value=wraps_features
with (
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(
"/forgot-password/validity",
@ -134,10 +138,10 @@ class TestForgotPasswordResetApi:
mock_session_cls.return_value.__enter__.return_value = mock_session
wraps_features = SimpleNamespace(enable_email_password_login=True)
with patch("controllers.console.auth.forgot_password.db", SimpleNamespace(engine="engine")), patch(
"controllers.console.wraps.dify_config", SimpleNamespace(EDITION="CLOUD")
), patch(
"controllers.console.wraps.FeatureService.get_system_features", return_value=wraps_features
with (
patch("controllers.console.auth.forgot_password.db", SimpleNamespace(engine="engine")),
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(
"/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)
with patch("controllers.console.setup.console_ns", mock_console_ns), patch(
"controllers.console.setup.get_setup_status", return_value=False
), patch(
"controllers.console.setup.TenantService.get_tenant_count", return_value=0
), patch(
"controllers.console.setup.get_init_validate_status", return_value=True
), patch(
"controllers.console.setup.extract_remote_ip", return_value="127.0.0.1"
), patch(
"controllers.console.setup.request", object()
), patch(
"controllers.console.setup.RegisterService.setup"
) as mock_register:
with (
patch("controllers.console.setup.console_ns", mock_console_ns),
patch("controllers.console.setup.get_setup_status", return_value=False),
patch("controllers.console.setup.TenantService.get_tenant_count", return_value=0),
patch("controllers.console.setup.get_init_validate_status", return_value=True),
patch("controllers.console.setup.extract_remote_ip", return_value="127.0.0.1"),
patch("controllers.console.setup.request", object()),
patch("controllers.console.setup.RegisterService.setup") as mock_register,
):
response, status = setup_api.post()
assert response == {"result": "success"}