Contents

# frozen_string_literal: true

require "much-stub"

module Assert
  def self.stubs
    MuchStub.stubs
  end

  def self.stub(*pargs, **kargs, &block)
    MuchStub.stub(*pargs, **kargs, &block)
  end

  def self.stub_on_call(*pargs, **kargs, &block)
    MuchStub.stub_on_call(*pargs, **kargs, &block)
  end

  def self.unstub(*pargs, **kargs)
    MuchStub.unstub(*pargs, **kargs)
  end

  def self.unstub!
    MuchStub.unstub!
  end

  def self.stub_send(*pargs, **kargs, &block)
    orig_caller = caller_locations
    begin
      MuchStub.stub_send(*pargs, **kargs, &block)
    rescue MuchStub::NotStubbedError => ex
      ex.set_backtrace(orig_caller.map(&:to_s))
      raise ex
    end
  end

  def self.stub_tap(*pargs, **kargs, &block)
    MuchStub.tap(*pargs, **kargs, &block)
  end

  def self.stub_tap_on_call(*pargs, **kargs, &block)
    MuchStub.tap_on_call(*pargs, **kargs, &block)
  end

  def self.stub_spy(*pargs, **kargs, &block)
    MuchStub.spy(*pargs, **kargs, &block)
  end

  StubCall = MuchStub::Call
  class StubCall
    def self.name
      super.gsub("MuchStub::Call", "Assert::StubCall")
    end

    def self.to_s
      super.gsub("MuchStub::Call", "Assert::StubCall")
    end

    def self.inspect
      super.gsub("MuchStub::Call", "Assert::StubCall")
    end

    def inspect
      super.gsub("MuchStub::Call", "Assert::StubCall")
    end
  end

  StubCallSpy = MuchStub::CallSpy
  class StubCallSpy
    def self.name
      super.gsub("MuchStub::CallSpy", "Assert::StubCallSpy")
    end

    def self.to_s
      super.gsub("MuchStub::CallSpy", "Assert::StubCallSpy")
    end

    def self.inspect
      super.gsub("MuchStub::CallSpy", "Assert::StubCallSpy")
    end

    # See MuchStub::CallSpy#inspect.
    def inspect
      "#<Assert::StubCallSpy:#{"0x0%x" % (__id__ << 1)}>"
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
assert-2.19.8	lib/assert/stub.rb
assert-2.19.7	lib/assert/stub.rb