github-api.test.ts

import { describe, it, expect } from 'vitest';
import { isSuccessComment, isPRModificationFailureComment, isBotComment, isStartWorkComment } from './github-api';
import type { GitHubComment } from './types';

describe('Comment detection', () => {
  it('should detect issue success comment', () => {
    const comment: GitHubComment = {
      id: 1,
      html_url: 'https://github.com/All-Hands-AI/OpenHands/issues/1#comment-1',
      created_at: '2023-11-28T00:01:00Z',
      user: { login: 'github-actions[bot]', type: 'Bot' },
      body: 'A potential fix has been generated and a draft PR #5364 has been created. Please review the changes.',
    };
    expect(isSuccessComment(comment)).toBe(true);
  });

  it('should detect PR failure comment', () => {
    const comment: GitHubComment = {
      id: 1,
      html_url: 'https://github.com/All-Hands-AI/OpenHands/issues/1#comment-1',
      created_at: '2023-11-28T00:01:00Z',
      user: { login: 'github-actions[bot]', type: 'Bot' },
      body: 'The workflow to fix this issue encountered an error. Please check the workflow logs for more information.',
    };
    expect(isPRModificationFailureComment(comment)).toBe(true);
  });

  it('should detect openhands-ai[bot] as a bot', () => {
    const comment: GitHubComment = {
      id: 1,
      html_url: 'https://github.com/All-Hands-AI/OpenHands/issues/1#comment-1',
      created_at: '2023-11-28T00:01:00Z',
      user: { login: 'openhands-ai[bot]', type: 'Bot' },
      body: 'Openhands is working, @neubig can click here to continue refining the PR',
    };
    expect(isBotComment(comment)).toBe(true);
  });

  it('should detect new start work format', () => {
    const comment: GitHubComment = {
      id: 1,
      html_url: 'https://github.com/All-Hands-AI/OpenHands/issues/1#comment-1',
      created_at: '2023-11-28T00:01:00Z',
      user: { login: 'openhands-ai[bot]', type: 'Bot' },
      body: 'Openhands is working, @neubig can click here to continue refining the PR',
    };
    expect(isStartWorkComment(comment)).toBe(true);
  });

  it('should detect new success format', () => {
    const comment: GitHubComment = {
      id: 1,
      html_url: 'https://github.com/All-Hands-AI/OpenHands/issues/1#comment-1',
      created_at: '2023-11-28T00:01:00Z',
      user: { login: 'openhands-ai[bot]', type: 'Bot' },
      body: '@neubig can click here to [continue refining the PR](https://app.all-hands.dev/conversations/e06b225de3a24fea97504511c80afc44)',
    };
    expect(isSuccessComment(comment)).toBe(true);
  });
});