summaryrefslogtreecommitdiff
path: root/spec/ruby/core/thread/thread_variable_set_spec.rb
blob: eadee76afba00d7a7be6765c77c9bc2ee5e42490 (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
62
require_relative '../../spec_helper'

describe "Thread#thread_variable_set" do
  before :each do
    @t = Thread.new { }
  end

  after :each do
    @t.join
  end

  it "returns the value set" do
    @t.thread_variable_set(:a, 2).should == 2
  end

  it "sets a value that will be returned by #thread_variable_get" do
    @t.thread_variable_set(:a, 49)
    @t.thread_variable_get(:a).should == 49
  end

  it "sets a value private to self" do
    @t.thread_variable_set(:thread_variable_get_spec, 82)
    @t.thread_variable_get(:thread_variable_get_spec).should == 82
    Thread.current.thread_variable_get(:thread_variable_get_spec).should be_nil
  end

  it "accepts String and Symbol keys interchangeably" do
    @t.thread_variable_set('a', 49)
    @t.thread_variable_get('a').should == 49

    @t.thread_variable_set(:a, 50)
    @t.thread_variable_get('a').should == 50
  end

  it "converts a key that is neither String nor Symbol with #to_str" do
    key = mock('key')
    key.should_receive(:to_str).and_return('a')
    @t.thread_variable_set(key, 49)
    @t.thread_variable_get(:a).should == 49
  end

  it "removes a key if the value is nil" do
    @t.thread_variable_set(:a, 52)
    @t.thread_variable_set(:a, nil)
    @t.thread_variable?(:a).should be_false
  end

  it "raises a FrozenError if the thread is frozen" do
    @t.freeze
    -> { @t.thread_variable_set(:a, 1) }.should raise_error(FrozenError, "can't modify frozen thread locals")
  end

  it "raises a TypeError if the key is neither Symbol nor String, nor responds to #to_str" do
    -> { @t.thread_variable_set(123, 1) }.should raise_error(TypeError, /123 is not a symbol/)
  end

  it "does not try to convert the key with #to_sym" do
    key = mock('key')
    key.should_not_receive(:to_sym)
    -> { @t.thread_variable_set(key, 42) }.should raise_error(TypeError, /#{Regexp.quote(key.inspect)} is not a symbol/)
  end
end