puppetlabs-stdlib/spec/monkey_patches/publicize_methods.rb

11 lines
400 B
Ruby
Executable file

#! /usr/bin/env ruby -S rspec
# Some monkey-patching to allow us to test private methods.
class Class
def publicize_methods(*methods)
saved_private_instance_methods = methods.empty? ? self.private_instance_methods : methods
self.class_eval { public(*saved_private_instance_methods) }
yield
self.class_eval { private(*saved_private_instance_methods) }
end
end