update test

Author: nytian

test/Grpc.IntegrationTests/GrpcDurableTaskClientIntegrationTests.cs

@@ -231,9 +231,9 @@ public async Task RestartAsync_EndToEnd(bool restartWithNewInstanceId)
     {
         await using HostTestLifetime server = await this.StartAsync();
 
-        // Start an initial orchestration
+        // Start an initial orchestration with shouldThrow = false to ensure it completes successfully
         string originalInstanceId = await server.Client.ScheduleNewOrchestrationInstanceAsync(
-            OrchestrationName, input: "test-input");
+            OrchestrationName, input: false);
 
         // Wait for it to start and then complete
         await server.Client.WaitForInstanceStartAsync(originalInstanceId, default);
@@ -265,7 +265,7 @@ public async Task RestartAsync_EndToEnd(bool restartWithNewInstanceId)
         OrchestrationMetadata? restartedMetadata = await server.Client.GetInstanceAsync(restartedInstanceId, true);
         restartedMetadata.Should().NotBeNull();
         restartedMetadata!.Name.Should().Be(OrchestrationName);
-        restartedMetadata.SerializedInput.Should().Be("\"test-input\"");
+        restartedMetadata.SerializedInput.Should().Be("false");
 
         // Complete the restarted orchestration
         await server.Client.RaiseEventAsync(restartedInstanceId, "event", default);

test/Grpc.IntegrationTests/GrpcSidecar/Grpc/TaskHubGrpcServer.cs

@@ -333,6 +333,60 @@ static P.GetInstanceResponse CreateGetInstanceResponse(OrchestrationState state,
         return new P.ResumeResponse();
     }
 
+    public override async Task<P.RestartInstanceResponse> RestartInstance(P.RestartInstanceRequest request, ServerCallContext context)
+    {
+        try
+        {
+            // Get the original orchestration state
+            IList<OrchestrationState> states = await this.client.GetOrchestrationStateAsync(request.InstanceId, false);
+            if (states == null || states.Count == 0)
+            {
+                throw new RpcException(new Status(StatusCode.NotFound, $"An orchestration with the instanceId {request.InstanceId} was not found."));
+            }
+
+            OrchestrationState state = states[0];
+            string newInstanceId = request.RestartWithNewInstanceId ? Guid.NewGuid().ToString("N") : request.InstanceId;
+
+            // Create a new orchestration instance
+            OrchestrationInstance newInstance = new()
+            {
+                InstanceId = newInstanceId,
+                ExecutionId = Guid.NewGuid().ToString("N"),
+            };
+
+            // Create an ExecutionStartedEvent with the original input
+            ExecutionStartedEvent startedEvent = new(-1, state.Name)
+            {
+                Input = state.Input, // Use the original serialized input
+                Version = state.Version ?? string.Empty,
+                OrchestrationInstance = newInstance,
+            };
+
+            TaskMessage taskMessage = new()
+            {
+                OrchestrationInstance = newInstance,
+                Event = startedEvent,
+            };
+
+            await this.client.CreateTaskOrchestrationAsync(taskMessage);
+
+            return new P.RestartInstanceResponse
+            {
+                InstanceId = newInstanceId,
+            };
+        }
+        catch (RpcException)
+        {
+            // Re-throw RpcException as-is
+            throw;
+        }
+        catch (Exception ex)
+        {
+            this.log.LogError(ex, "Error restarting orchestration instance {InstanceId}", request.InstanceId);
+            throw new RpcException(new Status(StatusCode.Internal, ex.Message));
+        }
+    }
+
     static P.TaskFailureDetails? GetFailureDetails(FailureDetails? failureDetails)
     {
         if (failureDetails == null)