1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
require 'abstract_unit'
begin
require 'openssl'
OpenSSL::PKCS5
rescue LoadError, NameError
$stderr.puts "Skipping KeyGenerator test: broken OpenSSL install"
else
require 'active_support/time'
require 'active_support/json'
class KeyGeneratorTest < ActiveSupport::TestCase
def setup
@secret = SecureRandom.hex(64)
@generator = ActiveSupport::KeyGenerator.new(@secret, :iterations=>2)
end
test "Generating a key of the default length" do
derived_key = @generator.generate_key("some_salt")
assert_kind_of String, derived_key
assert_equal OpenSSL::Digest::SHA1.new.block_length, derived_key.length, "Should have generated a key of the default size"
end
test "Generating a key of an alternative length" do
derived_key = @generator.generate_key("some_salt", 32)
assert_kind_of String, derived_key
assert_equal 32, derived_key.length, "Should have generated a key of the right size"
end
end
class CachingKeyGeneratorTest < ActiveSupport::TestCase
def setup
@secret = SecureRandom.hex(64)
@generator = ActiveSupport::KeyGenerator.new(@secret, :iterations=>2)
@caching_generator = ActiveSupport::CachingKeyGenerator.new(@generator)
end
test "Generating a cached key for same salt and key size" do
derived_key = @caching_generator.generate_key("some_salt", 32)
cached_key = @caching_generator.generate_key("some_salt", 32)
assert_equal derived_key, cached_key
assert_equal derived_key.object_id, cached_key.object_id
end
test "Does not cache key for different salt" do
derived_key = @caching_generator.generate_key("some_salt", 32)
different_salt_key = @caching_generator.generate_key("other_salt", 32)
assert_not_equal derived_key, different_salt_key
end
test "Does not cache key for different length" do
derived_key = @caching_generator.generate_key("some_salt", 32)
different_length_key = @caching_generator.generate_key("some_salt", 64)
assert_not_equal derived_key, different_length_key
end
end
end
|
