/*
 * 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 NUnit.Framework;
using QuantConnect.Lean.Engine.DataFeeds;
using QuantConnect.Orders;
using QuantConnect.Report;
using QuantConnect.Securities;
using QuantConnect.Brokerages;
using System;
using System.Collections.Generic;
using System.Linq;

namespace QuantConnect.Tests.Report
{
    [TestFixture]
    public class PortfolioLooperAlgorithmTests
    {
        private PortfolioLooperAlgorithm CreateAlgorithm(IEnumerable<Order> orders, AlgorithmConfiguration algorithmConfiguration = null)
        {
            var algorithm = new PortfolioLooperAlgorithm(100000m, orders, algorithmConfiguration);

            // Create MHDB and Symbol properties DB instances for the DataManager
            var marketHoursDatabase = MarketHoursDatabase.FromDataFolder();
            var symbolPropertiesDataBase = SymbolPropertiesDatabase.FromDataFolder();
            var dataPermissionManager = new DataPermissionManager();
            var dataManager = new DataManager(new QuantConnect.Report.MockDataFeed(),
                new UniverseSelection(
                    algorithm,
                    new SecurityService(algorithm.Portfolio.CashBook,
                        marketHoursDatabase,
                        symbolPropertiesDataBase,
                        algorithm,
                        RegisteredSecurityDataTypesProvider.Null,
                        new SecurityCacheProvider(algorithm.Portfolio),
                        algorithm: algorithm),
                    dataPermissionManager,
                    TestGlobals.DataProvider),
                algorithm,
                algorithm.TimeKeeper,
                marketHoursDatabase,
                false,
                RegisteredSecurityDataTypesProvider.Null,
                dataPermissionManager);

            var securityService = new SecurityService(algorithm.Portfolio.CashBook,
                marketHoursDatabase,
                symbolPropertiesDataBase,
                algorithm,
                RegisteredSecurityDataTypesProvider.Null,
                new SecurityCacheProvider(algorithm.Portfolio),
                algorithm: algorithm);

            // Initialize security services and other properties so that we
            // don't get null reference exceptions during our re-calculation
            algorithm.Securities.SetSecurityService(securityService);
            algorithm.SubscriptionManager.SetDataManager(dataManager);

            return algorithm;
        }

        [Test]
        public void Algorithm_CanSetLeverageOnAllSecurityTypes()
        {
            var orders = new Symbol[]
            {
                Symbol.Create("AAPL", SecurityType.Equity, Market.USA),
                Symbol.CreateOption("AAPL", Market.USA, OptionStyle.American, OptionRight.Call, 120m, new DateTime(2020, 5, 21)),
                Symbol.Create("EURUSD", SecurityType.Forex, Market.Oanda),
                Symbol.Create("XAUUSD", SecurityType.Cfd, Market.Oanda),
                Symbol.CreateFuture(Futures.Energy.CrudeOilWTI, Market.NYMEX, new DateTime(2020, 5, 21)),
                Symbol.Create("BTCUSD", SecurityType.Crypto, Market.Coinbase)
            }.Select(s => new MarketOrder(s, 1m, new DateTime(2020, 1, 1))).ToList();

            var algorithm = CreateAlgorithm(orders);
            Assert.DoesNotThrow(() => algorithm.FromOrders(orders));
        }

        [Test]
        public void Algorithm_UsesExpectedLeverageOnAllSecurityTypes()
        {
            var orders = new Symbol[]
            {
                Symbol.Create("AAPL", SecurityType.Equity, Market.USA),
                Symbol.CreateOption("AAPL", Market.USA, OptionStyle.American, OptionRight.Call, 120m, new DateTime(2020, 5, 21)),
                Symbol.Create("EURUSD", SecurityType.Forex, Market.Oanda),
                Symbol.Create("XAUUSD", SecurityType.Cfd, Market.Oanda),
                Symbol.CreateFuture(Futures.Energy.CrudeOilWTI, Market.NYMEX, new DateTime(2020, 5, 21)),
                Symbol.Create("BTCUSD", SecurityType.Crypto, Market.Coinbase)
            }.Select(s => new MarketOrder(s, 1m, new DateTime(2020, 1, 1)));

            var algorithm = CreateAlgorithm(orders);

            Assert.IsTrue(algorithm.Securities.Where(x => x.Key.SecurityType == SecurityType.Equity).All(x => x.Value.BuyingPowerModel.GetLeverage(x.Value) == 10000m));
            Assert.IsTrue(algorithm.Securities.Where(x => x.Key.SecurityType == SecurityType.Option).All(x => x.Value.BuyingPowerModel.GetLeverage(x.Value) == 1m));
            Assert.IsTrue(algorithm.Securities.Where(x => x.Key.SecurityType == SecurityType.Forex).All(x => x.Value.BuyingPowerModel.GetLeverage(x.Value) == 10000m));
            Assert.IsTrue(algorithm.Securities.Where(x => x.Key.SecurityType == SecurityType.Cfd).All(x => x.Value.BuyingPowerModel.GetLeverage(x.Value) == 10000m));
            Assert.IsTrue(algorithm.Securities.Where(x => x.Key.SecurityType == SecurityType.Future).All(x => x.Value.BuyingPowerModel.GetLeverage(x.Value) == 1m));
            Assert.IsTrue(algorithm.Securities.Where(x => x.Key.SecurityType == SecurityType.Crypto).All(x => x.Value.BuyingPowerModel.GetLeverage(x.Value) == 10000m));
        }

        [TestCase("BTC", BrokerageName.Binance, AccountType.Cash)]
        [TestCase("USDT", BrokerageName.Coinbase, AccountType.Cash)]
        [TestCase("EUR", BrokerageName.Bitfinex, AccountType.Margin)]
        public void SetsTheRightAlgorithmConfiguration(string currency, BrokerageName brokerageName, AccountType accountType)
        {
            var algorithm = CreateAlgorithm(new List<Order>(),
                new AlgorithmConfiguration("AlgorightmName", new HashSet<string>(), currency, brokerageName, accountType,
                    new Dictionary<string, string>(), DateTime.MinValue, DateTime.MinValue, null));
            algorithm.Initialize();

            Assert.AreEqual(currency, algorithm.AccountCurrency);
            Assert.AreEqual(brokerageName, BrokerageModel.GetBrokerageName(algorithm.BrokerageModel));
            Assert.AreEqual(accountType, algorithm.BrokerageModel.AccountType);
        }
    }
}