summaryrefslogtreecommitdiff
path: root/spec/ruby/core/objectspace/weakmap/element_set_spec.rb
blob: 858887715829b85d922438ef5abea7ef8a68f239 (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
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

  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