110 lines
2.9 KiB
Go
110 lines
2.9 KiB
Go
package xiaohongshu
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/require"
|
||
"github.com/xpzouying/xiaohongshu-mcp/browser"
|
||
)
|
||
|
||
func TestSearch(t *testing.T) {
|
||
|
||
t.Skip("SKIP: 测试发布")
|
||
|
||
b := browser.NewBrowser(false)
|
||
defer b.Close()
|
||
|
||
page := b.NewPage()
|
||
defer page.Close()
|
||
|
||
action := NewSearchAction(page)
|
||
|
||
feeds, err := action.Search(context.Background(), "Kimi")
|
||
require.NoError(t, err)
|
||
require.NotEmpty(t, feeds, "feeds should not be empty")
|
||
|
||
fmt.Printf("成功获取到 %d 个 Feed\n", len(feeds))
|
||
|
||
for _, feed := range feeds {
|
||
fmt.Printf("Feed ID: %s\n", feed.ID)
|
||
fmt.Printf("Feed Title: %s\n", feed.NoteCard.DisplayTitle)
|
||
}
|
||
}
|
||
|
||
func TestSearchWithFilters(t *testing.T) {
|
||
|
||
t.Skip("SKIP: 测试筛选功能")
|
||
|
||
b := browser.NewBrowser(false)
|
||
defer b.Close()
|
||
|
||
page := b.NewPage()
|
||
defer page.Close()
|
||
|
||
action := NewSearchAction(page)
|
||
|
||
// 方式1：直接使用索引
|
||
filters1 := []FilterOption{
|
||
{FiltersIndex: 2, TagsIndex: 3, Text: "图文"}, // 笔记类型 -> 图文
|
||
{FiltersIndex: 3, TagsIndex: 2, Text: "一天内"}, // 发布时间 -> 一天内
|
||
}
|
||
|
||
feeds1, err := action.Search(context.Background(), "dn432", filters1...)
|
||
require.NoError(t, err)
|
||
require.NotEmpty(t, feeds1, "feeds should not be empty")
|
||
|
||
fmt.Printf("方式1 - 成功获取到 %d 个筛选后的 Feed\n", len(feeds1))
|
||
|
||
// 方式2：使用便利函数
|
||
filter2, err := NoteType("图文")
|
||
require.NoError(t, err)
|
||
|
||
filter3, err := TimeRange("一天内")
|
||
require.NoError(t, err)
|
||
|
||
filters2 := []FilterOption{filter2, filter3}
|
||
feeds2, err := action.Search(context.Background(), "dn432", filters2...)
|
||
require.NoError(t, err)
|
||
require.NotEmpty(t, feeds2, "feeds should not be empty")
|
||
|
||
fmt.Printf("方式2 - 成功获取到 %d 个筛选后的 Feed\n", len(feeds2))
|
||
|
||
for _, feed := range feeds2 {
|
||
fmt.Printf("Feed ID: %s\n", feed.ID)
|
||
fmt.Printf("Feed Title: %s\n", feed.NoteCard.DisplayTitle)
|
||
}
|
||
}
|
||
|
||
func TestFilterValidation(t *testing.T) {
|
||
// 测试有效的筛选选项
|
||
validFilter := FilterOption{FiltersIndex: 2, TagsIndex: 3, Text: "图文"}
|
||
err := validateFilterOption(validFilter)
|
||
require.NoError(t, err)
|
||
|
||
// 测试无效的筛选组索引
|
||
invalidFilterGroup := FilterOption{FiltersIndex: 6, TagsIndex: 1, Text: "无效"}
|
||
err = validateFilterOption(invalidFilterGroup)
|
||
require.Error(t, err)
|
||
require.Contains(t, err.Error(), "无效的筛选组索引")
|
||
|
||
// 测试无效的标签索引
|
||
invalidTagIndex := FilterOption{FiltersIndex: 2, TagsIndex: 5, Text: "无效"}
|
||
err = validateFilterOption(invalidTagIndex)
|
||
require.Error(t, err)
|
||
require.Contains(t, err.Error(), "标签索引 5 超出范围")
|
||
|
||
// 测试便利函数
|
||
filter, err := NoteType("图文")
|
||
require.NoError(t, err)
|
||
require.Equal(t, 2, filter.FiltersIndex)
|
||
require.Equal(t, 3, filter.TagsIndex)
|
||
require.Equal(t, "图文", filter.Text)
|
||
|
||
// 测试不存在的文本
|
||
_, err = NoteType("不存在的类型")
|
||
require.Error(t, err)
|
||
require.Contains(t, err.Error(), "未找到文本")
|
||
}
|