src/test/java/com/example/g8backend/service/PostServiceRecommendTest.java

package com.example.g8backend.service;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.example.g8backend.entity.Post;
import com.example.g8backend.entity.PostView;
import com.example.g8backend.entity.UserTagPreference;
import com.example.g8backend.mapper.*;
import com.example.g8backend.service.impl.PostServiceImpl;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.transaction.annotation.Transactional;
import java.sql.Timestamp;
import java.time.Instant;
import java.util.*;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.*;
@ExtendWith(MockitoExtension.class)
@Transactional
public class PostServiceRecommendTest {
    @Mock
    private PostMapper postMapper;
    @Mock
    private PostViewMapper postViewMapper;
    @Mock
    private CommentMapper commentMapper;
    @Mock
    private UserTagPreferenceMapper userTagPreferenceMapper;
    @Mock
    private PostTagMapper postTagMapper;
    @InjectMocks
    private PostServiceImpl postService;
    private Post mockPost;
    private PostView mockPostView;
    @BeforeEach
    void setUp() {
        // 初始化测试数据
        mockPost = new Post()
                .setPostId(1L)
                .setUserId(100L)
                .setPostTitle("Test Post")
                .setViewCount(50)
                .setCreatedAt(Timestamp.from(Instant.now().minusSeconds(7200)))
                .setHotScore(5.0);
        mockPostView = new PostView()
                .setViewId(1L)
                .setUserId(200L)
                .setPostId(1L)
                .setViewTime(Timestamp.from(Instant.now()).toLocalDateTime());
    }
    @Test
    void testRecordViewHistory_NormalCase() {
        // 调用方法
        postService.recordViewHistory(100L, 1L);
        // 验证 postTagMapper 被调用
        verify(postTagMapper).findTagIdsByPostId(1L);
        verify(postViewMapper).insert(any(PostView.class));
        verify(postMapper).incrementViewCount(1L);
    }
    @Test
    public void testGetRecommendedPosts_ExcludesViewedPosts() {
        // 模拟用户已浏览的帖子ID
        Long userId = 200L;
        when(postViewMapper.findViewedPostIds(userId))
                .thenReturn(Arrays.asList(1L, 2L));
        // 模拟推荐结果（未浏览的帖子）
        Post recommendedPost = new Post().setPostId(3L).setHotScore(8.0);
        when(postMapper.selectPage(any(Page.class), any(QueryWrapper.class)))
                .thenReturn(new Page<Post>().setRecords(Collections.singletonList(recommendedPost)));
        // 调用推荐接口
        Page<Post> result = postService.getRecommendedPosts(1, 10, userId);
        // 验证结果
        assertEquals(1, result.getRecords().size(), "应返回1条推荐结果");
        assertEquals(3L, result.getRecords().get(0).getPostId(), "推荐结果应为未浏览的帖子ID 3");
        assertFalse(result.getRecords().stream().anyMatch(p -> p.getPostId() == 1L), "结果中不应包含已浏览的帖子ID 1");
    }
    @Test
    public void testGetRecommendedPosts_NoViewedPosts() {
        // 模拟用户未浏览任何帖子
        Long userId = 300L;
        when(postViewMapper.findViewedPostIds(userId))
                .thenReturn(Collections.emptyList());
        // 模拟推荐结果（所有帖子按热度排序）
        Post post1 = new Post().setPostId(1L).setHotScore(7.5);
        Post post2 = new Post().setPostId(2L).setHotScore(9.0);
        when(postMapper.selectPage(any(Page.class), any(QueryWrapper.class)))
                .thenReturn(new Page<Post>().setRecords(Arrays.asList(post2, post1)));
        // 调用推荐接口
        Page<Post> result = postService.getRecommendedPosts(1, 10, userId);
        // 验证结果
        assertEquals(2, result.getRecords().size(), "应返回所有帖子");
        assertEquals(2L, result.getRecords().get(0).getPostId(), "热度更高的帖子应排在前面");
    }
    @Test
    public void testCalculateHotScores_UpdatesHotScoreCorrectly() {
        // 设置存根
        when(postMapper.selectList(any(QueryWrapper.class)))
                .thenReturn(Collections.singletonList(mockPost));
        when(postMapper.selectLikeCount(anyLong())).thenReturn(30L);
        when(commentMapper.selectCountByPostId(anyLong())).thenReturn(20L);
        when(postMapper.batchUpdateHotScore(anyList())).thenReturn(1);
        // 执行并验证
        postService.calculateHotScores();
        double expectedScore = (Math.log(51) * 0.2 + 30 * 0.5 + 20 * 0.3) / Math.pow(4, 1.5);
        assertEquals(expectedScore, mockPost.getHotScore(), 0.01);
        verify(postMapper).batchUpdateHotScore(anyList());
    }
    //--------------------- 测试冷启动逻辑 ---------------------
    @Test
    public void testCreatePost_SetsInitialHotScore() {
        Post newPost = new Post().setPostId(4L).setPostTitle("New Post");
        postService.createPost(newPost);
        assertEquals(5.0, newPost.getHotScore(), "新帖子的初始热度应为5.0");
        assertNotNull(newPost.getCreatedAt(), "创建时间不应为空");
    }
    @Test
    public void testConcurrentViewCountUpdate() {
        // 设置存根
        doNothing().when(postMapper).incrementViewCount(anyLong());
        postService.recordViewHistory(100L, 1L);
        postService.recordViewHistory(200L, 1L);
        verify(postMapper, times(2)).incrementViewCount(1L);
        verify(postViewMapper, times(2)).insert(any(PostView.class));
    }
    @Test
    public void testGetRecommendedByTags_WithPreferredTags() {
        // 模拟用户偏好标签
        Long userId = 200L;
        UserTagPreference pref1 = new UserTagPreference().setTagId(100L).setWeight(2.0);
        UserTagPreference pref2 = new UserTagPreference().setTagId(200L).setWeight(1.5);
        when(userTagPreferenceMapper.selectByUserId(userId))
                .thenReturn(Arrays.asList(pref1, pref2));
        // 模拟标签关联的帖子ID
        List<Long> expectedPostIds = Arrays.asList(3L, 4L, 5L);
        when(postTagMapper.findPostIdsByTagIds(Arrays.asList(100L, 200L)))
                .thenReturn(expectedPostIds);
        // 使用 ArgumentCaptor 捕获 QueryWrapper 对象
        ArgumentCaptor<QueryWrapper<Post>> wrapperCaptor = ArgumentCaptor.forClass(QueryWrapper.class);
        Page<Post> mockPage = new Page<Post>().setRecords(Arrays.asList(
                new Post().setPostId(5L).setHotScore(9.0),
                new Post().setPostId(3L).setHotScore(8.0),
                new Post().setPostId(4L).setHotScore(7.5)
        ));
        when(postMapper.selectPage(any(Page.class), wrapperCaptor.capture()))
                .thenReturn(mockPage);
        // 调用标签推荐接口
        Page<Post> result = postService.getRecommendedByTags(1, 10, userId);
        // ---------- 验证查询条件 ----------
        QueryWrapper<Post> actualWrapper = wrapperCaptor.getValue();
        String sqlSegment = actualWrapper.getSqlSegment();
        // 验证 SQL 条件格式
        assertTrue(
                sqlSegment.matches(".*post_id\\s+IN\\s*\\(.*\\).*"),
                "应包含 post_id IN 条件，实际条件：" + sqlSegment
        );
        // 验证参数值（忽略顺序）
        Map<String, Object> params = actualWrapper.getParamNameValuePairs();
        List<Object> actualPostIds = new ArrayList<>(params.values());
        assertEquals(3, actualPostIds.size(), "IN 条件应包含3个参数");
        assertTrue(
                actualPostIds.containsAll(expectedPostIds),
                "参数应包含所有预期帖子ID，实际参数：" + actualPostIds
        );
        // ---------- 验证结果排序和内容 ----------
        assertEquals(3, result.getRecords().size(), "应返回3条结果");
        assertEquals(5L, result.getRecords().get(0).getPostId(), "热度最高的帖子应为ID 5");
        assertEquals(3L, result.getRecords().get(1).getPostId(), "热度次高的帖子应为ID 3");
        assertEquals(4L, result.getRecords().get(2).getPostId(), "热度最低的帖子应为ID 4");
    }
    @Test
    public void testGetRecommendedByTags_NoPreferredTags() {
        // 模拟用户无偏好标签
        Long userId = 300L;
        when(userTagPreferenceMapper.selectByUserId(userId))
                .thenReturn(Collections.emptyList());
        // 调用标签推荐接口
        Page<Post> result = postService.getRecommendedByTags(1, 10, userId);
        // 验证结果为空或兜底逻辑
        assertTrue(result.getRecords().isEmpty(), "无偏好标签时应返回空结果");
    }
    @Test
    public void testGetRecommendedByTags_WithNonExistingTags() {
        // 模拟用户偏好标签（但无关联帖子）
        Long userId = 400L;
        UserTagPreference pref = new UserTagPreference().setTagId(999L).setWeight(2.0);
        when(userTagPreferenceMapper.selectByUserId(userId))
                .thenReturn(Collections.singletonList(pref));
        when(postTagMapper.findPostIdsByTagIds(Collections.singletonList(999L)))
                .thenReturn(Collections.emptyList());
        // 调用标签推荐接口
        Page<Post> result = postService.getRecommendedByTags(1, 10, userId);
        // 验证结果为空
        assertNotNull(result, "分页结果不应为null");
        assertTrue(result.getRecords().isEmpty(), "无关联帖子时应返回空结果");
        // 验证postMapper.selectPage未被调用
        verify(postMapper, never()).selectPage(any(Page.class), any(QueryWrapper.class));
    }
}