summaryrefslogtreecommitdiff
path: root/spec/ruby/core/objectspace/weakmap/element_set_spec.rb
blob: 2b536501482f854abab52538d5bfad9ca3d94f0f (plain)
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
require_relative '../../../spec_helper'

describe "ObjectSpace::WeakMap#[]=" do
  def should_accept(map, key, value)
    (map[key] = value).should == value
    map.should.key?(key)
    map[key].should == value
  end

  it "is correct" do
    map = ObjectSpace::WeakMap.new
    key1, key2 = %w[a b].map(&:upcase)
    ref1, ref2 = %w[x y]
    should_accept(map, key1, ref1)
    should_accept(map, key1, ref1)
    should_accept(map, key2, ref2)
    map[key1].should == ref1
  end

  ruby_version_is ""..."2.7" do
    it "does not accept primitive or frozen keys or values" do
      map = ObjectSpace::WeakMap.new
      x = Object.new
      -> { map[true] = x }.should raise_error(ArgumentError)
      -> { map[false] = x }.should raise_error(ArgumentError)
      -> { map[nil] = x }.should raise_error(ArgumentError)
      -> { map[42] = x }.should raise_error(ArgumentError)
      -> { map[:foo] = x }.should raise_error(ArgumentError)
      -> { map[x] = true }.should raise_error(ArgumentError)
      -> { map[x] = false }.should raise_error(ArgumentError)
      -> { map[x] = nil }.should raise_error(ArgumentError)
      -> { map[x] = 42 }.should raise_error(ArgumentError)
      -> { map[x] = :foo }.should raise_error(ArgumentError)

      y = Object.new.freeze
      -> { map[x] = y}.should raise_error(FrozenError)
      -> { map[y] = x}.should raise_error(FrozenError)
    end
  end

  ruby_version_is "2.7" do
    it "accepts primitive or frozen keys or values" do
      map = ObjectSpace::WeakMap.new
      x = Object.new
      should_accept(map, true, x)
      should_accept(map, false, x)
      should_accept(map, nil, x)
      should_accept(map, 42, x)
      should_accept(map, :foo, x)

      should_accept(map, x, true)
      should_accept(map, x, false)
      should_accept(map, x, 42)
      should_accept(map, x, :foo)

      y = Object.new.freeze
      should_accept(map, x, y)
      should_accept(map, y, x)
    end
  end
end