Files
quantconnect--lean/Tests/Common/Securities/SecurityServiceTests.cs
T
2026-07-13 13:02:50 +08:00

316 lines
15 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 System.Collections.Generic;
using System.Linq;
using Moq;
using NUnit.Framework;
using QuantConnect.Algorithm.CSharp;
using QuantConnect.Configuration;
using QuantConnect.Data;
using QuantConnect.Data.Auxiliary;
using QuantConnect.Data.Market;
using QuantConnect.Data.UniverseSelection;
using QuantConnect.Interfaces;
using QuantConnect.Securities;
using QuantConnect.Tests.Engine.DataFeeds;
using QuantConnect.ToolBox.RandomDataGenerator;
using QuantConnect.Util;
namespace QuantConnect.Tests.Common.Securities
{
public class SecurityServiceTests : ISecurityInitializerProvider
{
private ISecurityService _securityService;
private SubscriptionManager _subscriptionManager;
private MarketHoursDatabase _marketHoursDatabase;
public ISecurityInitializer SecurityInitializer => QuantConnect.Securities.SecurityInitializer.Null;
[SetUp]
public void Setup()
{
SymbolCache.Clear();
_subscriptionManager = new SubscriptionManager(NullTimeKeeper.Instance);
var dataManager = new DataManagerStub();
_subscriptionManager.SetDataManager(dataManager);
_marketHoursDatabase = MarketHoursDatabase.FromDataFolder();
_securityService = dataManager.SecurityService;
}
[TestCase("EURUSD", SecurityType.Forex, Market.FXCM)]
[TestCase("EURUSD", SecurityType.Forex, Market.Oanda)]
[TestCase("BTCUSD", SecurityType.Crypto, Market.Coinbase)]
public void CanCreate_ForexOrCrypto_WithCorrectSubscriptions(string ticker, SecurityType type, string market)
{
var symbol = Symbol.Create(ticker, type, market);
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(typeof(QuoteBar), symbol, Resolution.Second, false, false, false);
var actual = _securityService.CreateSecurity(symbol, configs, 1.0m, false);
Assert.AreEqual(actual.Subscriptions.Count(), 1);
Assert.AreEqual(actual.Subscriptions.First().Type, typeof(QuoteBar));
Assert.AreEqual(actual.Subscriptions.First().TickType, TickType.Quote);
}
[Test]
public void CanCreate_CanonicalOption_WithCorrectSubscriptions()
{
var optionSymbol = Symbol.Create("GOOG", SecurityType.Option, Market.USA);
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(typeof(OptionUniverse), optionSymbol, Resolution.Minute, false, false, false);
var option = _securityService.CreateSecurity(optionSymbol, configs, 1.0m, false);
Assert.AreEqual(option.Subscriptions.Count(), 1);
Assert.AreEqual(option.Subscriptions.First().Type, typeof(OptionUniverse));
Assert.AreEqual(option.Subscriptions.First().TickType, TickType.Quote);
}
[Test]
public void CanCreate_Equity_WithCorrectSubscriptions()
{
var equitySymbol = Symbol.Create("AAPL", SecurityType.Equity, Market.USA);
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(typeof(TradeBar), equitySymbol, Resolution.Second, false, false, false);
var equity = _securityService.CreateSecurity(equitySymbol, configs, 1.0m, false);
Assert.AreEqual(equity.Subscriptions.Count(), 1);
Assert.AreEqual(equity.Subscriptions.First().Type, typeof(TradeBar));
Assert.AreEqual(equity.Subscriptions.First().TickType, TickType.Trade);
}
[Test]
public void CanCreate_Cfd_WithCorrectSubscriptions()
{
var symbol = Symbol.Create("abc", SecurityType.Cfd, Market.USA);
_marketHoursDatabase.SetEntryAlwaysOpen(Market.USA, "abc", SecurityType.Cfd, TimeZones.NewYork);
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(typeof(QuoteBar), symbol, Resolution.Second, false, false, false);
var cfd = _securityService.CreateSecurity(symbol, configs, 1.0m, false);
Assert.AreEqual(cfd.Subscriptions.Count(), 1);
Assert.AreEqual(cfd.Subscriptions.First().Type, typeof(QuoteBar));
Assert.AreEqual(cfd.Subscriptions.First().TickType, TickType.Quote);
}
[Test]
public void CanCreate_CustomSecurities_WithCorrectSubscriptions()
{
var symbol = new Symbol(SecurityIdentifier.GenerateBase(null, "BTC", Market.USA), "BTC");
_marketHoursDatabase.SetEntryAlwaysOpen(Market.USA, "BTC", SecurityType.Base, TimeZones.NewYork);
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(typeof(LiveTradingFeaturesAlgorithm.Bitcoin), symbol, Resolution.Second, false, false, false);
var security = _securityService.CreateSecurity(symbol, configs, 1.0m, false);
Assert.AreEqual(security.Subscriptions.Count(), 1);
Assert.AreEqual(security.Subscriptions.First().Type, typeof(LiveTradingFeaturesAlgorithm.Bitcoin));
Assert.AreEqual(security.Subscriptions.First().TickType, TickType.Trade);
}
[Test]
public void ThrowOnCreateCryptoNotDescribedInCSV()
{
var symbol = Symbol.Create("ABCDEFG", SecurityType.Crypto, Market.Coinbase);
Assert.Throws<ArgumentException>(() =>
{
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(typeof(QuoteBar), symbol, Resolution.Minute, false, false, false);
var actual = _securityService.CreateSecurity(symbol, configs, 1.0m, false);
}, "Symbol can't be found in the Symbol Properties Database");
}
[Test]
public void CanCreate_ConcreteOptions_WithCorrectSubscriptions()
{
var optionSymbol = Symbol.CreateOption(Symbols.SPY, Market.USA, OptionStyle.European, OptionRight.Put, 195.50m,
new DateTime(2015, 09, 18));
var subscriptionTypes = new List<Tuple<Type, TickType>>
{
new Tuple<Type, TickType>(typeof(TradeBar), TickType.Trade),
new Tuple<Type, TickType>(typeof(QuoteBar), TickType.Quote),
new Tuple<Type, TickType>(typeof(OpenInterest), TickType.OpenInterest)
};
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(optionSymbol, Resolution.Minute, false, false, false, false, false, subscriptionTypes);
var security = _securityService.CreateSecurity(optionSymbol, configs, 1.0m, false);
Assert.IsFalse(optionSymbol.IsCanonical());
Assert.AreEqual(security.Subscriptions.Count(), 3);
Assert.IsTrue(security.Subscriptions.Any(x => x.TickType == TickType.OpenInterest && x.Type == typeof(OpenInterest)));
Assert.IsTrue(security.Subscriptions.Any(x => x.TickType == TickType.Quote && x.Type == typeof(QuoteBar)));
Assert.IsTrue(security.Subscriptions.Any(x => x.TickType == TickType.Trade && x.Type == typeof(TradeBar)));
}
[Test]
public void CanCreate_ConcreteFutures_WithCorrectSubscriptions()
{
var identifier = SecurityIdentifier.GenerateFuture(new DateTime(2020, 12, 15), "ED", Market.CME);
var symbol = new Symbol(identifier, "ED", Symbol.Empty);
var subscriptionTypes = new List<Tuple<Type, TickType>>
{
new Tuple<Type, TickType>(typeof(TradeBar), TickType.Trade),
new Tuple<Type, TickType>(typeof(QuoteBar), TickType.Quote),
new Tuple<Type, TickType>(typeof(OpenInterest), TickType.OpenInterest)
};
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(symbol, Resolution.Second, false, false, false, false, false, subscriptionTypes);
var security = _securityService.CreateSecurity(symbol, configs, 1.0m, false);
Assert.IsFalse(symbol.IsCanonical());
Assert.AreEqual(security.Subscriptions.Count(), 3);
Assert.IsTrue(security.Subscriptions.Any(x => x.TickType == TickType.OpenInterest && x.Type == typeof(OpenInterest)));
Assert.IsTrue(security.Subscriptions.Any(x => x.TickType == TickType.Quote && x.Type == typeof(QuoteBar)));
Assert.IsTrue(security.Subscriptions.Any(x => x.TickType == TickType.Trade && x.Type == typeof(TradeBar)));
}
[TestCase("BTGUSDT", SecurityType.CryptoFuture, Market.Binance)]
[TestCase("USDTEUR", SecurityType.Forex, Market.Oanda)]
public void CannotCreateSecurityWhenBaseCurrencyNotFound(string ticker, SecurityType securityType, string market)
{
var symbol = QuantConnect.Symbol.Create(ticker, securityType, market);
var subscriptionTypes = new List<Tuple<Type, TickType>>
{
new Tuple<Type, TickType>(typeof(TradeBar), TickType.Trade),
new Tuple<Type, TickType>(typeof(QuoteBar), TickType.Quote),
new Tuple<Type, TickType>(typeof(OpenInterest), TickType.OpenInterest)
};
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(symbol, Resolution.Second, false, false, false, false, false, subscriptionTypes);
Assert.Throws<ArgumentException>(() => _securityService.CreateSecurity(symbol, configs, 1.0m, false));
}
[Test]
public void CreatesEquityOptionWithContractMultiplierEqualsToContractUnitOfTrade()
{
var underlying = Symbol.Create("TWX", SecurityType.Equity, Market.USA);
var equityOption = Symbol.CreateOption(
underlying,
Market.USA,
OptionStyle.American,
OptionRight.Call,
320m,
new DateTime(2020, 12, 18));
var subscriptionTypes = new List<Tuple<Type, TickType>>
{
new Tuple<Type, TickType>(typeof(TradeBar), TickType.Trade),
new Tuple<Type, TickType>(typeof(QuoteBar), TickType.Quote),
new Tuple<Type, TickType>(typeof(OpenInterest), TickType.OpenInterest)
};
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(equityOption, Resolution.Minute, true, false, false, false, false, subscriptionTypes);
var equityOptionSecurity = (QuantConnect.Securities.Option.Option)_securityService.CreateSecurity(equityOption, configs, 1.0m);
Assert.AreEqual(100, equityOptionSecurity.ContractMultiplier);
Assert.AreEqual(100, equityOptionSecurity.ContractUnitOfTrade);
}
[Test]
public void CreatesFutureOptionWithContractMultiplierEqualsToFutureContractMultiplier()
{
var underlying = Symbol.CreateFuture(
QuantConnect.Securities.Futures.Indices.SP500EMini,
Market.CME,
new DateTime(2020, 12, 18));
var futureOption = Symbol.CreateOption(
underlying,
Market.CME,
OptionStyle.American,
OptionRight.Call,
3250m,
new DateTime(2020, 12, 18));
var subscriptionTypes = new List<Tuple<Type, TickType>>
{
new Tuple<Type, TickType>(typeof(TradeBar), TickType.Trade),
new Tuple<Type, TickType>(typeof(QuoteBar), TickType.Quote),
new Tuple<Type, TickType>(typeof(OpenInterest), TickType.OpenInterest)
};
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(futureOption, Resolution.Minute, true, false, false, false, false, subscriptionTypes);
var futureOptionSecurity = (QuantConnect.Securities.Option.Option)_securityService.CreateSecurity(futureOption, configs, 1.0m);
Assert.AreEqual(50, futureOptionSecurity.ContractMultiplier);
Assert.AreEqual(1, futureOptionSecurity.ContractUnitOfTrade);
}
[Test]
public void AddPrimaryExchangeToSecurityObject()
{
// Arrange
var equitySymbol = Symbol.Create("AAPL", SecurityType.Equity, Market.USA);
var mockedPrimaryExchangeProvider = new Mock<IPrimaryExchangeProvider>();
mockedPrimaryExchangeProvider.Setup(pep => pep.GetPrimaryExchange(equitySymbol.ID)).Returns(Exchange.NASDAQ);
var algorithm = new AlgorithmStub();
var securityService = new SecurityService(algorithm.Portfolio.CashBook,
MarketHoursDatabase.FromDataFolder(),
SymbolPropertiesDatabase.FromDataFolder(),
algorithm,
new RegisteredSecurityDataTypesProvider(),
new SecurityCacheProvider(algorithm.Portfolio),
mockedPrimaryExchangeProvider.Object,
algorithm: algorithm);
var configs = _subscriptionManager.SubscriptionDataConfigService.Add(typeof(TradeBar), equitySymbol, Resolution.Second, false, false, false);
// Act
var equity = securityService.CreateSecurity(equitySymbol, configs, 1.0m, false);
// Assert
Assert.AreEqual(equity.Subscriptions.Count(), 1);
Assert.AreEqual(equity.Subscriptions.First().Type, typeof(TradeBar));
Assert.AreEqual(equity.Subscriptions.First().TickType, TickType.Trade);
Assert.AreEqual(((QuantConnect.Securities.Equity.Equity)equity).PrimaryExchange, Exchange.NASDAQ);
}
[Test]
public void CreateSecurityDoesNotThrowWithNullAlgorithm()
{
var startDate = new DateTime(2024, 1, 1);
var securityManager = new SecurityManager(new TimeKeeper(startDate, new[] { TimeZones.Utc }));
var securityService = new SecurityService(
new CashBook(),
MarketHoursDatabase.FromDataFolder(),
SymbolPropertiesDatabase.FromDataFolder(),
new SecurityInitializerProvider(new FuncSecurityInitializer(security => { })),
RegisteredSecurityDataTypesProvider.Null,
new SecurityCacheProvider(
new SecurityPortfolioManager(securityManager,
new SecurityTransactionManager(null, securityManager),
new AlgorithmSettings())),
new MapFilePrimaryExchangeProvider(
Composer.Instance.GetExportedValueByTypeName<IMapFileProvider>(
Config.Get("map-file-provider", "LocalDiskMapFileProvider"))),
algorithm: null
);
securityManager.SetSecurityService(securityService);
Assert.DoesNotThrow(() =>
{
var symbol = Symbol.Create("TEST", SecurityType.Equity, Market.USA);
var security = securityManager.CreateSecurity(symbol, new List<SubscriptionDataConfig>(), underlying: null);
});
}
}
}