From c8d7a4e10fb1f54b5fafc8264bc85aee98dbdfe1 Mon Sep 17 00:00:00 2001 From: "autofix-ci[bot]" <114827586+autofix-ci[bot]@users.noreply.github.com> Date: Mon, 22 Dec 2025 06:00:51 +0000 Subject: [PATCH] [autofix.ci] apply automated fixes --- .../console/auth/test_email_register.py | 27 +++++++++---------- .../console/auth/test_forgot_password.py | 26 ++++++++++-------- .../controllers/console/test_setup.py | 22 +++++++-------- 3 files changed, 37 insertions(+), 38 deletions(-) diff --git a/api/tests/unit_tests/controllers/console/auth/test_email_register.py b/api/tests/unit_tests/controllers/console/auth/test_email_register.py index 06b4017e67..724c80f18c 100644 --- a/api/tests/unit_tests/controllers/console/auth/test_email_register.py +++ b/api/tests/unit_tests/controllers/console/auth/test_email_register.py @@ -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", diff --git a/api/tests/unit_tests/controllers/console/auth/test_forgot_password.py b/api/tests/unit_tests/controllers/console/auth/test_forgot_password.py index d512b8ad71..291efd842b 100644 --- a/api/tests/unit_tests/controllers/console/auth/test_forgot_password.py +++ b/api/tests/unit_tests/controllers/console/auth/test_forgot_password.py @@ -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", diff --git a/api/tests/unit_tests/controllers/console/test_setup.py b/api/tests/unit_tests/controllers/console/test_setup.py index 900bdbc8a9..e7882dcd2b 100644 --- a/api/tests/unit_tests/controllers/console/test_setup.py +++ b/api/tests/unit_tests/controllers/console/test_setup.py @@ -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"}