1 files changed, 34 insertions, 0 deletions
diff --git a/spec/ruby/library/thread/sizedqueue/shared/enque.rb b/spec/ruby/library/thread/sizedqueue/shared/enque.rb
new file mode 100644
index 0000000000..627f8bd745
--- /dev/null
+++ b/spec/ruby/library/thread/sizedqueue/shared/enque.rb
@@ -0,0 +1,34 @@
+describe :sizedqueue_enq, shared: true do
+  it "blocks if queued elements exceed size" do
+    q = SizedQueue.new(1)
+
+    q.size.should == 0
+    q.send(@method, :first_element)
+    q.size.should == 1
+
+    blocked_thread = Thread.new { q.send(@method, :second_element) }
+    sleep 0.01 until blocked_thread.stop?
+
+    q.size.should == 1
+    q.pop.should == :first_element
+
+    blocked_thread.join
+    q.size.should == 1
+    q.pop.should == :second_element
+    q.size.should == 0
+  end
+
+  it "raises a ThreadError if queued elements exceed size when not blocking" do
+    q = SizedQueue.new(2)
+
+    non_blocking = true
+    add_to_queue = lambda { q.send(@method, Object.new, non_blocking) }
+
+    q.size.should == 0
+    add_to_queue.call
+    q.size.should == 1
+    add_to_queue.call
+    q.size.should == 2
+    add_to_queue.should raise_error(ThreadError)
+  end
+end