update restart logic at inmemoryorchestrationservice

Author: nytian

test/Grpc.IntegrationTests/GrpcDurableTaskClientIntegrationTests.cs

@@ -265,7 +265,8 @@ public async Task RestartAsync_EndToEnd(bool restartWithNewInstanceId)
         using var completionCts = new CancellationTokenSource(TimeSpan.FromSeconds(30));
         await server.Client.WaitForInstanceCompletionAsync(restartedInstanceId, completionCts.Token);
 
-        // Verify the restarted orchestration completed
+        // Verify the restarted orchestration completed.
+        // Also verify input and orchestrator name are matched.
         var restartedMetadata = await server.Client.GetInstanceAsync(restartedInstanceId, true);
         restartedMetadata.Should().NotBeNull();
         restartedMetadata!.Name.Should().Be(OrchestrationName);

test/Grpc.IntegrationTests/GrpcSidecar/InMemoryOrchestrationService.cs

@@ -451,10 +451,8 @@ public void AddMessage(TaskMessage message)
                     // For restart scenarios, clear the history and reset the state
                     if (isRestart && state.IsCompleted)
                     {
-                        state.HistoryEventsJson.Clear();
-                        state.ExecutionId = executionId; // Always update the execution ID
-                        state.IsCompleted = false;
-                        state.IsLoaded = false; // Reset the loaded state for restart scenarios
+                        state.ExecutionId = executionId;
+                        state.IsLoaded = false;
                     }
 
                     OrchestrationState newStatusRecord = new()
@@ -471,6 +469,7 @@ public void AddMessage(TaskMessage message)
                     };
 
                     state.StatusRecordJson = JsonValue.Create(newStatusRecord);
+                    state.HistoryEventsJson.Clear();
                     state.IsCompleted = false;
                 }
                 else if (state.IsCompleted)