require 'timeout'
require 'rubygems'
require 'bundler'
Bundler.setup :default, :development, :test
require 'pry'
require 'pathname'

$LOAD_PATH << ::File.expand_path('../..', __FILE__)
SUPPORT_PATH = Pathname.new(::File.expand_path('../support', __FILE__))
PROTOS_PATH = SUPPORT_PATH.join('protos')
$LOAD_PATH << SUPPORT_PATH

require 'protobuf'
require 'protobuf/rpc/server'
require SUPPORT_PATH.join('all')

$LOAD_PATH << ::File.expand_path("../../lib/protobuf/descriptors", __FILE__)
require 'google/protobuf/compiler/plugin.pb'

# Including a way to turn on debug logger for spec runs
if ENV.key?('DEBUG')
  debug_log = ::File.expand_path('../../debug_specs.log', __FILE__)
  ::Protobuf::Logging.initialize_logger(debug_log, ::Logger::DEBUG)
else
  ::Protobuf::Logging.initialize_logger('/dev/null')
end

# Get rid of the deprecation env var if present (messes with specs).
ENV.delete("PB_IGNORE_DEPRECATIONS")

::Protobuf::Rpc::Client.class_eval do
  def ==(other)
    connector.options == other.options && \
      success_cb == other.success_cb && \
      failure_cb == other.failure_cb
  end
end

def reset_service_location(service)
  service.host = nil
  service.port = nil
end