Fix missing actor context on test start (#346)

src/Akka.Hosting.TestKit.Tests/DiPropsFailTest.cs

@@ -0,0 +1,46 @@
+// -----------------------------------------------------------------------
+//  <copyright file="DiPropsFailTest.cs" company="Akka.NET Project">
+//      Copyright (C) 2013-2023 .NET Foundation <https://github.com/akkadotnet/akka.net>
+//  </copyright>
+// -----------------------------------------------------------------------
+
+using System;
+using Akka.Actor;
+using Akka.DependencyInjection;
+using Microsoft.Extensions.Logging;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace Akka.Hosting.TestKit.Tests;
+
+// Regression test for https://github.com/akkadotnet/Akka.Hosting/issues/343
+public class DiPropsFailTest: TestKit
+{
+    public DiPropsFailTest(ITestOutputHelper output) : base(nameof(DiPropsFailTest), output)
+    {}
+
+    protected override void ConfigureAkka(AkkaConfigurationBuilder builder, IServiceProvider provider)
+    { }
+
+    [Fact]
+    public void DiTest()
+    {
+        var actor = Sys.ActorOf(NonRootActorWithDi.Props());
+        actor.Tell("test");
+        ExpectMsg<string>("test");
+    }
+
+    private class NonRootActorWithDi: ReceiveActor
+    {
+        public static Props Props() => DependencyResolver.For(Context.System).Props<NonRootActorWithDi>();
+
+        public NonRootActorWithDi(ILogger<NonRootActorWithDi> log)
+        {
+            ReceiveAny(msg =>
+            {
+                log.LogInformation("Received {Msg}", msg);
+                Sender.Tell(msg);
+            });
+        }
+    }
+}

src/Akka.Hosting.TestKit/TestKit.cs

@@ -8,6 +8,7 @@
 using System.Threading;
 using System.Threading.Tasks;
 using Akka.Actor;
+using Akka.Actor.Internal;
 using Akka.Actor.Setup;
 using Akka.Annotations;
 using Akka.Configuration;
@@ -53,7 +54,7 @@ public IHost Host
         public ITestOutputHelper? Output { get; }
         public LogLevel LogLevel { get; }
 
-        private TaskCompletionSource<Done> _initialized = new TaskCompletionSource<Done>();
+        private readonly TaskCompletionSource<Done> _initialized = new TaskCompletionSource<Done>();
 
         protected TestKit(string? actorSystemName = null, ITestOutputHelper? output = null, TimeSpan? startupTimeout = null, LogLevel logLevel = LogLevel.Information)
         : base(Assertions)
@@ -102,8 +103,15 @@ private void InternalConfigureServices(HostBuilderContext context, IServiceColle
 
                 builder.AddStartup((system, _) =>
                 {
-                    base.InitializeTest(system, (ActorSystemSetup)null!, null, null);
-                    _initialized.SetResult(Done.Instance);
+                    try
+                    {
+                        base.InitializeTest(system, (ActorSystemSetup)null!, null, null);
+                        _initialized.SetResult(Done.Instance);
+                    }
+                    catch (Exception e)
+                    {
+                        _initialized.SetException(e);
+                    }
                 });
             });
         }
@@ -154,6 +162,10 @@ public async Task InitializeAsync()
             }
 
             await _initialized.Task;
+
+            if (this is not INoImplicitSender && InternalCurrentActorCellKeeper.Current is null)
+                InternalCurrentActorCellKeeper.Current = (ActorCell)((ActorRefWithCell)TestActor).Underlying;
+
             await BeforeTestStart();
         }