import { describe, it, expect, vi } from 'vitest';
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
import { StdioTransport } from '../../transports/stdio.js';

describe('StdioTransport', () => {
  it('should create instance with transport', () => {
    const transport = new StdioTransport();
    expect(transport).toBeDefined();
    expect(transport.getName()).toBe('stdio');
  });

  it('should return underlying transport', () => {
    const transport = new StdioTransport();
    const underlying = transport.getUnderlyingTransport();
    expect(underlying).toBeDefined();
    expect(underlying.constructor.name).toBe('StdioServerTransport');
  });

  it('should connect to server', async () => {
    const transport = new StdioTransport();
    const mockServer = {
      connect: vi.fn<() => Promise<any>>().mockResolvedValue(undefined as never),
    } as unknown as Server;

    await transport.connect(mockServer);
    expect(mockServer.connect).toHaveBeenCalledWith(transport.getUnderlyingTransport());
  });

  it('should add connect method to underlying transport', () => {
    const transport = new StdioTransport();
    const underlying = transport.getUnderlyingTransport();
    // The transport should have a connect method added via our workaround
    expect('connect' in underlying).toBe(true);
    expect(typeof (underlying as { connect?: unknown }).connect).toBe('function');
  });
});