require "minitest/autorun"
require "minitest/reporters"

ENV["RACK_ENV"] = "test"
MiniTest::Reporters.use!

require 'rubygems'
require 'bundler'

Bundler.require(:default, :test)

# Preload memcache-client
require 'memcache'

@trace_dir = File.dirname(__FILE__) + "/../tmp/"
$trace_file = @trace_dir + "trace_output.bson"

# Create a oboe-ruby/tmp dir to store trace output
Dir.mkdir @trace_dir unless File.exists?(@trace_dir) and File.directory?(@trace_dir)

# Configure Oboe
Oboe::Config[:tracing_mode] = "always"
Oboe::Config[:sample_rate] = 1000000
Oboe::Ruby.initialize
Oboe.logger.level = Logger::DEBUG

Oboe.logger.debug "[oboe/test] Debug log output test."

##
# clear_all_traces
#
# Truncates the trace output file to zero
#
def clear_all_traces
  File.truncate($trace_file, 0)
end

##
# get_all_traces
#
# Retrieves all traces written to the trace file
#
def get_all_traces
  traces = []
  f = File.open($trace_file)
  until f.eof?
    traces << BSON.read_bson_document(f)
  end
  traces
end

##
# validate_outer_layers
#
# Validates that the KVs in kvs are present
# in event
#
def validate_outer_layers(traces, layer)
  traces.first['Layer'].must_equal layer
  traces.first['Label'].must_equal 'entry'
  traces.last['Layer'].must_equal layer
  traces.last['Label'].must_equal 'exit'
end

##
# validate_event_keys
#
# Validates that the KVs in kvs are present
# in event
#
def validate_event_keys(event, kvs)
  kvs.each do |k, v|
    event.has_key?(k).must_equal true
    event[k].must_equal v
  end
end

##
# layer_has_key
#
# Checks an array of trace events if a specific layer (regardless of event type)
# has he specified key
#
def layer_has_key(traces, layer, key)
  has_key = false

  traces.each do |t|
    if t["Layer"] == layer and t.has_key?(key)
      has_key = true

      (t["Backtrace"].length > 0).must_equal true
    end
  end

  has_key.must_equal true
end

##
# layer_doesnt_have_key
#
# Checks an array of trace events to assure that a specific layer 
# (regardless of event type) doesn't have the specified key
#
def layer_doesnt_have_key(traces, layer, key)
  has_key = false

  traces.each do |t|
    has_key = true if t["Layer"] == layer and t.has_key?(key)
  end

  has_key.must_equal false
end