require 'cucumber/rails/world'
require 'facebooker/rails/integration_session'

module Facebooker
  module Rails
    module Cucumber
      class World < ::Cucumber::Rails::World
        def open_session
          session = Facebooker::Rails::IntegrationSession.new

          # delegate the fixture accessors back to the test instance
          extras = Module.new { attr_accessor :delegate, :test_result }
          if self.class.respond_to?(:fixture_table_names)
            self.class.fixture_table_names.each do |table_name|
              name = table_name.tr(".", "_")
              next unless respond_to?(name)
              extras.__send__(:define_method, name) { |*args| delegate.send(name, *args) }
            end
          end

          # delegate add_assertion to the test case
          extras.__send__(:define_method, :add_assertion) { test_result.add_assertion }
          session.extend(extras)
          session.delegate = self
          session.test_result = @_result

          yield session if block_given?
          session
        end
        
        def without_canvas
          in_canvas = @integration_session.canvas
          @integration_session.canvas = false
          yield
          @integration_session.canvas = in_canvas
        end
      end
    end
  end
end

World do
  w = Facebooker::Rails::Cucumber::World.new
  w.reset!
  w
end