root/tools/gauge/test/log_writer_test.rb

require "#{File.dirname(__FILE__)}/test_setup"
require 'gauge/log_writer'
require 'stringio'

class LogWriterTest < Test::Unit::TestCase
  FIXTURES = "#{File.dirname(__FILE__)}/fixtures"

  def test_no_prior_logs
    assert !File.exist?("#{FIXTURES}/logger/pristine/testing-0000.log")
    logger = Gauge::LogWriter.new("testing", "#{FIXTURES}/logger/pristine")
    assert File.exist?("#{FIXTURES}/logger/pristine/testing-0000.log")
    assert_equal 0, logger.number
    assert_equal "testing", logger.application
  ensure
    File.delete("#{FIXTURES}/logger/pristine/testing-0000.log") rescue nil
  end

  def test_existing_logs
    logger = Gauge::LogWriter.new("testing", "#{FIXTURES}/logger/existing")
    assert File.exist?("#{FIXTURES}/logger/existing/testing-0001.log")
    assert_equal 1, logger.number
    assert_equal "testing", logger.application
  end

  def test_rotate_log
    logger = Gauge::LogWriter.new("testing", "#{FIXTURES}/logger/existing")
    assert File.exist?("#{FIXTURES}/logger/existing/testing-0001.log")
    assert_equal 1, logger.number
    logger.next!
    assert File.exist?("#{FIXTURES}/logger/existing/testing-0002.log")
    assert_equal 2, logger.number
  ensure
    File.delete("#{FIXTURES}/logger/existing/testing-0002.log") rescue nil
  end

  def test_explicit_io_object
    io = StringIO.new
    logger = Gauge::LogWriter.new("testing", io)
    assert_nil logger.number
    logger.record "testing", "hello"
    assert_match %r{testing.*hello}, io.string
  end

  def test_rotate_with_explicit_io_object
    io = StringIO.new
    logger = Gauge::LogWriter.new("testing", io)
    assert_raises(ArgumentError) { logger.next! }
  end

  def test_logger_default_max_size
    io = StringIO.new
    logger = Gauge::LogWriter.new("testing", io)
    assert_equal 100*1024*1024, logger.max_size
  end

  def test_logger_max_size
    io = StringIO.new
    logger = Gauge::LogWriter.new("testing", io, :max_size => 1024)
    assert_equal 1024, logger.max_size
  end

  def test_logger_rotate_by_max_size_exceeded
    logger = Gauge::LogWriter.new("testing", "#{FIXTURES}/logger/existing",
      :max_size => 0)
    assert_equal 1, logger.number
    logger.record "testing", "hello"
    assert_equal 2, logger.number
    assert_match %r{testing.*hello}, File.read("#{FIXTURES}/logger/existing/testing-0002.log")
  ensure
    File.delete("#{FIXTURES}/logger/existing/testing-0002.log") rescue nil
  end
end