star-chart-search-enhancer/tests/spread-info.test.ts

import { describe, expect, test, vi } from "vitest";

import {
  buildSpreadInfoColumns,
  buildSpreadInfoUrl,
  createSpreadInfoClient,
  DEFAULT_SPREAD_INFO_CONFIGS,
  matchesSpreadThresholds,
  mapSpreadInfoResponse
} from "../src/content/market/spread-info";

describe("spread-info", () => {
  test("builds the spread info url with all request parameters", () => {
    expect(
      buildSpreadInfoUrl(
        "7361012802036695050",
        {
          flowType: 1,
          onlyAssign: true,
          range: 2,
          type: 2
        },
        "https://www.xingtu.cn"
      )
    ).toBe(
      "https://www.xingtu.cn/gw/api/data_sp/get_author_spread_info?o_author_id=7361012802036695050&platform_source=1&platform_channel=1&type=2&flow_type=1&only_assign=true&range=2"
    );
  });

  test("defines personal video ranges with fixed non-prefix parameters", () => {
    const personalConfigs = DEFAULT_SPREAD_INFO_CONFIGS.filter(
      (config) => config.type === 1
    );

    expect(personalConfigs).toEqual([
      {
        flowType: 0,
        onlyAssign: false,
        range: 2,
        type: 1
      },
      {
        flowType: 0,
        onlyAssign: false,
        range: 3,
        type: 1
      }
    ]);
    expect(buildSpreadInfoColumns(personalConfigs).slice(0, 2)).toEqual([
      "个人视频_近30天_完播率",
      "个人视频_近30天_播放量中位数"
    ]);
  });

  test("defines all xingtu video assign flow and range combinations", () => {
    const xingtuConfigs = DEFAULT_SPREAD_INFO_CONFIGS.filter(
      (config) => config.type === 2
    );

    expect(xingtuConfigs).toHaveLength(8);
    expect(xingtuConfigs).toContainEqual({
      flowType: 1,
      onlyAssign: true,
      range: 2,
      type: 2
    });
    expect(xingtuConfigs).toContainEqual({
      flowType: 0,
      onlyAssign: false,
      range: 3,
      type: 2
    });
    expect(buildSpreadInfoColumns([
      {
        flowType: 1,
        onlyAssign: true,
        range: 2,
        type: 2
      }
    ])).toEqual([
      "只看指派_排除营销流量_星图视频_近30天_完播率",
      "只看指派_排除营销流量_星图视频_近30天_播放量中位数",
      "只看指派_排除营销流量_星图视频_近30天_互动率",
      "只看指派_排除营销流量_星图视频_近30天_作品平均时长",
      "只看指派_排除营销流量_星图视频_近30天_作品平均评论数",
      "只看指派_排除营销流量_星图视频_近30天_作品平均点赞数",
      "只看指派_排除营销流量_星图视频_近30天_作品平均转发数"
    ]);
  });

  test("maps spread info response values into display values", () => {
    expect(
      mapSpreadInfoResponse({
        avg_duration: "5600",
        comment_avg: "7502",
        interact_rate: {
          value: 402
        },
        item_rate: {
          play_mid: {
            value: 10913233
          }
        },
        like_avg: "494458",
        play_over_rate: {
          value: 2824
        },
        share_avg: "188267"
      })
    ).toEqual({
      averageCommentCount: "7502",
      averageDuration: "56",
      averageLikeCount: "494458",
      averageShareCount: "188267",
      finishRate: "28.24%",
      interactionRate: "4.02%",
      playMedian: "10913233"
    });
  });

  test("loads each configured spread metric column for one author", async () => {
    const fetchImpl = vi.fn(async () => ({
      json: async () => ({
        avg_duration: "5600",
        comment_avg: "7502",
        interact_rate: {
          value: 402
        },
        like_avg: "494458",
        play_mid: "10913233",
        play_over_rate: {
          value: 2824
        },
        share_avg: "188267"
      }),
      ok: true
    }));
    const client = createSpreadInfoClient({
      configs: [
        {
          flowType: 0,
          onlyAssign: false,
          range: 2,
          type: 1
        }
      ],
      fetchImpl
    });

    await expect(
      client.loadAuthorSpreadMetrics("7361012802036695050")
    ).resolves.toEqual({
      "个人视频_近30天_互动率": "4.02%",
      "个人视频_近30天_作品平均点赞数": "494458",
      "个人视频_近30天_作品平均评论数": "7502",
      "个人视频_近30天_作品平均时长": "56",
      "个人视频_近30天_作品平均转发数": "188267",
      "个人视频_近30天_完播率": "28.24%",
      "个人视频_近30天_播放量中位数": "10913233"
    });
    expect(fetchImpl).toHaveBeenCalledWith(
      "https://www.xingtu.cn/gw/api/data_sp/get_author_spread_info?o_author_id=7361012802036695050&platform_source=1&platform_channel=1&type=1&flow_type=0&only_assign=false&range=2",
      expect.objectContaining({
        credentials: "include",
        method: "GET"
      })
    );
  });

  test("matches thresholds using display values and requires every filled threshold", () => {
    expect(
      matchesSpreadThresholds(
        {
          averageDuration: "56",
          finishRate: "28.24%",
          interactionRate: "4.02%",
          playMedian: "10913233"
        },
        {
          averageDuration: 50,
          finishRate: 28,
          interactionRate: 4,
          playMedian: 10000000
        }
      )
    ).toBe(true);

    expect(
      matchesSpreadThresholds(
        {
          averageDuration: "56",
          finishRate: "28.24%",
          interactionRate: "4.02%",
          playMedian: "10913233"
        },
        {
          averageDuration: 57
        }
      )
    ).toBe(false);

    expect(
      matchesSpreadThresholds(
        {
          finishRate: "28.24%"
        },
        {
          finishRate: 20,
          interactionRate: 1
        }
      )
    ).toBe(false);
  });
});