Maybe/test/models/security/price_test.rb

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

    provider_response = provider_error_response(Provider::ProviderError.new("Test error"))

    @provider.expects(:fetch_security_price)
             .with(security, date: Date.current)
             .returns(provider_response)

    assert_not @security.find_or_fetch_price(date: Date.current)
  end

  test "upserts historical prices from provider" do
    Security::Price.delete_all

    # Will be overwritten by upsert
    Security::Price.create!(security: @security, date: 1.day.ago.to_date, price: 190, currency: "USD")

    expect_provider_prices(security: @security, start_date: 2.days.ago.to_date, end_date: Date.current, prices: [
      Security::Price.new(security: @security, date: Date.current, price: 215, currency: "USD"),
      Security::Price.new(security: @security, date: 1.day.ago.to_date, price: 214, currency: "USD"),
      Security::Price.new(security: @security, date: 2.days.ago.to_date, price: 213, currency: "USD")
    ])

    @security.sync_provider_prices(start_date: 2.days.ago.to_date)

    assert_equal 215, @security.prices.find_by(date: Date.current).price
    assert_equal 214, @security.prices.find_by(date: 1.day.ago.to_date).price
    assert_equal 213, @security.prices.find_by(date: 2.days.ago.to_date).price
  end

  private
    def expect_provider_price(security:, price:, date:)
      @provider.expects(:fetch_security_price)
               .with(security, date: date)
               .returns(provider_success_response(Security::Provideable::PriceData.new(price: 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(Security::Provideable::PricesData.new(prices: prices)))
    end
end