Contents

require File.dirname(__FILE__) + '/test_helper'

DataMapper.setup(:default, "sqlite3::memory:")

class Client
  include DataMapper::Resource
  
  property :id, Serial
  property :encrypted_email, String
  property :encrypted_credentials, Text
  property :salt, String
  
  attr_encrypted :email, :key => 'a secret key'
  attr_encrypted :credentials, :key => Proc.new { |client| Huberry::Encryptor.encrypt(:value => client.salt, :key => 'some private key') }
  
  def initialize(attrs = {})
    super attrs
    self.salt ||= Digest::SHA1.hexdigest((Time.now.to_i * rand(5)).to_s)
    self.credentials ||= { :username => 'example', :password => 'test' }
  end
end

DataMapper.auto_migrate!

class DataMapperTest < Test::Unit::TestCase
  
  def setup
    Client.all.each(&:destroy)
  end
  
  def test_should_encrypt_email
    @person = Client.new :email => '[e-mail]'
    assert @person.save
    assert_not_nil @person.encrypted_email
    assert_not_equal @person.email, @person.encrypted_email
    assert_equal @person.email, Client.first.email
  end
  
  def test_should_marshal_and_encrypt_credentials
    @person = Client.new
    assert @person.save
    assert_not_nil @person.encrypted_credentials
    assert_not_equal @person.credentials, @person.encrypted_credentials
    assert_equal @person.credentials, Client.first.credentials
    assert Client.first.credentials.is_a?(Hash)
  end
  
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
shuber-attr_encrypted-1.0.1	test/data_mapper_test.rb
shuber-attr_encrypted-1.0.2	test/data_mapper_test.rb
shuber-attr_encrypted-1.0.3	test/data_mapper_test.rb
shuber-attr_encrypted-1.0.4	test/data_mapper_test.rb