Allow testing the WorkerManager

Author: Toastbrot236

Refresh.Workers/WorkerManager.cs

@@ -42,7 +42,7 @@ public void AddJob(WorkerJob worker)
         this._jobs.Add(worker);
     }
 
-    private void RunWorkCycle()
+    public void RunWorkCycle()
     {
         WorkContext context = new()
         {
@@ -99,25 +99,30 @@ private void RunWorkCycle()
         }
     }
 
+    public void RemoveUnusedJobStates()
+    {
+        GameDatabaseContext database = this._databaseProvider.GetContext();
+        string[] jobsFromWorkerManager = this._jobs.Select(j => j.GetType().Name).ToArray();
+        string[] jobsFromDatabase = database.GetAllJobIds(WorkerClass.Refresh).ToArray();
+        
+        foreach (string jobId in jobsFromDatabase)
+        {
+            if (!jobsFromWorkerManager.Any(j => j == jobId))
+            {
+                this._logger.LogInfo(RefreshContext.Worker, $"Removing job state for {jobId} because it doesn't exist in RefreshWorkerManager (likely from a newer/different Refresh build).");
+                database.RemoveJobState(jobId, false);
+            }
+        }
+        database.SaveChanges();
+    }
+
     public void Start()
     {
         this._logger.LogDebug(RefreshContext.Startup, "Starting the worker thread");
         this._threadShouldRun = true;
         Thread thread = new(() =>
         {
-            GameDatabaseContext database = this._databaseProvider.GetContext();
-            string[] jobsFromWorkerManager = this._jobs.Select(j => j.GetType().Name).ToArray();
-            string[] jobsFromDatabase = database.GetAllJobIds(WorkerClass.Refresh).ToArray();
-            
-            foreach (string jobId in jobsFromDatabase)
-            {
-                if (!jobsFromWorkerManager.Any(j => j == jobId))
-                {
-                    this._logger.LogInfo(RefreshContext.Worker, $"Removing job state for {jobId} because it doesn't exist in RefreshWorkerManager (likely from a newer/different Refresh build).");
-                    database.RemoveJobState(jobId, false);
-                }
-            }
-            database.SaveChanges();
+            this.RemoveUnusedJobStates();
 
             while (this._threadShouldRun)
             {

RefreshTests.GameServer/TestContext.cs

@@ -237,11 +237,16 @@ public WorkContext GetWorkContext()
         {
             Database = this.Database,
             Logger = this.Server.Value.Logger,
-            DataStore = (IDataStore)this.GetService<StorageService>()
-                .AddParameterToEndpoint(null!, new BunkumParameterInfo(typeof(IDataStore), ""), null!)!,
+            DataStore = this.GetDataStore(),
         };
     }
 
+    public IDataStore GetDataStore()
+    {
+        return (IDataStore)this.GetService<StorageService>()
+            .AddParameterToEndpoint(null!, new BunkumParameterInfo(typeof(IDataStore), ""), null!)!;
+    }
+
     public void Dispose()
     {
         this.Database.Dispose();