bird/tests/runtime-query-ids.test.ts

import { randomUUID } from 'node:crypto';
import { mkdir, readFile, writeFile } from 'node:fs/promises';
import os from 'node:os';
import path from 'node:path';
import { afterEach, describe, expect, it, vi } from 'vitest';
import { createRuntimeQueryIdStore } from '../src/lib/runtime-query-ids.js';

describe('runtime-query-ids', () => {
  afterEach(() => {
    vi.unstubAllGlobals();
    delete process.env.BIRD_QUERY_IDS_CACHE;
  });

  it('refreshes IDs by scanning discovered bundles and persists cache', async () => {
    const cacheDir = path.join(os.tmpdir(), `bird-test-${randomUUID()}`);
    await mkdir(cacheDir, { recursive: true });
    const cachePath = path.join(cacheDir, 'query-ids.json');

    const html = `
      <html>
        <script src="https://abs.twimg.com/responsive-web/client-web/main.test.js"></script>
      </html>
    `;
    const js = [
      `e.exports={queryId:"AAA",operationName:"CreateTweet"}`,
      `e.exports={queryId:"BBB",operationName:"CreateRetweet"}`,
      `e.exports={queryId:"CCC",operationName:"FavoriteTweet"}`,
      `e.exports={queryId:"DDD",operationName:"TweetDetail"}`,
      `e.exports={queryId:"EEE",operationName:"SearchTimeline"}`,
      `e.exports={queryId:"FFF",operationName:"UserArticlesTweets"}`,
    ].join('\n');

    const fetchMock = vi.fn(async (url: string | URL) => {
      const asString = String(url);
      if (asString.startsWith('https://x.com/')) {
        return new Response(html, { status: 200 });
      }
      if (asString.startsWith('https://abs.twimg.com/')) {
        return new Response(js, { status: 200 });
      }
      return new Response('not found', { status: 404 });
    });

    vi.stubGlobal('fetch', fetchMock);

    const store = createRuntimeQueryIdStore({
      cachePath,
      ttlMs: 1000,
      fetchImpl: fetchMock as unknown as typeof fetch,
    });
    const info = await store.refresh(
      ['CreateTweet', 'CreateRetweet', 'FavoriteTweet', 'TweetDetail', 'SearchTimeline', 'UserArticlesTweets'],
      { force: true },
    );

    expect(info?.snapshot.ids.CreateTweet).toBe('AAA');
    expect(info?.snapshot.ids.TweetDetail).toBe('DDD');
    expect(info?.cachePath).toBe(cachePath);

    const raw = await readFile(cachePath, 'utf8');
    const parsed = JSON.parse(raw) as { ids?: Record<string, string> };
    expect(parsed.ids?.SearchTimeline).toBe('EEE');
  });

  it('uses env cache path override and returns fresh snapshot without refreshing', async () => {
    const cacheDir = path.join(os.tmpdir(), `bird-test-${randomUUID()}`);
    await mkdir(cacheDir, { recursive: true });
    const cachePath = path.join(cacheDir, 'query-ids-cache.json');
    process.env.BIRD_QUERY_IDS_CACHE = cachePath;

    const snapshot = {
      fetchedAt: new Date().toISOString(),
      ttlMs: 60_000,
      ids: { CreateTweet: 'AAA' },
      discovery: { pages: ['https://x.com/'], bundles: ['main.js'] },
    };
    await writeFile(cachePath, JSON.stringify(snapshot), 'utf8');

    const fetchMock = vi.fn();
    const store = createRuntimeQueryIdStore({ fetchImpl: fetchMock as unknown as typeof fetch });
    const info = await store.refresh(['CreateTweet']);

    expect(store.cachePath).toBe(cachePath);
    expect(info?.snapshot.ids.CreateTweet).toBe('AAA');
    expect(fetchMock).not.toHaveBeenCalled();

    store.clearMemory();
  });

  it('returns current snapshot when refresh finds no matching operations', async () => {
    const cacheDir = path.join(os.tmpdir(), `bird-test-${randomUUID()}`);
    await mkdir(cacheDir, { recursive: true });
    const cachePath = path.join(cacheDir, 'query-ids.json');

    const snapshot = {
      fetchedAt: new Date(0).toISOString(),
      ttlMs: 1,
      ids: { CreateTweet: 'AAA' },
      discovery: { pages: ['https://x.com/'], bundles: ['main.test.js'] },
    };
    await writeFile(cachePath, JSON.stringify(snapshot), 'utf8');

    const html = `<script src="https://abs.twimg.com/responsive-web/client-web/main.test.js"></script>`;
    const js = 'console.log("no ops here")';

    const fetchMock = vi.fn(async (url: string | URL) => {
      const asString = String(url);
      if (asString.startsWith('https://x.com/')) {
        return new Response(html, { status: 200 });
      }
      if (asString.startsWith('https://abs.twimg.com/')) {
        return new Response(js, { status: 200 });
      }
      return new Response('not found', { status: 404 });
    });

    const store = createRuntimeQueryIdStore({
      cachePath,
      ttlMs: 1,
      fetchImpl: fetchMock as unknown as typeof fetch,
    });

    const info = await store.refresh(['CreateTweet']);
    expect(info?.snapshot.ids.CreateTweet).toBe('AAA');
  });

  it('throws when no bundles can be discovered', async () => {
    const cacheDir = path.join(os.tmpdir(), `bird-test-${randomUUID()}`);
    await mkdir(cacheDir, { recursive: true });
    const cachePath = path.join(cacheDir, 'query-ids.json');

    const fetchMock = vi.fn(async () => new Response('nope', { status: 500 }));

    const store = createRuntimeQueryIdStore({
      cachePath,
      fetchImpl: fetchMock as unknown as typeof fetch,
    });

    await expect(store.refresh(['CreateTweet'], { force: true })).rejects.toThrow(
      'No client bundles discovered; x.com layout may have changed.',
    );
  });
});