forked from binarylogic/authlogic
-
Notifications
You must be signed in to change notification settings - Fork 1
/
random_test.rb
49 lines (41 loc) · 1.56 KB
/
random_test.rb
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
require File.dirname(__FILE__) + '/test_helper.rb'
class RandomTest < ActiveSupport::TestCase
def test_random_tokens_have_consisten_length
with_any_random do
assert_equal 128, Authlogic::Random.hex_token.length
assert_equal 20, Authlogic::Random.friendly_token.length
end
end
def test_random_tokens_are_indeed_random
# this might fail if you are *really* unlucky :)
with_any_random do
assert_not_equal Authlogic::Random.hex_token, Authlogic::Random.hex_token
assert_not_equal Authlogic::Random.friendly_token, Authlogic::Random.friendly_token
end
end
private
def with_any_random(&block)
[true, false].each {|val| with_secure_random_enabled(val, &block)}
end
def with_secure_random_enabled(enabled = true)
# can't really test SecureRandom if we don't have an implementation
return if enabled && !Authlogic::Random::SecureRandom
current_sec_rand = Authlogic::Random::SecureRandom
reload_authlogic_with_sec_random!(current_sec_rand, enabled)
yield
ensure
reload_authlogic_with_sec_random!(current_sec_rand)
end
def reload_authlogic_with_sec_random!(secure_random, enabled = true)
silence_warnings do
secure_random.parent.const_set(secure_random.name.sub("#{secure_random.parent}::", ''), enabled ? secure_random : nil)
load(File.dirname(__FILE__) + '/../lib/authlogic/random.rb')
end
end
def silence_warnings
old_verbose, $VERBOSE = $VERBOSE, nil
yield
ensure
$VERBOSE = old_verbose
end
end