fix: try to fix failed tests in ci

meysamhadeli · Dec 18, 2024 · bd345ae · bd345ae
1 parent e871a37
commit bd345ae
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/src/BuildingBlocks/PersistMessageProcessor/Extensions.cs b/src/BuildingBlocks/PersistMessageProcessor/Extensions.cs
@@ -51,7 +51,10 @@ IWebHostEnvironment env
 
         services.AddScoped<IPersistMessageProcessor, PersistMessageProcessor>();
 
-        services.AddHostedService<PersistMessageBackgroundService>();
+        if (env.EnvironmentName != "test")
+        {
+            services.AddHostedService<PersistMessageBackgroundService>();
+        }
 
         return services;
     }

diff --git a/src/BuildingBlocks/PersistMessageProcessor/PersistMessageBackgroundService.cs b/src/BuildingBlocks/PersistMessageProcessor/PersistMessageBackgroundService.cs
@@ -14,15 +14,12 @@ IOptions<PersistMessageOptions> options
 {
     private PersistMessageOptions _options = options.Value;
 
-    private Task? _executingTask;
-
-    protected override Task ExecuteAsync(CancellationToken stoppingToken)
+    protected override async Task ExecuteAsync(CancellationToken stoppingToken)
     {
         logger.LogInformation("PersistMessage Background Service Start");
 
-        _executingTask = ProcessAsync(stoppingToken);
+        await ProcessAsync(stoppingToken);
 
-        return _executingTask;
     }
 
     public override Task StopAsync(CancellationToken cancellationToken)

diff --git a/src/BuildingBlocks/TestBase/TestBase.cs b/src/BuildingBlocks/TestBase/TestBase.cs
@@ -95,8 +95,7 @@ protected TestFixture()
                         {
                             TestRegistrationServices?.Invoke(services);
                             services.ReplaceSingleton(AddHttpContextAccessorMock);
-                            services.RemoveAll<IHostedService>();
-
+                            // services.RemoveAll<IHostedService>();
                             services.AddSingleton<PersistMessageBackgroundService>();
 
                             // Register all ITestDataSeeder implementations dynamically