From 1e9929a3acc076b37e4f6aaeaefeeb0b9b1f7829 Mon Sep 17 00:00:00 2001
From: Roger Johansson <roger@asynkron.se>
Date: Thu, 29 Feb 2024 19:59:54 +0100
Subject: [PATCH] rollback testprovider

---
 src/Proto.Cluster.TestProvider/TestProvider.cs | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/src/Proto.Cluster.TestProvider/TestProvider.cs b/src/Proto.Cluster.TestProvider/TestProvider.cs
index 283a6b65ef..6cfdc3fa36 100644
--- a/src/Proto.Cluster.TestProvider/TestProvider.cs
+++ b/src/Proto.Cluster.TestProvider/TestProvider.cs
@@ -28,9 +28,8 @@ public TestProvider(TestProviderOptions options, InMemAgent agent)
         _agent = agent;
     }
 
-    public async Task StartMemberAsync(Cluster cluster)
+    public Task StartMemberAsync(Cluster cluster)
     {
-        await Task.Yield();
         var memberList = cluster.MemberList;
         var (host, port) = cluster.System.GetAddress();
         var kinds = cluster.GetClusterKinds();
@@ -47,26 +46,30 @@ public async Task StartMemberAsync(Cluster cluster)
                 Port = port
             }
         );
+
+        return Task.CompletedTask;
     }
 
-    public async Task StartClientAsync(Cluster cluster)
+    public Task StartClientAsync(Cluster cluster)
     {
-        await Task.Yield();
         var memberList = cluster.MemberList;
 
         _id = cluster.System.Id;
         _memberList = memberList;
         _agent.StatusUpdate += AgentOnStatusUpdate;
         _agent.ForceUpdate();
+
+        return Task.CompletedTask;
     }
 
-    public async Task ShutdownAsync(bool graceful)
+    public Task ShutdownAsync(bool graceful)
     {
-        await Task.Delay(100);
         Logger.LogDebug("Unregistering service {Service}", _id);
 
         _ttlReportTimer?.Stop();
         _agent.DeregisterService(_id);
+
+        return Task.CompletedTask;
     }
 
     private void AgentOnStatusUpdate(object sender, EventArgs e) => NotifyStatuses();