path: root/activerecord/test/cases/adapters/postgresql/money_test.rb

# encoding: utf-8
require "cases/helper"
require 'support/schema_dumping_helper'

class PostgresqlMoneyTest < ActiveRecord::TestCase
  include SchemaDumpingHelper

  class PostgresqlMoney < ActiveRecord::Base; end

  setup do
    @connection = ActiveRecord::Base.connection
    @connection.execute("set lc_monetary = 'C'")
    @connection.create_table('postgresql_moneys', force: true) do |t|
      t.column "wealth", "money"
      t.column "depth", "money", default: "150.55"
    end
  end

  teardown do
    @connection.execute 'DROP TABLE IF EXISTS postgresql_moneys'
  end

  def test_column
    column = PostgresqlMoney.columns_hash["wealth"]
    assert_equal :money, column.type
    assert_equal "money", column.sql_type
    assert_equal 2, column.scale
    assert_not column.array?

    type = PostgresqlMoney.type_for_attribute("wealth")
    assert type.number?
    assert_not type.binary?
  end

  def test_default
    assert_equal BigDecimal.new("150.55"), PostgresqlMoney.column_defaults['depth']
    assert_equal BigDecimal.new("150.55"), PostgresqlMoney.new.depth
  end

  def test_money_values
    @connection.execute("INSERT INTO postgresql_moneys (id, wealth) VALUES (1, '567.89'::money)")
    @connection.execute("INSERT INTO postgresql_moneys (id, wealth) VALUES (2, '-567.89'::money)")

    first_money = PostgresqlMoney.find(1)
    second_money = PostgresqlMoney.find(2)
    assert_equal 567.89, first_money.wealth
    assert_equal(-567.89, second_money.wealth)
  end

  def test_money_type_cast
    type = PostgresqlMoney.type_for_attribute('wealth')
    assert_equal(12345678.12, type.type_cast_from_user("$12,345,678.12"))
    assert_equal(12345678.12, type.type_cast_from_user("$12.345.678,12"))
    assert_equal(-1.15, type.type_cast_from_user("-$1.15"))
    assert_equal(-2.25, type.type_cast_from_user("($2.25)"))
  end

  def test_schema_dumping
    output = dump_table_schema("postgresql_moneys")
    assert_match %r{t\.money\s+"wealth",\s+scale: 2$}, output
    assert_match %r{t\.money\s+"depth",\s+scale: 2,\s+default: 150\.55$}, output
  end

  def test_create_and_update_money
    money = PostgresqlMoney.create(wealth: "987.65")
    assert_equal 987.65, money.wealth

    new_value = BigDecimal.new('123.45')
    money.wealth = new_value
    money.save!
    money.reload
    assert_equal new_value, money.wealth
  end

  def test_update_all_with_money_string
    money = PostgresqlMoney.create!
    PostgresqlMoney.update_all(wealth: "987.65")
    money.reload

    assert_equal 987.65, money.wealth
  end

  def test_update_all_with_money_big_decimal
    money = PostgresqlMoney.create!
    PostgresqlMoney.update_all(wealth: '123.45'.to_d)
    money.reload

    assert_equal 123.45, money.wealth
  end

  def test_update_all_with_money_numeric
    money = PostgresqlMoney.create!
    PostgresqlMoney.update_all(wealth: 123.45)
    money.reload

    assert_equal 123.45, money.wealth
  end
end