addressing copilot comments

Author: Sophia Tevosyan

src/Worker/Grpc/GrpcEntityRunner.cs

@@ -135,7 +135,7 @@ public static async Task<string> LoadAndRunAsync(
 
         if (addToExtendedSessions)
         {
-            extendedSessions.Set(
+            extendedSessions!.Set(
                 request.InstanceId,
                 result.EntityState,
                 new MemoryCacheEntryOptions { SlidingExpiration = TimeSpan.FromSeconds(extendedSessionIdleTimeoutInSeconds) });

src/Worker/Grpc/GrpcInstanceRunnerUtils.cs

@@ -1,9 +1,6 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 
-using System;
-using System.Collections.Generic;
-using System.Text;
 using Microsoft.Extensions.Caching.Memory;
 
 namespace Microsoft.DurableTask.Worker.Grpc;
@@ -19,7 +16,7 @@ static class GrpcInstanceRunnerUtils
     /// </summary>
     /// <remarks>
     /// If any request property is missing or invalid (i.e. the key is misspelled or the value is of the wrong type),
-    /// extended sessions are not enabled and default values are assigned are assigned to the returns.
+    /// extended sessions are not enabled and default values are assigned to the returns.
     /// </remarks>
     /// <param name="properties">
     /// A dictionary containing request properties used to configure extended session behavior.

src/Worker/Grpc/GrpcOrchestrationRunner.cs

@@ -207,7 +207,7 @@ public static string LoadAndRun(
 
                 if (addToExtendedSessions && !executor.IsCompleted)
                 {
-                    extendedSessions.Set<ExtendedSessionState>(
+                    extendedSessions!.Set<ExtendedSessionState>(
                         request.InstanceId,
                         new(runtimeState, shim, executor),
                         new MemoryCacheEntryOptions { SlidingExpiration = TimeSpan.FromSeconds(extendedSessionIdleTimeoutInSeconds) });

test/Worker/Grpc.Tests/GrpcEntityRunnerTests.cs

@@ -73,8 +73,7 @@ public async Task MalformedRequestParameters_Means_CacheNotInitialized_Async()
             { "ExtendedSessionsIdleTimeoutInSeconds", Value.ForNumber(DefaultExtendedSessionIdleTimeoutInSeconds) } });
         byte[] requestBytes = entityRequest.ToByteArray();
         string requestString = Convert.ToBase64String(requestBytes);
-        string responseString = await GrpcEntityRunner.LoadAndRunAsync(requestString, new SimpleEntity(), extendedSessions);
-        Protobuf.OrchestratorResponse response = Protobuf.OrchestratorResponse.Parser.ParseFrom(Convert.FromBase64String(responseString));
+        await GrpcEntityRunner.LoadAndRunAsync(requestString, new SimpleEntity(), extendedSessions);
         Assert.False(extendedSessions.IsInitialized);
 
         // Wrong value type for extended session timeout key
@@ -373,7 +372,7 @@ static Protobuf.EntityBatchRequest CreateEntityRequest(IEnumerable<Protobuf.Oper
         return entityBatchRequest;
     }
 
-    class SimpleEntity : TaskEntity<int?>
+    sealed class SimpleEntity : TaskEntity<int?>
     {
         public void Set(int value)
         {

test/Worker/Grpc.Tests/GrpcOrchestrationRunnerTests.cs

@@ -120,8 +120,7 @@ public void MalformedRequestParameters_Means_CacheNotInitialized()
             { "ExtendedSessionsIdleTimeoutInSeconds", Value.ForNumber(DefaultExtendedSessionIdleTimeoutInSeconds) } });
         byte[] requestBytes = orchestratorRequest.ToByteArray();
         string requestString = Convert.ToBase64String(requestBytes);
-        string responseString = GrpcOrchestrationRunner.LoadAndRun(requestString, new SimpleOrchestrator(), extendedSessions);
-        Protobuf.OrchestratorResponse response = Protobuf.OrchestratorResponse.Parser.ParseFrom(Convert.FromBase64String(responseString));
+        GrpcOrchestrationRunner.LoadAndRun(requestString, new SimpleOrchestrator(), extendedSessions);
         Assert.False(extendedSessions.IsInitialized);
 
         // Wrong value type for extended session timeout key