mirror of
https://github.com/langgenius/dify.git
synced 2026-11-10 21:35:45 +08:00
28dabd92ac
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: yyh <yuanyouhuilyz@gmail.com>
80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from controllers.common.fields import Parameters, Site
|
|
from core.app.app_config.common.parameters_mapping import get_parameters_from_feature_dict
|
|
from models.model import IconType
|
|
|
|
|
|
def test_parameters_model_round_trip() -> None:
|
|
parameters = get_parameters_from_feature_dict(features_dict={}, user_input_form=[])
|
|
|
|
model = Parameters.model_validate(parameters)
|
|
|
|
assert model.model_dump(mode="json") == parameters
|
|
|
|
|
|
@pytest.mark.parametrize("enabled", [False, True])
|
|
def test_parameters_serializes_show_thoughts(enabled: bool) -> None:
|
|
parameters = get_parameters_from_feature_dict(
|
|
features_dict={"show_thoughts": {"enabled": enabled}}, user_input_form=[]
|
|
)
|
|
|
|
assert Parameters.model_validate(parameters).model_dump(mode="json")["show_thoughts"] == {"enabled": enabled}
|
|
|
|
|
|
def test_parameters_defaults_show_thoughts_for_legacy_payload() -> None:
|
|
parameters = dict(get_parameters_from_feature_dict(features_dict={}, user_input_form=[]))
|
|
parameters.pop("show_thoughts")
|
|
|
|
assert Parameters.model_validate(parameters).model_dump(mode="json")["show_thoughts"] == {"enabled": False}
|
|
|
|
|
|
def test_site_icon_url_uses_signed_url_for_image_icon() -> None:
|
|
site = Site(
|
|
title="Example",
|
|
chat_color_theme=None,
|
|
chat_color_theme_inverted=False,
|
|
icon_type=IconType.IMAGE,
|
|
icon="file-id",
|
|
icon_background=None,
|
|
description=None,
|
|
copyright=None,
|
|
privacy_policy=None,
|
|
custom_disclaimer=None,
|
|
default_language="en-US",
|
|
show_workflow_steps=True,
|
|
use_icon_as_answer_icon=False,
|
|
)
|
|
|
|
with patch("controllers.common.fields.file_helpers.get_signed_file_url", return_value="signed") as mock_helper:
|
|
model = Site.model_validate(site)
|
|
|
|
assert model.icon_url == "signed"
|
|
mock_helper.assert_called_once_with("file-id")
|
|
|
|
|
|
def test_site_icon_url_is_none_for_non_image_icon() -> None:
|
|
site = Site(
|
|
title="Example",
|
|
chat_color_theme=None,
|
|
chat_color_theme_inverted=False,
|
|
icon_type=IconType.EMOJI,
|
|
icon="file-id",
|
|
icon_background=None,
|
|
description=None,
|
|
copyright=None,
|
|
privacy_policy=None,
|
|
custom_disclaimer=None,
|
|
default_language="en-US",
|
|
show_workflow_steps=True,
|
|
use_icon_as_answer_icon=False,
|
|
)
|
|
|
|
with patch("controllers.common.fields.file_helpers.get_signed_file_url") as mock_helper:
|
|
model = Site.model_validate(site)
|
|
|
|
assert model.icon_url is None
|
|
mock_helper.assert_not_called()
|