Files
quantconnect--lean/Tests/Brokerages/LevelOneOrderBook/LevelOneMarketDataTests.cs
T
2026-07-13 13:02:50 +08:00

241 lines
11 KiB
C#

/*
* QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
* Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
using System;
using NUnit.Framework;
using QuantConnect.Data.Market;
using QuantConnect.Brokerages.LevelOneOrderBook;
namespace QuantConnect.Tests.Brokerages.LevelOneOrderBook
{
[TestFixture]
public class LevelOneMarketDataTests
{
private readonly Symbol _aapl = Symbols.AAPL;
private readonly DateTime _mockDateTime = new DateTime(2026, 06, 20);
[TestCase(true)]
[TestCase(false)]
public void LevelOneMarketDataShouldNotRaiseEventWhenTimestampIsInvalid(bool useInvalidTimestamp)
{
var invalidTimestamp = useInvalidTimestamp ? default(DateTime?) : new DateTime();
var levelOneMarketData = new LevelOneMarketData(_aapl);
var eventRaised = default(bool);
levelOneMarketData.BaseDataReceived += (_, e) => eventRaised = true;
levelOneMarketData.UpdateLastTrade(invalidTimestamp, null, null);
levelOneMarketData.UpdateQuote(invalidTimestamp, null, null, null, null);
levelOneMarketData.UpdateOpenInterest(invalidTimestamp, null);
Assert.IsFalse(eventRaised, "BaseDataReceived should not be raised with invalid timestamp.");
}
[TestCase(false, null)]
[TestCase(true, 0)]
[TestCase(true, 100)]
public void UpdateOpenInterestShouldRaiseEventBasedOnValuePresence(bool shouldRaiseEvent, decimal? openInterest)
{
var levelOneMarketData = new LevelOneMarketData(_aapl);
var eventRaised = default(bool);
levelOneMarketData.BaseDataReceived += (_, e) => eventRaised = true;
levelOneMarketData.UpdateOpenInterest(_mockDateTime, openInterest);
Assert.AreEqual(shouldRaiseEvent, eventRaised, $"Expected event to {(shouldRaiseEvent ? "" : "not ")}be raised for open interest value: {openInterest?.ToStringInvariant() ?? "null"}");
}
[TestCase(false, null, null)]
[TestCase(false, 1, null)]
[TestCase(true, null, 1)]
[TestCase(true, 1, 1)]
public void UpdateLastTradeShouldRaiseEventBasedOnPriceAndQuantity(bool shouldRaiseEvent, decimal? lastQuantity, decimal? lastPrice)
{
var levelOneMarketData = new LevelOneMarketData(_aapl);
var eventRaised = default(bool);
levelOneMarketData.BaseDataReceived += (_, e) => eventRaised = true;
levelOneMarketData.UpdateLastTrade(_mockDateTime, lastQuantity, lastPrice);
Assert.AreEqual(shouldRaiseEvent, eventRaised, $"Expected event {(shouldRaiseEvent ? "" : "not ")}to be raised for (Qty={lastQuantity}, Price={lastPrice}).");
}
[TestCase(true, 1, 1, 2, 2)]
[TestCase(true, 1, 1, null, 2)]
[TestCase(true, 1, 1, 2, null)]
[TestCase(false, 1, 1, null, null)]
public void UpdateLastTradeShouldTrackLatestTradeCorrectlyOnMultipleUpdates(bool shouldRaiseEvent, decimal? firstLastQuantity, decimal? firstLastPrice, decimal? secondLastQuantity, decimal? secondLastPrice)
{
var levelOneMarketData = new LevelOneMarketData(_aapl);
var eventRaised = default(bool);
levelOneMarketData.BaseDataReceived += (_, e) => eventRaised = true;
levelOneMarketData.UpdateLastTrade(_mockDateTime, firstLastQuantity, firstLastPrice);
Assert.IsTrue(eventRaised, "Expected event to be raised on first update.");
Assert.AreEqual(firstLastQuantity, levelOneMarketData.LastTradeSize, "First update: unexpected LastTradeSize.");
Assert.AreEqual(firstLastPrice, levelOneMarketData.LastTradePrice, "First update: unexpected LastTradePrice.");
// Reset event flag for second update
eventRaised = false;
levelOneMarketData.UpdateLastTrade(_mockDateTime, secondLastQuantity, secondLastPrice);
Assert.AreEqual(shouldRaiseEvent, eventRaised, $"Second update: Expected event {(shouldRaiseEvent ? "" : "not ")}to be raised.");
var expectedSize = secondLastQuantity ?? firstLastQuantity;
var expectedPrice = secondLastPrice ?? firstLastPrice;
Assert.AreEqual(expectedSize, levelOneMarketData.LastTradeSize, "Final LastTradeSize mismatch.");
Assert.AreEqual(expectedPrice, levelOneMarketData.LastTradePrice, "Final LastTradePrice mismatch.");
}
[TestCase(false, null, null, null, null)]
[TestCase(true, 1, null, 1, null)]
[TestCase(true, 1, 0, null, null)]
[TestCase(true, null, null, 1, 0)]
[TestCase(true, 1, 1, 1, 0)]
public void UpdateQuoteShouldRaiseEventBasedOnAskAndBid(bool shouldRaiseEvent, decimal? bidPrice, decimal? bidSize, decimal? askPrice, decimal? askSize)
{
var levelOneMarketData = new LevelOneMarketData(_aapl);
var eventRaised = default(bool);
levelOneMarketData.BaseDataReceived += (_, e) => eventRaised = true;
levelOneMarketData.UpdateQuote(_mockDateTime, bidPrice, bidSize, askPrice, askSize);
Assert.AreEqual(shouldRaiseEvent, eventRaised);
}
[TestCase(false, 1, 1, 1, 1, null, null, null, null)]
[TestCase(false, 1, 1, 1, 1, 1, 1, 1, 1)]
[TestCase(true, 1, 1, 1, 1, 2, 2, 2, 2)]
public void UpdateQuoteShouldTrackQuoteCorrectlyOnMultipleUpdates(
bool shouldRaiseEvent,
decimal? firstBidPrice, decimal? firstBidSize, decimal? firstAskPrice, decimal? firstAskSize,
decimal? secondBidPrice, decimal? secondBidSize, decimal? secondAskPrice, decimal? secondAskSize)
{
var levelOneMarketData = new LevelOneMarketData(_aapl);
var eventRaised = default(bool);
levelOneMarketData.BaseDataReceived += (_, e) => eventRaised = true;
levelOneMarketData.UpdateQuote(_mockDateTime, firstBidPrice, firstBidSize, firstAskPrice, firstAskSize);
Assert.IsTrue(eventRaised);
// Reset event flag for second update
eventRaised = false;
levelOneMarketData.UpdateQuote(_mockDateTime, secondBidPrice, secondBidSize, secondAskPrice, secondAskSize);
Assert.AreEqual(shouldRaiseEvent, eventRaised, $"Second update: Expected event {(shouldRaiseEvent ? "" : "not ")}to be raised.");
}
[TestCase(true, 2, 0, 2, 0)]
[TestCase(false, 2, 0, 2, 0)]
public void UpdateQuoteShouldIgnoreZeroSizeUpdatesCorrectly(bool ignoreZeroSizeUpdates, decimal? bidPrice, decimal? bidSize, decimal? askPrice, decimal? askSize)
{
var levelOneMarketData = new LevelOneMarketData(_aapl)
{
IgnoreZeroSizeUpdates = ignoreZeroSizeUpdates
};
var expectedBestBidSize = 1;
var expectedBestAskSize = 1;
levelOneMarketData.UpdateQuote(_mockDateTime, 1, expectedBestBidSize, 1, expectedBestAskSize);
levelOneMarketData.UpdateQuote(_mockDateTime, bidPrice, bidSize, askPrice, askSize);
Assert.AreEqual(bidPrice, levelOneMarketData.BestBidPrice, "Bid price should update.");
Assert.AreEqual(askPrice, levelOneMarketData.BestAskPrice, "Ask price should update.");
if (ignoreZeroSizeUpdates)
{
Assert.AreEqual(expectedBestAskSize, levelOneMarketData.BestAskSize, "Bid size should remain unchanged when ignoring zero size.");
Assert.AreEqual(expectedBestBidSize, levelOneMarketData.BestBidSize, "Ask size should remain unchanged when ignoring zero size.");
}
else
{
Assert.AreEqual(bidSize, levelOneMarketData.BestAskSize, "Bid size should be overwritten with 0.");
Assert.AreEqual(askSize, levelOneMarketData.BestBidSize, "Ask size should be overwritten with 0.");
}
}
[TestCase("T", "NASDAQ")]
[TestCase("Z", "BATS")]
[TestCase("N", "NYSE")]
[TestCase("P", "ARCA")]
[TestCase("", "")]
[TestCase(null, "")]
public void LevelOneMarketDataShouldMapExchangeStringOnPublishedTicks(string exchange, string expectedExchange)
{
var levelOneMarketData = new LevelOneMarketData(_aapl);
var quoteTick = default(Tick);
var tradeTick = default(Tick);
levelOneMarketData.BaseDataReceived += (_, e) =>
{
var tick = (Tick)e.BaseData;
if (tick.TickType == TickType.Quote)
{
quoteTick = tick;
}
else if (tick.TickType == TickType.Trade)
{
tradeTick = tick;
}
};
levelOneMarketData.UpdateQuote(_mockDateTime, 1, 1, 1, 1, saleCondition: string.Empty, exchange: exchange);
levelOneMarketData.UpdateLastTrade(_mockDateTime, 1, 1, exchange: exchange);
Assert.IsNotNull(quoteTick, "Expected BaseDataReceived to publish a quote tick.");
Assert.IsNotNull(tradeTick, "Expected BaseDataReceived to publish a trade tick.");
Assert.AreEqual(expectedExchange, quoteTick.Exchange, "Quote tick exchange mismatch.");
Assert.AreEqual(expectedExchange, tradeTick.Exchange, "Trade tick exchange mismatch.");
}
[TestCase(true, 0, 2)]
[TestCase(false, 0, 2)]
public void UpdateLastTradeShouldIgnoreZeroSizeUpdatesCorrectly(bool ignoreZeroSizeUpdates, decimal? lastQuantity, decimal? lastPrice)
{
var levelOneMarketData = new LevelOneMarketData(_aapl)
{
IgnoreZeroSizeUpdates = ignoreZeroSizeUpdates
};
var expectedLastQuantity = 1;
levelOneMarketData.UpdateLastTrade(_mockDateTime, expectedLastQuantity, 1);
levelOneMarketData.UpdateLastTrade(_mockDateTime, lastQuantity, lastPrice);
Assert.AreEqual(lastPrice, levelOneMarketData.LastTradePrice, "LastTradePrice should update.");
if (ignoreZeroSizeUpdates)
{
Assert.AreEqual(expectedLastQuantity, levelOneMarketData.LastTradeSize);
}
else
{
Assert.AreEqual(lastQuantity, levelOneMarketData.LastTradeSize);
}
}
}
}