Contents

# Core ext methods

class Object < BasicObject
  def stub_ivar(name, value = nil)
    instance_variable_set(name, value)
    yield
  ensure
    remove_instance_variable(name)
  end

  def stub_cvar(name, value = nil)
    class_variable_set(name, value)
    yield
  ensure
    remove_class_variable(name)
  end

  def stub_const(name, value)
    if const_defined?(name)
      old_value = const_get(name)
      remove_const(name)
    end

    const_set(name, value)
    yield
  ensure
    remove_const(name)
    const_set(name, old_value) if defined? old_value
  end

  def stub_imethod(name, &block)
    lambda do |&run_block|
      alias_method(:"_old_#{name}", name) if (reset_old = method_defined?(name))
      define_method(name, &block)
      run_block.call
    ensure
      undef_method(name)
      if reset_old
        alias_method(name, :"_old_#{name}")
        undef_method(:"_old_#{name}")
      end
    end
  end

  def get_reset_ivar(name, *extra, &block)
    instance_variable_set(name, extra.first) if extra.any?
    instance_exec(&block)

    instance_variable_get(name).tap do
      remove_instance_variable(name) if instance_variable_defined?(name)
    end
  end

  def get_reset_cvar(name, *extra, &block)
    class_variable_set(name, extra.first) if extra.any?
    instance_exec(&block)

    class_variable_get(name).tap do
      remove_class_variable(name) if class_variable_defined?(name)
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
rails-graphql-0.2.1	test/test_ext.rb
rails-graphql-0.2.0	test/test_ext.rb