1 files changed, 64 insertions, 0 deletions
diff --git a/spec/rubyspec/core/hash/shift_spec.rb b/spec/rubyspec/core/hash/shift_spec.rb
new file mode 100644
index 0000000000..3991da9656
--- /dev/null
+++ b/spec/rubyspec/core/hash/shift_spec.rb
@@ -0,0 +1,64 @@
+require File.expand_path('../../../spec_helper', __FILE__)
+require File.expand_path('../fixtures/classes', __FILE__)
+
+describe "Hash#shift" do
+  it "removes a pair from hash and return it" do
+    h = { a: 1, b: 2, "c" => 3, nil => 4, [] => 5 }
+    h2 = h.dup
+
+    h.size.times do |i|
+      r = h.shift
+      r.should be_kind_of(Array)
+      h2[r.first].should == r.last
+      h.size.should == h2.size - i - 1
+    end
+
+    h.should == {}
+  end
+
+  # MRI explicitly implements this behavior
+  it "allows shifting entries while iterating" do
+    h = { a: 1, b: 2, c: 3 }
+    visited = []
+    shifted = []
+    h.each_pair { |k,v|
+      visited << k
+      shifted << h.shift
+    }
+    visited.should == [:a, :b, :c]
+    shifted.should == [[:a, 1], [:b, 2], [:c, 3]]
+    h.should == {}
+  end
+
+  it "calls #default with nil if the Hash is empty" do
+    h = {}
+    def h.default(key)
+      key.should == nil
+      :foo
+    end
+    h.shift.should == :foo
+  end
+
+  it "returns nil from an empty hash" do
+    {}.shift.should == nil
+  end
+
+  it "returns (computed) default for empty hashes" do
+    Hash.new(5).shift.should == 5
+    h = Hash.new { |*args| args }
+    h.shift.should == [h, nil]
+  end
+
+  it "preserves Hash invariants when removing the last item" do
+    h = { :a => 1, :b => 2 }
+    h.shift.should == [:a, 1]
+    h.shift.should == [:b, 2]
+    h[:c] = 3
+    h.should == {:c => 3}
+  end
+
+  it "raises a RuntimeError if called on a frozen instance" do
+    lambda { HashSpecs.frozen_hash.shift  }.should raise_error(RuntimeError)
+    lambda { HashSpecs.empty_frozen_hash.shift }.should raise_error(RuntimeError)
+  end
+end