spec/ruby/core/module/private_class_method_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

require_relative '../../spec_helper'
require_relative 'fixtures/classes'

describe "Module#private_class_method" do
  before :each do
    # This is not in classes.rb because after marking a class method private it
    # will stay private.
    class << ModuleSpecs::Parent
      public
      def private_method_1; end
      def private_method_2; end
    end
  end

  after :each do
    class << ModuleSpecs::Parent
      remove_method :private_method_1
      remove_method :private_method_2
    end
  end

  it "makes an existing class method private" do
    ModuleSpecs::Parent.private_method_1.should == nil
    ModuleSpecs::Parent.private_class_method :private_method_1
    -> { ModuleSpecs::Parent.private_method_1  }.should raise_error(NoMethodError)

    # Technically above we're testing the Singleton classes, class method(right?).
    # Try a "real" class method set private.
    -> { ModuleSpecs::Parent.private_method }.should raise_error(NoMethodError)
  end

  it "makes an existing class method private up the inheritance tree" do
    ModuleSpecs::Child.public_class_method :private_method_1
    ModuleSpecs::Child.private_method_1.should == nil
    ModuleSpecs::Child.private_class_method :private_method_1

    -> { ModuleSpecs::Child.private_method_1 }.should raise_error(NoMethodError)
    -> { ModuleSpecs::Child.private_method   }.should raise_error(NoMethodError)
  end

  it "accepts more than one method at a time" do
    ModuleSpecs::Parent.private_method_1.should == nil
    ModuleSpecs::Parent.private_method_2.should == nil

    ModuleSpecs::Child.private_class_method :private_method_1, :private_method_2

    -> { ModuleSpecs::Child.private_method_1 }.should raise_error(NoMethodError)
    -> { ModuleSpecs::Child.private_method_2 }.should raise_error(NoMethodError)
  end

  it "raises a NameError if class method doesn't exist" do
    -> do
      ModuleSpecs.private_class_method :no_method_here
    end.should raise_error(NameError)
  end

  it "makes a class method private" do
    c = Class.new do
      def self.foo() "foo" end
      private_class_method :foo
    end
    -> { c.foo }.should raise_error(NoMethodError)
  end

  it "raises a NameError when the given name is not a method" do
    -> do
      Class.new do
        private_class_method :foo
      end
    end.should raise_error(NameError)
  end

  it "raises a NameError when the given name is an instance method" do
    -> do
      Class.new do
        def foo() "foo" end
        private_class_method :foo
      end
    end.should raise_error(NameError)
  end
end