2 files changed, 72 insertions, 0 deletions
diff --git a/ChangeLog b/ChangeLog
index 428c8539c6..9b4c193671 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,8 @@
+Thu Nov 26 00:05:58 2009  NAKAMURA, Hiroshi  <nahi@ruby-lang.org>
+
+	* test/digest/test_digest_extend.rb: Added tests for current digest
+	  framework.
+
 Wed Nov 25 20:46:37 2009  Tanaka Akira  <akr@fsij.org>
 
 	* vm_eval.c (rb_search_method_entry): refine error message.
diff --git a/test/digest/test_digest_extend.rb b/test/digest/test_digest_extend.rb
new file mode 100644
index 0000000000..9b2e59cdf0
--- /dev/null
+++ b/test/digest/test_digest_extend.rb
@@ -0,0 +1,67 @@
+require 'test/unit'
+require 'digest'
+
+class TestDigestExtend < Test::Unit::TestCase
+  class MyDigest < Digest::Class
+    def initialize(*arg)
+      super
+      @buf = []
+    end
+
+    def update(arg)
+      @buf << arg
+      self
+    end
+
+    alias << update
+
+    def finish
+      (@buf.join.length % 256).chr
+    end
+
+    def reset
+      @buf.clear
+      self
+    end
+  end
+
+  def test_digest
+    assert_equal("\3", MyDigest.digest("foo"))
+  end
+
+  def test_hexdigest
+    assert_equal("03", MyDigest.hexdigest("foo"))
+  end
+
+  def test_context
+    digester = MyDigest.new
+    digester.update("foo")
+    assert_equal("\3", digester.digest)
+    digester.update("foobar")
+    assert_equal("\6", digester.digest)
+    digester.update("foo")
+    assert_equal("\3", digester.digest)
+  end
+
+  def test_to_s
+    digester = MyDigest.new
+    digester.update("foo")
+    assert_equal("03", digester.to_s)
+  end
+
+  def test_digest_length # breaks MyDigest#digest_length
+    assert_equal(1, MyDigest.new.digest_length)
+    MyDigest.class_eval do
+      def digest_length
+        2
+      end
+    end
+    assert_equal(2, MyDigest.new.digest_length)
+  end
+
+  def test_block_length
+    assert_raises(RuntimeError) do
+      MyDigest.new.block_length
+    end
+  end
+end