160 lines
3.5 KiB
Ruby
160 lines
3.5 KiB
Ruby
# frozen_string_literal: false
|
|
require 'test/unit'
|
|
require 'digest'
|
|
require_relative '../lib/with_different_ofs'
|
|
|
|
class TestDigestExtend < Test::Unit::TestCase
|
|
extend DifferentOFS
|
|
|
|
class MyDigest < Digest::Class
|
|
def initialize(*arg)
|
|
super
|
|
@buf = []
|
|
end
|
|
|
|
def initialize_copy(org)
|
|
@buf = org.buf.dup
|
|
end
|
|
|
|
def update(arg)
|
|
@buf << arg
|
|
self
|
|
end
|
|
|
|
alias << update
|
|
|
|
def finish
|
|
(@buf.join('').length % 256).chr
|
|
end
|
|
|
|
def reset
|
|
@buf.clear
|
|
self
|
|
end
|
|
|
|
protected
|
|
|
|
def buf
|
|
@buf
|
|
end
|
|
end
|
|
|
|
def setup
|
|
@MyDigest = Class.new(MyDigest)
|
|
end
|
|
|
|
def test_digest_s_hexencode
|
|
assert_equal('', Digest.hexencode(''))
|
|
assert_equal('0102', Digest.hexencode("\1\2"))
|
|
assert_equal(
|
|
(0..0xff).to_a.map { |c| sprintf("%02x", c ) }.join(''),
|
|
Digest.hexencode((0..0xff).to_a.map { |c| c.chr }.join(''))
|
|
)
|
|
assert_equal(Encoding::US_ASCII, Digest.hexencode("\1\2").encoding)
|
|
end
|
|
|
|
def test_class_reset
|
|
a = Digest::SHA1.new
|
|
base = a.to_s
|
|
assert_equal(base, a.reset.to_s)
|
|
b = a.new
|
|
assert_equal(base, b.to_s)
|
|
b.update('1')
|
|
assert_not_equal(base, b.to_s)
|
|
assert_equal(base, b.reset.to_s)
|
|
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("\t", digester.digest)
|
|
digester.update("foo")
|
|
assert_equal("\f", digester.digest)
|
|
end
|
|
|
|
def test_new
|
|
a = Digest::SHA1.new
|
|
b = a.new
|
|
obj = a.to_s
|
|
assert_equal(obj, a.to_s)
|
|
assert_equal(obj, b.to_s)
|
|
a.update('1')
|
|
assert_not_equal(obj, a.to_s)
|
|
assert_equal(obj, b.to_s)
|
|
end
|
|
|
|
def test_digest_hexdigest
|
|
[:digest, :hexdigest].each do |m|
|
|
exp_1st = "\3"; exp_1st = Digest.hexencode(exp_1st) if m == :hexdigest
|
|
exp_2nd = "\6"; exp_2nd = Digest.hexencode(exp_2nd) if m == :hexdigest
|
|
digester = @MyDigest.new
|
|
digester.update("foo")
|
|
obj = digester.send(m)
|
|
# digest w/o param does not reset the org digester.
|
|
assert_equal(exp_1st, obj)
|
|
digester.update("bar")
|
|
obj = digester.send(m)
|
|
assert_equal(exp_2nd, obj)
|
|
obj = digester.send(m, "baz")
|
|
# digest with param resets the org digester.
|
|
assert_equal(exp_1st, obj)
|
|
end
|
|
end
|
|
|
|
def test_digest_hexdigest_bang
|
|
[:digest!, :hexdigest!].each do |m|
|
|
exp_1st = "\3"; exp_1st = Digest.hexencode(exp_1st) if m == :hexdigest!
|
|
digester = @MyDigest.new
|
|
digester.update("foo")
|
|
obj = digester.send(m) # digest! always resets the org digester.
|
|
assert_equal(exp_1st, obj)
|
|
digester.update("bar")
|
|
obj = digester.send(m)
|
|
assert_equal(exp_1st, obj)
|
|
end
|
|
end
|
|
|
|
def test_to_s
|
|
digester = @MyDigest.new
|
|
digester.update("foo")
|
|
assert_equal("03", digester.to_s)
|
|
end
|
|
|
|
def test_length
|
|
@MyDigest.class_eval do
|
|
def digest_length
|
|
2
|
|
end
|
|
end
|
|
digester = @MyDigest.new
|
|
assert_equal(2, digester.length)
|
|
assert_equal(2, digester.size)
|
|
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_raise(RuntimeError) do
|
|
@MyDigest.new.block_length
|
|
end
|
|
end
|
|
end
|