/tests/stdin.test.js

import { test } from 'node:test';
import assert from 'node:assert/strict';
import { PassThrough } from 'node:stream';
import { readStdin, getProviderLabel } from '../dist/stdin.js';

test('readStdin returns null for TTY input', async () => {
  const originalIsTTY = process.stdin.isTTY;
  Object.defineProperty(process.stdin, 'isTTY', { value: true, configurable: true });

  try {
    const result = await readStdin();
    assert.equal(result, null);
  } finally {
    Object.defineProperty(process.stdin, 'isTTY', { value: originalIsTTY, configurable: true });
  }
});

test('readStdin returns null on stream errors', async () => {
  const originalIsTTY = process.stdin.isTTY;
  const originalSetEncoding = process.stdin.setEncoding;
  Object.defineProperty(process.stdin, 'isTTY', { value: false, configurable: true });
  process.stdin.setEncoding = () => {
    throw new Error('boom');
  };

  try {
    const result = await readStdin();
    assert.equal(result, null);
  } finally {
    process.stdin.setEncoding = originalSetEncoding;
    Object.defineProperty(process.stdin, 'isTTY', { value: originalIsTTY, configurable: true });
  }
});

function createPipe() {
  const stream = new PassThrough();
  Object.defineProperty(stream, 'isTTY', { value: false, configurable: true });
  return stream;
}

test('readStdin parses valid JSON without waiting for EOF', async () => {
  const stream = createPipe();
  const resultPromise = readStdin(stream, { firstByteTimeoutMs: 20, idleTimeoutMs: 10 });

  stream.write('{"cwd":"/tmp/project","model":{"display_name":"Opus"}}');

  const result = await resultPromise;
  assert.deepEqual(result, {
    cwd: '/tmp/project',
    model: { display_name: 'Opus' },
  });
});

test('readStdin parses JSON split across multiple chunks', async () => {
  const stream = createPipe();
  const resultPromise = readStdin(stream, { firstByteTimeoutMs: 20, idleTimeoutMs: 10 });

  stream.write('{"cwd":"/tmp/project",');
  stream.write('"model":{"display_name":"Opus"}}');

  const result = await resultPromise;
  assert.deepEqual(result, {
    cwd: '/tmp/project',
    model: { display_name: 'Opus' },
  });
});

test('readStdin returns null when a non-TTY stream never produces data', async () => {
  const stream = createPipe();
  const result = await readStdin(stream, { firstByteTimeoutMs: 10, idleTimeoutMs: 5 });
  assert.equal(result, null);
});

test('readStdin returns null when partial JSON goes idle', async () => {
  const stream = createPipe();
  const resultPromise = readStdin(stream, { firstByteTimeoutMs: 20, idleTimeoutMs: 10 });

  stream.write('{"cwd":"/tmp/project"');

  const result = await resultPromise;
  assert.equal(result, null);
});

test('readStdin returns null when stdin payload exceeds the safety limit', async () => {
  const stream = createPipe();
  const resultPromise = readStdin(stream, {
    firstByteTimeoutMs: 20,
    idleTimeoutMs: 10,
    maxBytes: 16,
  });

  stream.write('{"cwd":"/tmp/project"}');

  const result = await resultPromise;
  assert.equal(result, null);
});

// getProviderLabel tests

test('getProviderLabel returns Bedrock when CLAUDE_CODE_USE_BEDROCK=1', () => {
  const orig = process.env.CLAUDE_CODE_USE_BEDROCK;
  try {
    process.env.CLAUDE_CODE_USE_BEDROCK = '1';
    const result = getProviderLabel({ model: { id: 'claude-sonnet-4-6' } });
    assert.equal(result, 'Bedrock');
  } finally {
    if (orig === undefined) delete process.env.CLAUDE_CODE_USE_BEDROCK;
    else process.env.CLAUDE_CODE_USE_BEDROCK = orig;
  }
});

test('getProviderLabel returns null when CLAUDE_CODE_USE_BEDROCK is not set', () => {
  const orig = process.env.CLAUDE_CODE_USE_BEDROCK;
  try {
    delete process.env.CLAUDE_CODE_USE_BEDROCK;
    const result = getProviderLabel({ model: { id: 'anthropic.claude-sonnet-4-6' } });
    assert.equal(result, null);
  } finally {
    if (orig === undefined) delete process.env.CLAUDE_CODE_USE_BEDROCK;
    else process.env.CLAUDE_CODE_USE_BEDROCK = orig;
  }
});

test('getProviderLabel returns null for cross-region model ID without env var', () => {
  const orig = process.env.CLAUDE_CODE_USE_BEDROCK;
  try {
    delete process.env.CLAUDE_CODE_USE_BEDROCK;
    const result = getProviderLabel({ model: { id: 'us.anthropic.claude-sonnet-4-6' } });
    assert.equal(result, null);
  } finally {
    if (orig === undefined) delete process.env.CLAUDE_CODE_USE_BEDROCK;
    else process.env.CLAUDE_CODE_USE_BEDROCK = orig;
  }
});

test('getProviderLabel returns null when CLAUDE_CODE_USE_BEDROCK=0', () => {
  const orig = process.env.CLAUDE_CODE_USE_BEDROCK;
  try {
    process.env.CLAUDE_CODE_USE_BEDROCK = '0';
    const result = getProviderLabel({ model: { id: 'anthropic.claude-sonnet-4-6' } });
    assert.equal(result, null);
  } finally {
    if (orig === undefined) delete process.env.CLAUDE_CODE_USE_BEDROCK;
    else process.env.CLAUDE_CODE_USE_BEDROCK = orig;
  }
});