mirror of
https://github.com/maybe-finance/maybe.git
synced 2025-07-25 08:09:38 +02:00
8db95623cf
* Handle reverse calculator starting portfolio generation correctly * Fix current_holdings to handle different dates and hide zero quantities - Use DISTINCT ON to get most recent holding per security instead of assuming same date - Filter out zero quantity holdings from UI display - Maintain cash display regardless of zero balance - Use single efficient query with proper Rails syntax * Continue to process holdings even if one is not resolvable * Lint fixes
50 lines
1.9 KiB
Ruby
50 lines
1.9 KiB
Ruby
require "test_helper"
|
|
|
|
class Holding::PortfolioSnapshotTest < ActiveSupport::TestCase
|
|
include EntriesTestHelper
|
|
setup do
|
|
@account = accounts(:investment)
|
|
@aapl = securities(:aapl)
|
|
@msft = securities(:msft)
|
|
end
|
|
|
|
test "captures the most recent holding quantities for each security" do
|
|
# Clear any existing data
|
|
@account.holdings.destroy_all
|
|
@account.entries.destroy_all
|
|
|
|
# Create some trades to establish which securities are in the portfolio
|
|
create_trade(@aapl, account: @account, qty: 10, price: 100, date: 5.days.ago)
|
|
create_trade(@msft, account: @account, qty: 30, price: 200, date: 5.days.ago)
|
|
|
|
# Create holdings for AAPL at different dates
|
|
@account.holdings.create!(security: @aapl, date: 3.days.ago, qty: 10, price: 100, amount: 1000, currency: "USD")
|
|
@account.holdings.create!(security: @aapl, date: 1.day.ago, qty: 20, price: 150, amount: 3000, currency: "USD")
|
|
|
|
# Create holdings for MSFT at different dates
|
|
@account.holdings.create!(security: @msft, date: 5.days.ago, qty: 30, price: 200, amount: 6000, currency: "USD")
|
|
@account.holdings.create!(security: @msft, date: 2.days.ago, qty: 40, price: 250, amount: 10000, currency: "USD")
|
|
|
|
snapshot = Holding::PortfolioSnapshot.new(@account)
|
|
portfolio = snapshot.to_h
|
|
|
|
assert_equal 2, portfolio.size
|
|
assert_equal 20, portfolio[@aapl.id]
|
|
assert_equal 40, portfolio[@msft.id]
|
|
end
|
|
|
|
test "includes securities from trades with zero quantities when no holdings exist" do
|
|
# Clear any existing data
|
|
@account.holdings.destroy_all
|
|
@account.entries.destroy_all
|
|
|
|
# Create a trade to establish AAPL is in the portfolio
|
|
create_trade(@aapl, account: @account, qty: 10, price: 100, date: 5.days.ago)
|
|
|
|
snapshot = Holding::PortfolioSnapshot.new(@account)
|
|
portfolio = snapshot.to_h
|
|
|
|
assert_equal 1, portfolio.size
|
|
assert_equal 0, portfolio[@aapl.id]
|
|
end
|
|
end
|