module Spec::Rails::Example::RenderObserver

def register_verify_after_each #:nodoc:

:nodoc: 
def register_verify_after_each #:nodoc:
  proc = verify_rendered_proc
  Spec::Example::ExampleGroup.after(:each, &proc)
end

def render_proxy #:nodoc:

:nodoc: 
def render_proxy #:nodoc:
  @render_proxy ||= Spec::Mocks::Mock.new("render_proxy")
end

def should_not_receive(*args) 

def should_not_receive(*args)
  if args[0] == :render
    register_verify_after_each
    render_proxy.should_not_receive(:render)
  else
    super
  end
end

def should_receive(*args) 

def should_receive(*args)
  if args[0] == :render
    register_verify_after_each
    render_proxy.should_receive(:render, :expected_from => caller(1)[0])
  else
    super
  end
end

def stub(*args) 

def stub(*args)
  if args[0] == :render
    register_verify_after_each
    render_proxy.stub(args.first, :expected_from => caller(1)[0])
  else
    super
  end
end

def stub!(*args)

:expected_from. This will have to do for the moment.
above. Could delegate, but then we'd run into craziness handling
as expected in the else branch, so this is a duplicate of stub()
FIXME - for some reason, neither alias nor alias_method are working 
def stub!(*args)
  if args[0] == :render
    register_verify_after_each
    render_proxy.stub!(args.first, :expected_from => caller(1)[0])
  else
    super
  end
end

def unregister_verify_after_each #:nodoc:

:nodoc: 
def unregister_verify_after_each #:nodoc:
  proc = verify_rendered_proc
  Spec::Example::ExampleGroup.remove_after(:each, &proc)
end

def verify_rendered # :nodoc:

:nodoc: 
def verify_rendered # :nodoc:
  render_proxy.rspec_verify
end

def verify_rendered_proc #:nodoc:

:nodoc: 
def verify_rendered_proc #:nodoc:
  template = self
  @verify_rendered_proc ||= Proc.new do
    template.verify_rendered
    template.unregister_verify_after_each
  end
end