Class: Spy::Strategy::Wrap
- Inherits:
-
Object
- Object
- Spy::Strategy::Wrap
- Defined in:
- lib/spy/strategy/wrap.rb
Instance Method Summary collapse
- #apply ⇒ Object
-
#initialize(spy) ⇒ Wrap
constructor
A new instance of Wrap.
- #undo ⇒ Object
Constructor Details
#initialize(spy) ⇒ Wrap
Returns a new instance of Wrap.
7 8 9 10 |
# File 'lib/spy/strategy/wrap.rb', line 7 def initialize(spy) @spy = spy @visibility = DetermineVisibility.call(spy.original) end |
Instance Method Details
#apply ⇒ Object
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# File 'lib/spy/strategy/wrap.rb', line 12 def apply spy = @spy visibility = @visibility @spy.original.owner.class_eval do undef_method spy.original.name # Replace the method with the spy define_method spy.original.name do |*args, &block| Spy::Strategy::Base.call(spy, self, *args, &block) end # Make the visibility of the spy match the spied original send(visibility, spy.original.name) end end |
#undo ⇒ Object
28 29 30 31 32 33 34 35 36 |
# File 'lib/spy/strategy/wrap.rb', line 28 def undo spy = @spy visibility = @visibility spy.original.owner.class_eval do remove_method spy.original.name define_method spy.original.name, spy.original send(visibility, spy.original.name) end end |