1 files changed, 43 insertions, 0 deletions
diff --git a/spec/ruby/core/main/public_spec.rb b/spec/ruby/core/main/public_spec.rb
new file mode 100644
index 0000000000..89368ebb0d
--- /dev/null
+++ b/spec/ruby/core/main/public_spec.rb
@@ -0,0 +1,43 @@
+require_relative '../../spec_helper'
+require_relative 'fixtures/classes'
+
+describe "main#public" do
+  after :each do
+    Object.send(:private, :main_private_method)
+    Object.send(:private, :main_private_method2)
+  end
+
+  context "when single argument is passed and it is not an array" do
+    it "sets the visibility of the given methods to public" do
+      eval "public :main_private_method", TOPLEVEL_BINDING
+      Object.private_methods(true).should_not.include?(:main_private_method)
+    end
+  end
+
+  context "when multiple arguments are passed" do
+    it "sets the visibility of the given methods to public" do
+      eval "public :main_private_method, :main_private_method2", TOPLEVEL_BINDING
+      Object.private_methods(true).should_not.include?(:main_private_method)
+      Object.private_methods(true).should_not.include?(:main_private_method2)
+    end
+  end
+
+  context "when single argument is passed and is an array" do
+    it "sets the visibility of the given methods to public" do
+      eval "public [:main_private_method, :main_private_method2]", TOPLEVEL_BINDING
+      Object.private_methods(true).should_not.include?(:main_private_method)
+      Object.private_methods(true).should_not.include?(:main_private_method2)
+    end
+  end
+
+  it "returns argument" do
+    eval("public :main_private_method", TOPLEVEL_BINDING).should.equal?(:main_private_method)
+  end
+
+
+  it "raises a NameError when given an undefined name" do
+    -> do
+      eval "public :main_undefined_method", TOPLEVEL_BINDING
+    end.should.raise(NameError)
+  end
+end