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

import {
  AGENT_CHAT_ATTACHMENT_ACCEPT,
  AGENT_IMAGE_REMOVE_REQUIRED_MESSAGE,
  AGENT_IMAGE_UNSUPPORTED_MESSAGE,
  supportsAgentImageInput,
} from './imageCapability';

describe('agent chat image capability', () => {
  it('仅允许明确启用图片能力的发布模型', () => {
    expect(supportsAgentImageInput(true)).toBe(true);
    expect(supportsAgentImageInput(false)).toBe(false);
    expect(supportsAgentImageInput(null)).toBe(false);
    expect(supportsAgentImageInput(undefined)).toBe(false);
  });

  it('附件选择器同时展示文档和图片格式', () => {
    expect(AGENT_CHAT_ATTACHMENT_ACCEPT).toContain('.pdf');
    expect(AGENT_CHAT_ATTACHMENT_ACCEPT).toContain('.png');
    expect(AGENT_CHAT_ATTACHMENT_ACCEPT).toContain('image/png');
  });

  it('为视觉能力限制提供明确反馈', () => {
    expect(AGENT_IMAGE_UNSUPPORTED_MESSAGE).toBe(
      '当前大模型不支持视觉，无法上传图片',
    );
    expect(AGENT_IMAGE_REMOVE_REQUIRED_MESSAGE).toBe(
      '当前大模型不支持视觉，已添加的图片无法发送，请先移除',
    );
  });
});