2008-11-25 14:27:54 -05:00
|
|
|
require 'abstract_unit'
|
2010-05-01 08:40:02 -04:00
|
|
|
|
|
|
|
begin
|
|
|
|
require 'openssl'
|
|
|
|
OpenSSL::Digest::SHA1
|
|
|
|
rescue LoadError, NameError
|
|
|
|
$stderr.puts "Skipping MessageEncryptor test: broken OpenSSL install"
|
|
|
|
else
|
|
|
|
|
2009-11-14 14:37:06 -05:00
|
|
|
require 'active_support/time'
|
2008-11-25 14:27:54 -05:00
|
|
|
|
|
|
|
class MessageEncryptorTest < Test::Unit::TestCase
|
|
|
|
def setup
|
2011-05-23 07:02:06 -04:00
|
|
|
@encryptor = ActiveSupport::MessageEncryptor.new(SecureRandom.hex(64))
|
2010-03-29 01:51:48 -04:00
|
|
|
@data = { :some => "data", :now => Time.local(2010) }
|
2008-11-25 14:27:54 -05:00
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2008-11-25 14:27:54 -05:00
|
|
|
def test_simple_round_tripping
|
|
|
|
message = @encryptor.encrypt(@data)
|
|
|
|
assert_equal @data, @encryptor.decrypt(message)
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2008-11-25 14:27:54 -05:00
|
|
|
def test_encrypting_twice_yields_differing_cipher_text
|
|
|
|
first_messqage = @encryptor.encrypt(@data)
|
|
|
|
second_message = @encryptor.encrypt(@data)
|
|
|
|
assert_not_equal first_messqage, second_message
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2008-11-25 14:27:54 -05:00
|
|
|
def test_messing_with_either_value_causes_failure
|
|
|
|
text, iv = @encryptor.encrypt(@data).split("--")
|
|
|
|
assert_not_decrypted([iv, text] * "--")
|
|
|
|
assert_not_decrypted([text, munge(iv)] * "--")
|
|
|
|
assert_not_decrypted([munge(text), iv] * "--")
|
|
|
|
assert_not_decrypted([munge(text), munge(iv)] * "--")
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2008-11-25 14:27:54 -05:00
|
|
|
def test_signed_round_tripping
|
|
|
|
message = @encryptor.encrypt_and_sign(@data)
|
|
|
|
assert_equal @data, @encryptor.decrypt_and_verify(message)
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
|
|
|
|
2008-11-25 14:27:54 -05:00
|
|
|
private
|
|
|
|
def assert_not_decrypted(value)
|
2009-03-08 16:11:58 -04:00
|
|
|
assert_raise(ActiveSupport::MessageEncryptor::InvalidMessage) do
|
2008-11-25 14:27:54 -05:00
|
|
|
@encryptor.decrypt(value)
|
|
|
|
end
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2008-11-25 14:27:54 -05:00
|
|
|
def munge(base64_string)
|
|
|
|
bits = ActiveSupport::Base64.decode64(base64_string)
|
|
|
|
bits.reverse!
|
|
|
|
ActiveSupport::Base64.encode64s(bits)
|
|
|
|
end
|
|
|
|
end
|
2010-05-01 08:40:02 -04:00
|
|
|
|
|
|
|
end
|