Files
xiaohongshu-mcp/xiaohongshu/feeds_test.go
zy 108da414fd feat: 实现获取小红书首页 Feed 列表功能
- 添加 FeedsListAction 用于获取页面 window.__INITIAL_STATE__ 数据
- 定义完整的 Feed 数据结构,包含笔记、视频、用户信息等
- 实现 GetFeedsList 方法解析并返回 Feed 列表
- 添加单元测试验证数据获取和 JSON 序列化

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
2025-08-12 01:11:27 +08:00

87 lines
2.9 KiB
Go

package xiaohongshu
import (
"context"
"encoding/json"
"fmt"
"testing"
"github.com/stretchr/testify/require"
"github.com/xpzouying/xiaohongshu-mcp/browser"
)
func TestGetFeedsList(t *testing.T) {
t.Skip("SKIP: 测试发布")
_ = browser.Init(false)
defer browser.Close()
page := browser.NewPage()
defer page.Close()
// NewFeedsListAction 内部已经处理导航
action := NewFeedsListAction(page)
feeds, err := action.GetFeedsList(context.Background())
require.NoError(t, err)
require.NotEmpty(t, feeds, "feeds should not be empty")
fmt.Printf("成功获取到 %d 个 Feed\n", len(feeds))
// 验证 JSON 结构完整性
for i, feed := range feeds {
// 验证必填字段
require.NotEmpty(t, feed.ID, "Feed ID should not be empty")
require.NotEmpty(t, feed.ModelType, "ModelType should not be empty")
require.NotEmpty(t, feed.XsecToken, "XsecToken should not be empty")
require.NotEmpty(t, feed.TrackID, "TrackID should not be empty")
require.NotEmpty(t, feed.NoteCard.Type, "NoteCard Type should not be empty")
require.NotEmpty(t, feed.NoteCard.DisplayTitle, "DisplayTitle should not be empty")
require.NotEmpty(t, feed.NoteCard.User.UserID, "User ID should not be empty")
require.NotEmpty(t, feed.NoteCard.User.Nickname, "User nickname should not be empty")
// 如果是视频类型,检查视频信息
if feed.NoteCard.Type == "video" {
require.NotNil(t, feed.NoteCard.Video, "Video info should not be nil for video type")
if feed.NoteCard.Video != nil {
require.True(t, feed.NoteCard.Video.Capa.Duration > 0, "Video duration should be greater than 0")
}
}
// 只对第一个 Feed 进行完整 JSON 序列化检查
if i == 0 {
// 序列化为 JSON
jsonData, err := json.MarshalIndent(feed, "", " ")
require.NoError(t, err, "Failed to marshal feed")
fmt.Printf("\n第一个 Feed 的完整 JSON 结构:\n%s\n", string(jsonData))
// 反序列化检查
var checkFeed Feed
err = json.Unmarshal(jsonData, &checkFeed)
require.NoError(t, err, "Failed to unmarshal feed")
// 比较序列化前后是否一致
require.Equal(t, feed.ID, checkFeed.ID)
require.Equal(t, feed.ModelType, checkFeed.ModelType)
require.Equal(t, feed.NoteCard.Type, checkFeed.NoteCard.Type)
}
// 打印前3个 Feed 的信息
if i < 3 {
fmt.Printf("\nFeed %d 基本信息:\n", i+1)
fmt.Printf(" ID: %s\n", feed.ID)
fmt.Printf(" ModelType: %s\n", feed.ModelType)
fmt.Printf(" 标题: %s\n", feed.NoteCard.DisplayTitle)
fmt.Printf(" 类型: %s\n", feed.NoteCard.Type)
fmt.Printf(" 作者: %s (@%s)\n", feed.NoteCard.User.Nickname, feed.NoteCard.User.UserID)
fmt.Printf(" 点赞数: %s\n", feed.NoteCard.InteractInfo.LikedCount)
fmt.Printf(" 封面尺寸: %dx%d\n", feed.NoteCard.Cover.Width, feed.NoteCard.Cover.Height)
if feed.NoteCard.Type == "video" && feed.NoteCard.Video != nil {
fmt.Printf(" 视频时长: %d秒\n", feed.NoteCard.Video.Capa.Duration)
}
}
}
}