agent-ecosystem/test/renderer/utils/crossTeamPendingReplies.test.ts

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

import { computePendingCrossTeamReplies } from '@renderer/utils/crossTeamPendingReplies';

import type { InboxMessage } from '@shared/types';

function makeMessage(overrides: Partial<InboxMessage> = {}): InboxMessage {
  return {
    from: 'user',
    text: 'hello',
    timestamp: '2026-03-09T12:00:00.000Z',
    read: true,
    messageId: 'msg-1',
    ...overrides,
  };
}

describe('computePendingCrossTeamReplies', () => {
  it('returns pending entry for outbound cross-team message without reply', () => {
    const result = computePendingCrossTeamReplies([
      makeMessage({
        conversationId: 'conv-1',
        source: 'cross_team_sent',
        to: 'team-best.team-lead',
        timestamp: '2026-03-09T12:00:00.000Z',
      }),
    ]);

    expect(result).toEqual([
      {
        conversationId: 'conv-1',
        teamName: 'team-best',
        sentAtMs: Date.parse('2026-03-09T12:00:00.000Z'),
      },
    ]);
  });

  it('clears pending entry when a newer cross-team reply arrives in the same conversation', () => {
    const result = computePendingCrossTeamReplies([
      makeMessage({
        conversationId: 'conv-1',
        source: 'cross_team_sent',
        to: 'team-best.team-lead',
        timestamp: '2026-03-09T12:00:00.000Z',
      }),
      makeMessage({
        conversationId: 'conv-1',
        replyToConversationId: 'conv-1',
        from: 'team-best.team-lead',
        source: 'cross_team',
        timestamp: '2026-03-09T12:05:00.000Z',
        messageId: 'msg-2',
      }),
    ]);

    expect(result).toEqual([]);
  });

  it('keeps pending entry when the latest outbound is newer than the last reply', () => {
    const result = computePendingCrossTeamReplies([
      makeMessage({
        conversationId: 'conv-1',
        replyToConversationId: 'conv-1',
        from: 'team-best.team-lead',
        source: 'cross_team',
        timestamp: '2026-03-09T12:05:00.000Z',
        messageId: 'msg-1-reply',
      }),
      makeMessage({
        conversationId: 'conv-1',
        source: 'cross_team_sent',
        to: 'team-best.team-lead',
        timestamp: '2026-03-09T12:10:00.000Z',
        messageId: 'msg-2',
      }),
    ]);

    expect(result).toEqual([
      {
        conversationId: 'conv-1',
        teamName: 'team-best',
        sentAtMs: Date.parse('2026-03-09T12:10:00.000Z'),
      },
    ]);
  });

  it('keeps a pending conversation even when another team message arrives in a different conversation', () => {
    const result = computePendingCrossTeamReplies([
      makeMessage({
        conversationId: 'conv-1',
        source: 'cross_team_sent',
        to: 'team-best.team-lead',
        timestamp: '2026-03-09T12:00:00.000Z',
      }),
      makeMessage({
        conversationId: 'conv-2',
        from: 'team-best.team-lead',
        source: 'cross_team',
        timestamp: '2026-03-09T12:05:00.000Z',
        messageId: 'msg-2',
      }),
    ]);

    expect(result).toEqual([
      {
        conversationId: 'conv-1',
        teamName: 'team-best',
        sentAtMs: Date.parse('2026-03-09T12:00:00.000Z'),
      },
    ]);
  });

  it('ignores non-cross-team messages', () => {
    const result = computePendingCrossTeamReplies([
      makeMessage({
        from: 'alice',
        to: 'team-lead',
        timestamp: '2026-03-09T12:00:00.000Z',
      }),
    ]);

    expect(result).toEqual([]);
  });

  it('falls back to legacy team-level matching when conversationId is missing', () => {
    const result = computePendingCrossTeamReplies([
      makeMessage({
        source: 'cross_team_sent',
        to: 'team-best.team-lead',
        timestamp: '2026-03-09T12:00:00.000Z',
      }),
    ]);

    expect(result).toEqual([
      {
        teamName: 'team-best',
        sentAtMs: Date.parse('2026-03-09T12:00:00.000Z'),
      },
    ]);
  });
});