Sha256: c6a45684ff1c2fb988aff7fa0a0135ff44ddefb3ba09b30b5cc19a14b74ac7ee

Contents?: true

Size: 1.77 KB

Versions: 4

Compression:

Stored size: 1.77 KB

Contents

# frozen_string_literal: true

require_relative 'test_helper'

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

class LegacyClient
  include DataMapper::Resource
  self.attr_encrypted_options[:insecure_mode] = true
  self.attr_encrypted_options[:algorithm] = 'aes-256-cbc'
  self.attr_encrypted_options[:mode] = :single_iv_and_salt

  property :id, Serial
  property :encrypted_email, String
  property :encrypted_credentials, Text
  property :salt, String

  attr_encrypted :email, :key => 'a secret key', mode: :single_iv_and_salt
  attr_encrypted :credentials, :key => Proc.new { |client| Encryptor.encrypt(:value => client.salt, :key => 'some private key', insecure_mode: true, algorithm: 'aes-256-cbc') }, :marshal => true, mode: :single_iv_and_salt

  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 LegacyDataMapperTest < Minitest::Test

  def setup
    LegacyClient.all.each(&:destroy)
  end

  def test_should_encrypt_email
    @client = LegacyClient.new :email => 'test@example.com'
    assert @client.save
    refute_nil @client.encrypted_email
    refute_equal @client.email, @client.encrypted_email
    assert_equal @client.email, LegacyClient.first.email
  end

  def test_should_marshal_and_encrypt_credentials
    @client = LegacyClient.new
    assert @client.save
    refute_nil @client.encrypted_credentials
    refute_equal @client.credentials, @client.encrypted_credentials
    assert_equal @client.credentials, LegacyClient.first.credentials
    assert LegacyClient.first.credentials.is_a?(Hash)
  end

  def test_should_encode_by_default
    assert LegacyClient.attr_encrypted_options[:encode]
  end

end

Version data entries

4 entries across 4 versions & 2 rubygems

Version Path
attr_encrypted-4.0.0 test/legacy_data_mapper_test.rb
powerhome-attr_encrypted-1.2.0 test/legacy_data_mapper_test.rb
powerhome-attr_encrypted-1.1.0 test/legacy_data_mapper_test.rb
powerhome-attr_encrypted-1.0.1 test/legacy_data_mapper_test.rb