require 'rubygems'
require 'bundler/setup'

$LOAD_PATH.unshift(File.expand_path('../../lib', __FILE__))

require 'zk'
require 'benchmark'

ZK_TEST_PORT = 2181

LOG_FILE = File.open(File.join(ZK::ZK_ROOT, 'test.log'), 'a').tap { |f| f.sync = true }

ZK.logger = Logger.new(LOG_FILE).tap { |log| log.level = Logger::DEBUG }
#Zookeeper.logger = ZK.logger

# Requires supporting ruby files with custom matchers and macros, etc,
# in spec/support/ and its subdirectories.
Dir[File.expand_path("../support/**/*.rb", __FILE__)].each {|f| require f}

$stderr.sync = true

# COMMENT THESE LINES FOR REMOTE DEBUGGING
require 'ruby-debug'
require 'flexmock'

RSpec.configure do |config|
  config.mock_with :flexmock
  config.include(FlexMock::ArgumentTypes)

#   config.before(:all) do
#     unless $did_debug
#       $did_debug = true
#       $stderr.puts "debugger started? is #{Debugger.started?.inspect}"

#       Debugger.wait_connection = true
#       $stderr.puts "run 'rdebug -c -p #{Debugger::PORT}'"
#       Debugger.start_remote

#       config.debug = true
#     end
#   end
end

def logger
  ZK.logger
end

# method to wait until block passed returns true or timeout (default is 2 seconds) is reached 
def wait_until(timeout=2)
  time_to_stop = Time.now + timeout

  until yield 
    break if Time.now > time_to_stop
    Thread.pass
  end
end

def wait_while(timeout=2)
  time_to_stop = Time.now + timeout

  while yield 
    break if Time.now > time_to_stop
    Thread.pass
  end
end


class ::Thread
  # join with thread until given block is true, the thread joins successfully, 
  # or timeout seconds have passed
  #
  def join_until(timeout=2)
    time_to_stop = Time.now + timeout

    until yield
      break if Time.now > time_to_stop
      break if join(0.1)
    end
  end
  
  def join_while(timeout=2)
    time_to_stop = Time.now + timeout

    while yield
      break if Time.now > time_to_stop
      break if join(0.1)
    end
  end
end

def report_realtime(what)
  t = Benchmark.realtime { yield }
  $stderr.puts "#{what}: %0.3f" % [t.to_f]
end