1 files changed, 95 insertions, 0 deletions
diff --git a/test/test_raindrops.rb b/test/test_raindrops.rb
new file mode 100644
index 0000000..66fc208
--- /dev/null
+++ b/test/test_raindrops.rb
@@ -0,0 +1,95 @@
+# -*- encoding: binary -*-
+require 'test/unit'
+require 'raindrops'
+
+class TestRaindrops < Test::Unit::TestCase
+
+  def test_size
+    rd = Raindrops.new(4)
+    assert_equal 4, rd.size
+  end
+
+  def test_ary
+    rd = Raindrops.new(4)
+    assert_equal [0, 0, 0, 0] , rd.to_ary
+  end
+
+  def test_incr_no_args
+    rd = Raindrops.new(4)
+    assert_equal 1, rd.incr(0)
+    assert_equal [1, 0, 0, 0], rd.to_ary
+  end
+
+  def test_incr_args
+    rd = Raindrops.new(4)
+    assert_equal 6, rd.incr(3, 6)
+    assert_equal [0, 0, 0, 6], rd.to_ary
+  end
+
+  def test_decr_args
+    rd = Raindrops.new(4)
+    rd[3] = 6
+    assert_equal 5, rd.decr(3, 1)
+    assert_equal [0, 0, 0, 5], rd.to_ary
+  end
+
+  def test_incr_shared
+    rd = Raindrops.new(2)
+    5.times do
+      pid = fork { rd.incr(1) }
+      _, status = Process.waitpid2(pid)
+      assert status.success?
+    end
+    assert_equal [0, 5], rd.to_ary
+  end
+
+  def test_incr_decr
+    rd = Raindrops.new(1)
+    fork { 1000000.times { rd.incr(0) } }
+    1000.times { rd.decr(0) }
+    statuses = Process.waitall
+    statuses.each { |pid, status| assert status.success? }
+    assert_equal [999000], rd.to_ary
+  end
+
+  def test_bad_incr
+    rd = Raindrops.new(1)
+    assert_raises(ArgumentError) { rd.incr(-1) }
+    assert_raises(ArgumentError) { rd.incr(2) }
+    assert_raises(ArgumentError) { rd.incr(0xffffffff) }
+  end
+
+  def test_dup
+    @rd = Raindrops.new(1)
+    rd = @rd.dup
+    assert_equal 1, @rd.incr(0)
+    assert_equal 1, rd.incr(0)
+    assert_equal 2, rd.incr(0)
+    assert_equal 2, rd[0]
+    assert_equal 1, @rd[0]
+  end
+
+  def test_clone
+    @rd = Raindrops.new(1)
+    rd = @rd.clone
+    assert_equal 1, @rd.incr(0)
+    assert_equal 1, rd.incr(0)
+    assert_equal 2, rd.incr(0)
+    assert_equal 2, rd[0]
+    assert_equal 1, @rd[0]
+  end
+
+  def test_big
+    expect = 256.times.map { 0 }
+    rd = Raindrops.new(256)
+    assert_equal expect, rd.to_ary
+    assert_nothing_raised { rd[255] = 5 }
+    assert_equal 5, rd[255]
+    assert_nothing_raised { rd[2] = 2 }
+
+    expect[255] = 5
+    expect[2] = 2
+    assert_equal expect, rd.to_ary
+  end
+
+end