mirror of
https://github.com/maybe-finance/maybe.git
synced 2025-07-18 20:59:39 +02:00
e4ee06c9f6
* Setup health check * Security health checker cron * Use resolver throughout codebase * Use resolver for trade builder * Add security health checks to schedule * Handle no provider * Lint fixes
64 lines
2 KiB
Ruby
64 lines
2 KiB
Ruby
require "test_helper"
|
|
require "ostruct"
|
|
|
|
class Security::PriceTest < ActiveSupport::TestCase
|
|
include ProviderTestHelper
|
|
|
|
setup do
|
|
@provider = mock
|
|
Security.stubs(:provider).returns(@provider)
|
|
|
|
@security = securities(:aapl)
|
|
end
|
|
|
|
test "finds single security price in DB" do
|
|
@provider.expects(:fetch_security_price).never
|
|
price = security_prices(:one)
|
|
|
|
assert_equal price, @security.find_or_fetch_price(date: price.date)
|
|
end
|
|
|
|
test "caches prices from provider to DB" do
|
|
price_date = 10.days.ago.to_date
|
|
|
|
expected_price = Security::Price.new(
|
|
security: @security,
|
|
date: price_date,
|
|
price: 314.34,
|
|
currency: "USD"
|
|
)
|
|
|
|
expect_provider_price(security: @security, price: expected_price, date: price_date)
|
|
|
|
assert_difference "Security::Price.count", 1 do
|
|
fetched_price = @security.find_or_fetch_price(date: price_date, cache: true)
|
|
assert_equal expected_price.price, fetched_price.price
|
|
end
|
|
end
|
|
|
|
test "returns nil if no price found in DB or from provider" do
|
|
security = securities(:aapl)
|
|
Security::Price.delete_all # Clear any existing prices
|
|
|
|
with_provider_response = provider_error_response(StandardError.new("Test error"))
|
|
|
|
@provider.expects(:fetch_security_price)
|
|
.with(symbol: security.ticker, exchange_operating_mic: security.exchange_operating_mic, date: Date.current)
|
|
.returns(with_provider_response)
|
|
|
|
assert_not @security.find_or_fetch_price(date: Date.current)
|
|
end
|
|
|
|
private
|
|
def expect_provider_price(security:, price:, date:)
|
|
@provider.expects(:fetch_security_price)
|
|
.with(symbol: security.ticker, exchange_operating_mic: security.exchange_operating_mic, date: date)
|
|
.returns(provider_success_response(price))
|
|
end
|
|
|
|
def expect_provider_prices(security:, prices:, start_date:, end_date:)
|
|
@provider.expects(:fetch_security_prices)
|
|
.with(security, start_date: start_date, end_date: end_date)
|
|
.returns(provider_success_response(prices))
|
|
end
|
|
end
|