summaryrefslogtreecommitdiff
path: root/spec/ruby/core/data/to_h_spec.rb
blob: 64816b72510ae973435840406bd71cf3848b3719 (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
63
require_relative '../../spec_helper'
require_relative 'fixtures/classes'

describe "Data#to_h" do
  it "transforms the data object into a hash" do
    data = DataSpecs::Measure.new(amount: 42, unit: 'km')
    data.to_h.should == { amount: 42, unit: 'km' }
  end

  context "with block" do
    it "transforms [key, value] pairs returned by the block into a hash" do
      data = DataSpecs::Measure.new(amount: 42, unit: 'km')
      data.to_h { |key, value| [value, key] }.should == { 42 => :amount, 'km' => :unit }
    end

    it "passes to a block each pair's key and value as separate arguments" do
      ScratchPad.record []
      data = DataSpecs::Measure.new(amount: 42, unit: 'km')
      data.to_h { |k, v| ScratchPad << [k, v]; [k, v] }
      ScratchPad.recorded.sort.should == [[:amount, 42], [:unit, 'km']]

      ScratchPad.record []
      data.to_h { |*args| ScratchPad << args; [args[0], args[1]] }
      ScratchPad.recorded.sort.should == [[:amount, 42], [:unit, 'km']]
    end

    it "raises ArgumentError if block returns longer or shorter array" do
      data = DataSpecs::Measure.new(amount: 42, unit: 'km')
      -> do
        data.to_h { |k, v| [k.to_s, v*v, 1] }
      end.should raise_error(ArgumentError, /element has wrong array length/)

      -> do
        data.to_h { |k, v| [k] }
      end.should raise_error(ArgumentError, /element has wrong array length/)
    end

    it "raises TypeError if block returns something other than Array" do
      data = DataSpecs::Measure.new(amount: 42, unit: 'km')
      -> do
        data.to_h { |k, v| "not-array" }
      end.should raise_error(TypeError, /wrong element type String/)
    end

    it "coerces returned pair to Array with #to_ary" do
      x = mock('x')
      x.stub!(:to_ary).and_return([:b, 'b'])
      data = DataSpecs::Measure.new(amount: 42, unit: 'km')

      data.to_h { |k| x }.should == { :b => 'b' }
    end

    it "does not coerce returned pair to Array with #to_a" do
      x = mock('x')
      x.stub!(:to_a).and_return([:b, 'b'])
      data = DataSpecs::Measure.new(amount: 42, unit: 'km')

      -> do
        data.to_h { |k| x }
      end.should raise_error(TypeError, /wrong element type MockObject/)
    end
  end
end