ruby_llm_spec.rb

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe RubyLLM do
  describe '.logger' do
    let(:logger) { instance_double(Logger) }
    let(:log_file) { double }
    let(:log_level) { double }

    before do
      described_class.instance_variable_set(:@config, nil)
      described_class.instance_variable_set(:@logger, nil)
    end

    after do
      described_class.instance_variable_set(:@config, nil)
      described_class.instance_variable_set(:@logger, nil)
    end

    context 'with configuration options' do
      before do
        described_class.configure do |config|
          config.log_file = log_file
          config.log_level = log_level
        end
      end

      it 'returns a default Logger' do
        allow(Logger).to receive(:new).with(log_file, progname: 'RubyLLM', level: log_level).and_return(logger)

        expect(described_class.logger).to eq(logger)
      end
    end

    context 'with a custom logger' do
      before do
        described_class.configure do |config|
          config.logger = logger
          config.log_file = log_file
          config.log_level = log_level
        end
      end

      it 'returns a the custom Logger' do
        expect(described_class.logger).to eq(logger)
      end
    end

    context 'with RUBYLLM_LOG_FILE set' do
      around do |example|
        original_log_file = ENV.fetch('RUBYLLM_LOG_FILE', nil)
        ENV['RUBYLLM_LOG_FILE'] = '/tmp/ruby_llm.log'
        example.run
      ensure
        ENV['RUBYLLM_LOG_FILE'] = original_log_file
      end

      it 'uses the env var as the default log file' do
        allow(Logger).to receive(:new)
          .with('/tmp/ruby_llm.log', progname: 'RubyLLM', level: Logger::INFO)
          .and_return(logger)

        expect(described_class.logger).to eq(logger)
      end
    end
  end
end