jackwener__opencli/clis/twitter/reply.test.ts

import * as fs from 'node:fs';
import * as os from 'node:os';
import * as path from 'node:path';

import { describe, expect, it, vi } from 'vitest';

import { getRegistry } from '../../registry.js';
import type { IPage } from '../../types.js';
import { __test__ } from './reply.js';

function createPageMock(evaluateResults: any[], overrides: Partial<IPage> = {}): IPage {
  const evaluate = vi.fn();
  for (const result of evaluateResults) {
    evaluate.mockResolvedValueOnce(result);
  }

  return {
    goto: vi.fn().mockResolvedValue(undefined),
    evaluate,
    snapshot: vi.fn().mockResolvedValue(undefined),
    click: vi.fn().mockResolvedValue(undefined),
    typeText: vi.fn().mockResolvedValue(undefined),
    pressKey: vi.fn().mockResolvedValue(undefined),
    scrollTo: vi.fn().mockResolvedValue(undefined),
    getFormState: vi.fn().mockResolvedValue({ forms: [], orphanFields: [] }),
    wait: vi.fn().mockResolvedValue(undefined),
    tabs: vi.fn().mockResolvedValue([]),
    selectTab: vi.fn().mockResolvedValue(undefined),
    networkRequests: vi.fn().mockResolvedValue([]),
    consoleMessages: vi.fn().mockResolvedValue([]),
    scroll: vi.fn().mockResolvedValue(undefined),
    autoScroll: vi.fn().mockResolvedValue(undefined),
    installInterceptor: vi.fn().mockResolvedValue(undefined),
    getInterceptedRequests: vi.fn().mockResolvedValue([]),
    getCookies: vi.fn().mockResolvedValue([]),
    screenshot: vi.fn().mockResolvedValue(''),
    waitForCapture: vi.fn().mockResolvedValue(undefined),
    ...overrides,
  };
}

describe('twitter reply command', () => {
  it('keeps the text-only reply flow working', async () => {
    const cmd = getRegistry().get('twitter/reply');
    expect(cmd?.func).toBeTypeOf('function');

    const page = createPageMock([
      { ok: true, message: 'Reply posted successfully.' },
    ]);

    const result = await cmd!.func!(page, {
      url: 'https://x.com/_kop6/status/2040254679301718161?s=20',
      text: 'text-only reply',
    });

    expect(page.goto).toHaveBeenCalledWith('https://x.com/_kop6/status/2040254679301718161?s=20');
    expect(page.wait).toHaveBeenCalledWith({ selector: '[data-testid="primaryColumn"]' });
    expect(result).toEqual([
      {
        status: 'success',
        message: 'Reply posted successfully.',
        text: 'text-only reply',
      },
    ]);
  });

  it('uploads a local image through the dedicated reply composer when --image is provided', async () => {
    const cmd = getRegistry().get('twitter/reply');
    expect(cmd?.func).toBeTypeOf('function');

    const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'opencli-twitter-reply-'));
    const imagePath = path.join(tempDir, 'qr.png');
    fs.writeFileSync(imagePath, Buffer.from([0x89, 0x50, 0x4e, 0x47]));

    const setFileInput = vi.fn().mockResolvedValue(undefined);
    const page = createPageMock([
      { ok: true, previewCount: 1 },
      { ok: true, message: 'Reply posted successfully.' },
    ], {
      setFileInput,
    });

    const result = await cmd!.func!(page, {
      url: 'https://x.com/_kop6/status/2040254679301718161?s=20',
      text: 'reply with image',
      image: imagePath,
    });

    expect(page.goto).toHaveBeenCalledWith('https://x.com/compose/post?in_reply_to=2040254679301718161');
    expect(page.wait).toHaveBeenNthCalledWith(1, { selector: '[data-testid="tweetTextarea_0"]' });
    expect(page.wait).toHaveBeenNthCalledWith(2, { selector: 'input[type="file"][data-testid="fileInput"]', timeout: 20 });
    expect(setFileInput).toHaveBeenCalledWith([imagePath], 'input[type="file"][data-testid="fileInput"]');
    expect(result).toEqual([
      {
        status: 'success',
        message: 'Reply posted successfully.',
        text: 'reply with image',
        image: imagePath,
      },
    ]);
  });

  it('downloads a remote image before uploading when --image-url is provided', async () => {
    const cmd = getRegistry().get('twitter/reply');
    expect(cmd?.func).toBeTypeOf('function');

    const fetchMock = vi.fn().mockResolvedValue({
      ok: true,
      headers: {
        get: vi.fn().mockReturnValue('image/png'),
      },
      arrayBuffer: vi.fn().mockResolvedValue(Uint8Array.from([0x89, 0x50, 0x4e, 0x47]).buffer),
    });
    vi.stubGlobal('fetch', fetchMock);

    const setFileInput = vi.fn().mockResolvedValue(undefined);
    const page = createPageMock([
      { ok: true, previewCount: 1 },
      { ok: true, message: 'Reply posted successfully.' },
    ], {
      setFileInput,
    });

    const result = await cmd!.func!(page, {
      url: 'https://x.com/_kop6/status/2040254679301718161?s=20',
      text: 'reply with remote image',
      'image-url': 'https://example.com/qr',
    });

    expect(fetchMock).toHaveBeenCalledWith('https://example.com/qr');
    expect(setFileInput).toHaveBeenCalledTimes(1);
    const uploadedPath = setFileInput.mock.calls[0][0][0];
    expect(uploadedPath).toMatch(/opencli-twitter-reply-.*\/image\.png$/);
    expect(fs.existsSync(uploadedPath)).toBe(false);
    expect(result).toEqual([
      {
        status: 'success',
        message: 'Reply posted successfully.',
        text: 'reply with remote image',
        'image-url': 'https://example.com/qr',
      },
    ]);

    vi.unstubAllGlobals();
  });

  it('rejects invalid image paths early', async () => {
    await expect(() => __test__.resolveImagePath('/tmp/does-not-exist.png'))
      .toThrow('Image file not found');
  });

  it('rejects using --image and --image-url together', async () => {
    const cmd = getRegistry().get('twitter/reply');
    expect(cmd?.func).toBeTypeOf('function');

    const page = createPageMock([]);

    await expect(cmd!.func!(page, {
      url: 'https://x.com/_kop6/status/2040254679301718161?s=20',
      text: 'nope',
      image: '/tmp/a.png',
      'image-url': 'https://example.com/a.png',
    })).rejects.toThrow('Use either --image or --image-url, not both.');
  });

  it('extracts tweet ids from both user and i/status URLs', () => {
    expect(__test__.extractTweetId('https://x.com/_kop6/status/2040254679301718161?s=20')).toBe('2040254679301718161');
    expect(__test__.extractTweetId('https://x.com/i/status/2040318731105313143')).toBe('2040318731105313143');
    expect(__test__.buildReplyComposerUrl('https://x.com/i/status/2040318731105313143'))
      .toBe('https://x.com/compose/post?in_reply_to=2040318731105313143');
  });

  it('prefers content-type when resolving remote image extensions', () => {
    expect(__test__.resolveImageExtension('https://example.com/no-ext', 'image/webp')).toBe('.webp');
    expect(__test__.resolveImageExtension('https://example.com/a.jpeg?x=1', null)).toBe('.jpeg');
  });
});