logstash/qa/integration/framework/fixture.rb

require_relative "../services/service_locator"

# A class that holds all fixtures for a given test file. This deals with
# bootstrapping services, dealing with config files, inputs etc
class Fixture
  FIXTURES_DIR = File.expand_path(File.join("..", "..", "fixtures"), __FILE__)

  attr_reader :input
  attr_reader :actual_output
  attr_reader :test_dir

  class TemplateContext
    attr_reader :options

    def initialize(options)
      @options = options
    end

    def get_binding
      binding
    end
  end

  def initialize(test_file_location)
    @test_file_location = test_file_location
    @settings = TestSettings.new(@test_file_location)
    @service_locator = ServiceLocator.new(@settings)
    setup_services
    @input = File.join(FIXTURES_DIR, @settings.get("input")) if @settings.is_set?("input")
    @actual_output = @settings.get("actual_output")
  end

  def config(node = "root", options = nil)
    if node == "root"
      config = @settings.get("config")
    else
      config = @settings.get("config")[node]
    end

    if options != nil
       ERB.new(config, nil, "-").result(TemplateContext.new(options).get_binding)
    else
      config
    end
  end

  def get_service(name)
    @service_locator.get_service(name)
  end

  def output_equals_expected?
    FileUtils.identical?(@actual_output, @input)
  end

  def output_exists?
    File.exists?(@actual_output)
  end

  def teardown
    File.delete(@actual_output) if @settings.is_set?("actual_output") && output_exists?
    puts "Tearing down services"
    services = @settings.get("services")
    services.each do |name|
      @service_locator.get_service(name).teardown
    end
  end

  def setup_services
    puts "Setting up services"
    services = @settings.get("services")
    services.each do |name|
     @service_locator.get_service(name).setup
    end
    if @settings.is_set?("setup_script")
      puts "Setting up test specific fixtures"
      script = File.join(FIXTURES_DIR, @settings.get("setup_script"))
      `#{script}`
    end
  end
end