Update unit tests

QuantConnect · Jan 19, 2024 · 3ff685c · 3ff685c
1 parent 35a6e50
commit 3ff685c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/Engine/RealTime/LiveTradingRealTimeHandler.cs b/Engine/RealTime/LiveTradingRealTimeHandler.cs
@@ -203,7 +203,7 @@ protected virtual IEnumerable<MarketHoursSegment> GetMarketHours(DateTime time,
         /// </summary>
         protected virtual void ResetMarketHoursDatabase()
         {
-            MarketHoursDatabase.Reset();
+            MarketHoursDatabase.ReloadEntries();
         }
     }
 }
diff --git a/Tests/Engine/RealTime/LiveTradingRealTimeHandlerTests.cs b/Tests/Engine/RealTime/LiveTradingRealTimeHandlerTests.cs
@@ -167,9 +167,21 @@ public bool TryRequestAdditionalTime(int minutes)
         public class TestLiveTradingRealTimeHandler: LiveTradingRealTimeHandler
         {
             private static AutoResetEvent OnSecurityUpdated = new AutoResetEvent(false);
+            private MarketHoursDatabase newMarketHoursDatabase;
             public void SetMarketHoursDatabase(MarketHoursDatabase marketHoursDatabase)
             {
-                MarketHoursDatabase = marketHoursDatabase;
+                newMarketHoursDatabase = marketHoursDatabase;
+            }
+            protected override void ResetMarketHoursDatabase()
+            {
+                if (newMarketHoursDatabase != null)
+                {
+                    MarketHoursDatabase = newMarketHoursDatabase;
+                }
+                else
+                {
+                    base.ResetMarketHoursDatabase();
+                }
             }
 
             public void TestRefreshMarketHoursToday(Security security, DateTime time, MarketHoursSegment expectedSegment)