class Mocha::Mockery

def add_mock(mock)
  mocks << mock
  mock
end

def add_state_machine(state_machine)
  state_machines << state_machine
  state_machine
end

def check(action, description, signature_proc, backtrace = caller)
  treatment = Mocha.configuration.send(action)
  return if (treatment == :allow) || (block_given? && !yield)
  method_signature = signature_proc.call
  message = "stubbing #{description}: #{method_signature}"
  raise StubbingError.new(message, backtrace) if treatment == :prevent
  logger.warn(message) if treatment == :warn
end

def expectations
  mocks.map { |mock| mock.__expectations__.to_a }.flatten
end

def instance
  @instances.last || Null.new
end

def logger
  @logger ||= Logger.new($stderr)
end

def mocha_inspect
  lines = []
  lines << "unsatisfied expectations:\n- #{unsatisfied_expectations.map(&:mocha_inspect).join("\n- ")}\n" if unsatisfied_expectations.any?
  lines << "satisfied expectations:\n- #{satisfied_expectations.map(&:mocha_inspect).join("\n- ")}\n" if satisfied_expectations.any?
  lines << "states:\n- #{state_machines.map(&:mocha_inspect).join("\n- ")}\n" if state_machines.any?
  lines.join
end

def mock_impersonating(object)
  add_mock(Mock.new(self, ImpersonatingName.new(object), ObjectReceiver.new(object)))
end

def mock_impersonating_any_instance_of(klass)
  add_mock(Mock.new(self, ImpersonatingAnyInstanceName.new(klass), AnyInstanceReceiver.new(klass)))
end

def mocks
  @mocks ||= []
end

def named_mock(name)
  add_mock(Mock.new(self, Name.new(name)))
end

def new_state_machine(name)
  add_state_machine(StateMachine.new(name))
end

def on_stubbing(object, method)
  signature_proc = lambda { "#{object.mocha_inspect}.#{method}" }
  check(:stubbing_non_existent_method, 'non-existent method', signature_proc) do
    !(object.stubba_class.__method_exists__?(method, true) || object.respond_to?(method))
  end
  check(:stubbing_non_public_method, 'non-public method', signature_proc) do
    object.stubba_class.__method_exists__?(method, false)
  end
  check(:stubbing_method_on_nil, 'method on nil', signature_proc) { object.nil? }
  check(:stubbing_method_on_non_mock_object, 'method on non-mock object', signature_proc)
end

def reset
  @stubba = nil
  @mocks = nil
  @state_machines = nil
end

def satisfied_expectations
  expectations.select(&:verified?)
end

def sequences
  @sequences ||= []
end

def setup
  @instances ||= []
  mockery = new
  mockery.logger = instance.logger unless @instances.empty?
  @instances.push(mockery)
end

def state_machines
  @state_machines ||= []
end

def stubba
  @stubba ||= Central.new
end

def teardown(origin = nil)
  instance.teardown(origin)
ensure
  @instances.pop
end

def teardown(origin = nil)
  stubba.unstub_all
  mocks.each { |m| m.__expire__(origin) }
  reset
end

def unnamed_mock
  add_mock(Mock.new(self))
end

def unsatisfied_expectations
  expectations.reject(&:verified?)
end

def verify(*args)
  instance.verify(*args)
end

def verify(assertion_counter = nil)
  unless mocks.all? { |mock| mock.__verified__?(assertion_counter) }
    message = "not all expectations were satisfied\n#{mocha_inspect}"
    backtrace = if unsatisfied_expectations.empty?
                  caller
                else
                  unsatisfied_expectations[0].backtrace
                end
    raise ExpectationErrorFactory.build(message, backtrace)
  end
  expectations.reject(&:used?).each do |expectation|
    signature_proc = lambda { expectation.method_signature }
    check(:stubbing_method_unnecessarily, 'method unnecessarily', signature_proc, expectation.backtrace)
  end
end