use RdTask.GetAwaiter in AsyncRdTaskTest

JetBrains · Mar 15, 2024 · 063066d · 063066d
1 parent 1869eaf
commit 063066d
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/rd-net/Test.RdFramework/AsyncRdTaskTest.cs b/rd-net/Test.RdFramework/AsyncRdTaskTest.cs
@@ -20,7 +20,24 @@ protected override IScheduler CreateScheduler(bool isServer)
   }
 
   [Test]
-  public void BindableRdCallListTest()
+  public void BindableRdCallListUseSystemTaskTest()
+  {
+    BindableRdCallListTest(TaskKind.System);
+  }
+
+  [Test]
+  public void BindableRdCallListUseRdTaskTest()
+  {
+    BindableRdCallListTest(TaskKind.Rd);
+  }
+
+  private enum TaskKind
+  {
+    System,
+    Rd,
+  }
+
+  private void BindableRdCallListTest(TaskKind taskKind)
   {  
     ClientWire.AutoTransmitMode = true;
     ServerWire.AutoTransmitMode = true;
@@ -52,7 +69,12 @@ public void BindableRdCallListTest()
       {
         BindToClient(lifetime, callsite, entity_id);
 
-        var list = await callsite.Start(lifetime, Unit.Instance).AsTask();
+        var list = taskKind switch
+        {
+          TaskKind.System => await callsite.Start(lifetime, Unit.Instance).AsTask(),
+          TaskKind.Rd     => await callsite.Start(lifetime, Unit.Instance),
+          _ => throw new ArgumentOutOfRangeException(nameof(taskKind), taskKind, null)
+        };
         var count = 0;
 
         list.View(lifetime, (lt, index, value) =>