Nit changes

QuantConnect · Jan 2, 2024 · fdfa80d · fdfa80d
1 parent b13076c
commit fdfa80d
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/Tests/Engine/Setup/BrokerageSetupHandlerTests.cs b/Tests/Engine/Setup/BrokerageSetupHandlerTests.cs
@@ -598,7 +598,7 @@ private void TestLoadExistingHoldingsAndOrders(IAlgorithm algorithm, Func<List<H
             brokerage.Setup(x => x.GetAccountHoldings()).Returns(getHoldings);
             brokerage.Setup(x => x.GetOpenOrders()).Returns(getOrders);
 
-            var setupHandler = new TestBrokerageSetupHandler();
+            var setupHandler = new TestableBrokerageSetupHandler();
 
             IBrokerageFactory factory;
             setupHandler.CreateBrokerage(job, algorithm, out factory);
@@ -797,6 +797,11 @@ public void PublicGetOpenOrders(IAlgorithm algorithm, IResultHandler resultHandl
             {
                 GetOpenOrders(algorithm, resultHandler, transactionHandler, brokerage);
             }
+
+            public bool TestLoadExistingHoldingsAndOrders(IBrokerage brokerage, IAlgorithm algorithm, SetupHandlerParameters parameters)
+            {
+                return LoadExistingHoldingsAndOrders(brokerage, algorithm, parameters);
+            }
         }
     }
 
@@ -893,12 +898,4 @@ public override void Disconnect()
 
         #endregion
     }
-
-    public class TestBrokerageSetupHandler: BrokerageSetupHandler
-    {
-        public bool TestLoadExistingHoldingsAndOrders(IBrokerage brokerage, IAlgorithm algorithm, SetupHandlerParameters parameters)
-        {
-            return LoadExistingHoldingsAndOrders(brokerage, algorithm, parameters);
-        }
-    }
 }