Contents

require File.expand_path('../test_helper', __FILE__)

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

class Client
  include DataMapper::Resource

  property :id, Serial
  property :encrypted_email, String
  property :encrypted_email_iv, String
  property :encrypted_email_salt, String

  property :encrypted_credentials, Text
  property :encrypted_credentials_iv, Text
  property :encrypted_credentials_salt, Text

  self.attr_encrypted_options[:mode] = :per_attribute_iv_and_salt

  attr_encrypted :email, :key => SECRET_KEY
  attr_encrypted :credentials, :key => SECRET_KEY, :marshal => true

  def initialize(attrs = {})
    super attrs
    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
    @client = Client.new :email => '[e-mail]'
    assert @client.save
    assert_not_nil @client.encrypted_email
    assert_not_equal @client.email, @client.encrypted_email
    assert_equal @client.email, Client.first.email
  end

  def test_should_marshal_and_encrypt_credentials
    @client = Client.new
    assert @client.save
    assert_not_nil @client.encrypted_credentials
    assert_not_equal @client.credentials, @client.encrypted_credentials
    assert_equal @client.credentials, Client.first.credentials
    assert Client.first.credentials.is_a?(Hash)
  end

  def test_should_encode_by_default
    assert Client.attr_encrypted_options[:encode]
  end

end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
attr_encrypted-1.3.3	test/data_mapper_test.rb
attr_encrypted-1.3.2	test/data_mapper_test.rb
attr_encrypted-1.3.1	test/data_mapper_test.rb
attr_encrypted-1.3.0	test/data_mapper_test.rb