mirror of
https://github.com/we-promise/sure.git
synced 2026-04-18 03:24:09 +00:00
eac5d5e663
* Placeholder logic for missing prices * Generate holdings properly for "offline" securities * Separate forward and reverse calculators for holdings and balances * Remove unnecessary currency conversion during sync * Clearer sync process * Move price caching logic to dedicated model * Base holding calculator * Base calculator for balances * Finish balance calculators * Better naming * Logs cleanup * Remove stale data type * Remove stale test * Fix price lookup logic for holdings sync * Fix Plaid item sync regression * Remove temp logging * Calculate cash and holdings series * Add holdings, cash, and balance series dropdown for investments
64 lines
2.6 KiB
Ruby
64 lines
2.6 KiB
Ruby
require "test_helper"
|
|
|
|
class Account::Holding::PortfolioCacheTest < ActiveSupport::TestCase
|
|
include Account::EntriesTestHelper
|
|
|
|
setup do
|
|
# Prices, highest to lowest priority
|
|
@db_price = 210
|
|
@provider_price = 220
|
|
@trade_price = 200
|
|
@holding_price = 250
|
|
|
|
@account = families(:empty).accounts.create!(name: "Test Brokerage", balance: 10000, currency: "USD", accountable: Investment.new)
|
|
@test_security = Security.create!(name: "Test Security", ticker: "TEST")
|
|
|
|
@trade = create_trade(@test_security, account: @account, qty: 1, date: Date.current, price: @trade_price)
|
|
@holding = Account::Holding.create!(security: @test_security, account: @account, date: Date.current, qty: 1, price: @holding_price, amount: @holding_price, currency: "USD")
|
|
Security::Price.create!(security: @test_security, date: Date.current, price: @db_price)
|
|
end
|
|
|
|
test "gets price from DB if available" do
|
|
cache = Account::Holding::PortfolioCache.new(@account)
|
|
|
|
assert_equal @db_price, cache.get_price(@test_security.id, Date.current).price
|
|
end
|
|
|
|
test "if no price in DB, try fetching from provider" do
|
|
Security::Price.destroy_all
|
|
Security::Price.expects(:find_prices)
|
|
.with(security: @test_security, start_date: @account.start_date, end_date: Date.current)
|
|
.returns([
|
|
Security::Price.new(security: @test_security, date: Date.current, price: @provider_price, currency: "USD")
|
|
])
|
|
|
|
cache = Account::Holding::PortfolioCache.new(@account)
|
|
|
|
assert_equal @provider_price, cache.get_price(@test_security.id, Date.current).price
|
|
end
|
|
|
|
test "if no price from db or provider, try getting the price from trades" do
|
|
Security::Price.destroy_all # No DB prices
|
|
Security::Price.expects(:find_prices)
|
|
.with(security: @test_security, start_date: @account.start_date, end_date: Date.current)
|
|
.returns([]) # No provider prices
|
|
|
|
cache = Account::Holding::PortfolioCache.new(@account)
|
|
|
|
assert_equal @trade_price, cache.get_price(@test_security.id, Date.current).price
|
|
end
|
|
|
|
test "if no price from db, provider, or trades, search holdings" do
|
|
Security::Price.destroy_all # No DB prices
|
|
Security::Price.expects(:find_prices)
|
|
.with(security: @test_security, start_date: @account.start_date, end_date: Date.current)
|
|
.returns([]) # No provider prices
|
|
|
|
@account.entries.destroy_all # No prices from trades
|
|
|
|
cache = Account::Holding::PortfolioCache.new(@account, use_holdings: true)
|
|
|
|
assert_equal @holding_price, cache.get_price(@test_security.id, Date.current).price
|
|
end
|
|
end
|