kairee
d971e718e9
Implement WBI video search with legacy fallback, classification scoring, candidate pool, permanent seen-aid dedupe, blacklist filtering, and batched Telegram notifications. Add write client for fav deal/sort, target-folder selection, insertion position calculation, read-back verification, fav_action locking, retry on failure, ignore and blacklist actions. Wire discovery, candidate cleanup, Telegram long polling, and new discover/candidates CLI commands into the scheduler. Add migration v3 for candidate duration and index stats.
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""新视频插入位置纯函数测试: uv run python tests/test_fav_position.py"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
sys.stdout.reconfigure(encoding="utf-8")
|
|
|
|
from reaction_board.fav_position import VideoMeta, plan_insertion # noqa: E402
|
|
|
|
|
|
def meta(up_mid, pubdate):
|
|
return VideoMeta(up_mid=up_mid, pubdate=pubdate)
|
|
|
|
|
|
def test_empty_and_single():
|
|
assert plan_insertion([], 9, {}, meta(100, 200)).moves == ()
|
|
assert plan_insertion([], 9, {}, meta(100, 200)).target_position == 1
|
|
plan = plan_insertion([1], 9, {1: meta(200, 50)}, meta(100, 200))
|
|
assert plan.target_position == 1
|
|
assert plan.moves == ((1, 9), (9, 1))
|
|
|
|
|
|
def test_same_up_group():
|
|
current = [1, 2, 3, 4]
|
|
metadata = {
|
|
1: meta(100, 100),
|
|
2: meta(100, 90),
|
|
3: meta(200, 80),
|
|
4: meta(200, 70),
|
|
}
|
|
# 同 UP 组在中间：插到组头
|
|
plan = plan_insertion(current, 9, metadata, meta(200, 95))
|
|
assert plan.target_position == 3
|
|
assert plan.moves == ((2, 9),)
|
|
|
|
# 同 UP 组在首位：两次移动
|
|
plan = plan_insertion(current, 9, metadata, meta(100, 200))
|
|
assert plan.target_position == 1
|
|
assert plan.moves == ((1, 9), (9, 1))
|
|
|
|
|
|
def test_new_group_by_earliest_pubdate():
|
|
current = [1, 2, 3, 4]
|
|
metadata = {
|
|
1: meta(100, 100),
|
|
2: meta(100, 90),
|
|
3: meta(200, 80),
|
|
4: meta(200, 70),
|
|
}
|
|
# 新 UP 的组键 85，应插到组键 70 的组之前
|
|
plan = plan_insertion(current, 9, metadata, meta(300, 85))
|
|
assert plan.target_position == 3
|
|
assert plan.moves == ((2, 9),)
|
|
|
|
# 比所有组都旧：追加到末尾
|
|
plan = plan_insertion(current, 9, metadata, meta(300, 60))
|
|
assert plan.target_position == 5
|
|
assert plan.moves == ((4, 9),)
|
|
|
|
|
|
def test_excludes_new_aid_if_present():
|
|
current = [1, 9, 2]
|
|
metadata = {1: meta(100, 100), 2: meta(100, 90)}
|
|
plan = plan_insertion(current, 9, metadata, meta(100, 95))
|
|
assert plan.target_position == 1
|
|
assert plan.moves == ((1, 9), (9, 1))
|
|
|
|
|
|
def main():
|
|
test_empty_and_single()
|
|
test_same_up_group()
|
|
test_new_group_by_earliest_pubdate()
|
|
test_excludes_new_aid_if_present()
|
|
print("OK: 新视频插入位置计算测试全部通过")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|