class Cucumber::WireSupport::Configuration

def default_timeouts
  {
    'invoke' => 120,
    'begin_scenario' => 120,
    'end_scenario' => 120
  }
end

def initialize(wire_file)
  params = YAML.load_file(wire_file)
  @host = params['host']
  @port = params['port']
  @timeouts = default_timeouts.merge(params['timeout'] || {})
end

def timeout(message = nil)
  return @timeouts[message.to_s] || 3
end