module Typhoeus::Hydra::Stubbing

def self.included(base)
  base.extend(SharedMethods)
  base.class_eval do
    attr_accessor :stubs
  end
end

def assign_to_stub(request)
  m = find_stub_from_request(request)
  # Fallback to global stubs.
  m ||= self.class.find_stub_from_request(request)
  if m
    m.add_request(request)
    @active_stubs << m
    m
  else
    nil
  end
end