mirror of
https://github.com/langgenius/dify.git
synced 2026-05-05 09:58:04 +08:00
refactor: use EnumText for MessageFeedback and MessageFile columns (#33738)
This commit is contained in:
@ -4,6 +4,7 @@ from unittest.mock import MagicMock, patch
|
||||
import pytest
|
||||
|
||||
from libs.infinite_scroll_pagination import InfiniteScrollPagination
|
||||
from models.enums import FeedbackFromSource, FeedbackRating
|
||||
from models.model import App, AppMode, EndUser, Message
|
||||
from services.errors.message import (
|
||||
FirstMessageNotExistsError,
|
||||
@ -820,14 +821,14 @@ class TestMessageServiceFeedback:
|
||||
app_model=app,
|
||||
message_id="msg-123",
|
||||
user=user,
|
||||
rating="like",
|
||||
rating=FeedbackRating.LIKE,
|
||||
content="Good answer",
|
||||
)
|
||||
|
||||
# Assert
|
||||
assert result.rating == "like"
|
||||
assert result.rating == FeedbackRating.LIKE
|
||||
assert result.content == "Good answer"
|
||||
assert result.from_source == "user"
|
||||
assert result.from_source == FeedbackFromSource.USER
|
||||
mock_db.session.add.assert_called_once()
|
||||
mock_db.session.commit.assert_called_once()
|
||||
|
||||
@ -852,13 +853,13 @@ class TestMessageServiceFeedback:
|
||||
app_model=app,
|
||||
message_id="msg-123",
|
||||
user=user,
|
||||
rating="dislike",
|
||||
rating=FeedbackRating.DISLIKE,
|
||||
content="Bad answer",
|
||||
)
|
||||
|
||||
# Assert
|
||||
assert result == feedback
|
||||
assert feedback.rating == "dislike"
|
||||
assert feedback.rating == FeedbackRating.DISLIKE
|
||||
assert feedback.content == "Bad answer"
|
||||
mock_db.session.commit.assert_called_once()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user