* test/digest/test_digest_extend.rb: Added tests for current digest

nahi · nahi · commit e3f1c2179283 · 2009-11-25T15:08:06.000Z
framework.


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@25925 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
diff --git 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
@@ -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
