summaryrefslogtreecommitdiff
path: root/puppet/modules/stdlib/spec/monkey_patches/publicize_methods.rb
diff options
context:
space:
mode:
Diffstat (limited to 'puppet/modules/stdlib/spec/monkey_patches/publicize_methods.rb')
-rwxr-xr-xpuppet/modules/stdlib/spec/monkey_patches/publicize_methods.rb11
1 files changed, 11 insertions, 0 deletions
diff --git a/puppet/modules/stdlib/spec/monkey_patches/publicize_methods.rb b/puppet/modules/stdlib/spec/monkey_patches/publicize_methods.rb
new file mode 100755
index 00000000..3ae59f97
--- /dev/null
+++ b/puppet/modules/stdlib/spec/monkey_patches/publicize_methods.rb
@@ -0,0 +1,11 @@
+#! /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