mcs/class/corlib/Test/System.Runtime.CompilerServices/TaskAwaiterTest.cs

   1 //
   2 // TaskAwaiterTest.cs
   3 //
   4 // Authors:
   5 //      Marek Safar  <marek.safar@gmail.com>
   6 //
   7 // Copyright (C) 2011 Xamarin, Inc (http://www.xamarin.com)
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28
  29 #if NET_4_5
  30
  31 using System;
  32 using System.Threading;
  33 using System.Threading.Tasks;
  34 using NUnit.Framework;
  35 using System.Runtime.CompilerServices;
  36 using System.Collections.Generic;
  37
  38 namespace MonoTests.System.Runtime.CompilerServices
  39 {
  40         [TestFixture]
  41         public class TaskAwaiterTest
  42         {
  43                 class Scheduler : TaskScheduler
  44                 {
  45                         string name;
  46
  47                         public Scheduler (string name)
  48                         {
  49                                 this.name = name;
  50                         }
  51
  52                         public int InlineCalls { get; set; }
  53                         public int QueueCalls { get; set; }
  54
  55                         protected override IEnumerable<Task> GetScheduledTasks ()
  56                         {
  57                                 throw new NotImplementedException ();
  58                         }
  59
  60                         protected override void QueueTask (Task task)
  61                         {
  62                                 ++QueueCalls;
  63                                 ThreadPool.QueueUserWorkItem (o => {
  64                                         TryExecuteTask (task);
  65                                 });
  66                         }
  67
  68                         protected override bool TryExecuteTaskInline (Task task, bool taskWasPreviouslyQueued)
  69                         {
  70                                 ++InlineCalls;
  71                                 return false;
  72                         }
  73                 }
  74
  75                 [Test]
  76                 public void GetResultFaulted ()
  77                 {
  78                         TaskAwaiter awaiter;
  79
  80                         var task = new Task (() => { throw new ApplicationException (); });
  81                         awaiter = task.GetAwaiter ();
  82                         task.RunSynchronously (TaskScheduler.Current);
  83
  84
  85                         Assert.IsTrue (awaiter.IsCompleted);
  86
  87                         try {
  88                                 awaiter.GetResult ();
  89                                 Assert.Fail ();
  90                         } catch (ApplicationException) {
  91                         }
  92                 }
  93
  94                 [Test]
  95                 public void GetResultCanceled ()
  96                 {
  97                         TaskAwaiter awaiter;
  98
  99                         var token = new CancellationToken (true);
 100                         var task = new Task (() => { }, token);
 101                         awaiter = task.GetAwaiter ();
 102
 103                         try {
 104                                 awaiter.GetResult ();
 105                                 Assert.Fail ();
 106                         } catch (TaskCanceledException) {
 107                         }
 108                 }
 109
 110                 [Test]
 111                 public void GetResultWaitOnCompletion ()
 112                 {
 113                         TaskAwaiter awaiter;
 114
 115                         var task = Task.Delay (30);
 116                         awaiter = task.GetAwaiter ();
 117
 118                         awaiter.GetResult ();
 119                         Assert.AreEqual (TaskStatus.RanToCompletion, task.Status);
 120                 }
 121
 122                 [Test]
 123                 public void CustomScheduler ()
 124                 {
 125                         // some test runners (e.g. Touch.Unit) will execute this on the main thread and that would lock them
 126                         if (!Thread.CurrentThread.IsBackground)
 127                                 return;
 128
 129                         var a = new Scheduler ("a");
 130                         var b = new Scheduler ("b");
 131
 132                         var t = TestCS (a, b);
 133                         Assert.IsTrue (t.Wait (3000), "#0");
 134                         Assert.AreEqual (0, t.Result, "#1");
 135                         Assert.AreEqual (1, a.InlineCalls, "#2a");
 136                         Assert.AreEqual (0, b.InlineCalls, "#2b");
 137                         Assert.AreEqual (2, a.QueueCalls, "#3a");
 138                         Assert.AreEqual (1, b.QueueCalls, "#3b");
 139                 }
 140
 141                 static async Task<int> TestCS (TaskScheduler schedulerA, TaskScheduler schedulerB)
 142                 {
 143                         var res = await Task.Factory.StartNew (async () => {
 144                                 if (TaskScheduler.Current != schedulerA)
 145                                         return 1;
 146
 147                                 await Task.Factory.StartNew (
 148                                         () => {
 149                                                 if (TaskScheduler.Current != schedulerB)
 150                                                         return 2;
 151
 152                                                 return 0;
 153                                         }, CancellationToken.None, TaskCreationOptions.None, schedulerB);
 154
 155                                 if (TaskScheduler.Current != schedulerA)
 156                                         return 3;
 157
 158                                 return 0;
 159                         }, CancellationToken.None, TaskCreationOptions.None, schedulerA);
 160
 161                         return res.Result;
 162                 }
 163
 164                 [Test]
 165                 public void FinishedTaskOnCompleted ()
 166                 {
 167                         var mres = new ManualResetEvent (false);
 168                         var mres2 = new ManualResetEvent (false);
 169
 170                         var tcs = new TaskCompletionSource<object> ();
 171                         tcs.SetResult (null);
 172                         var task = tcs.Task;
 173
 174                         var awaiter = task.GetAwaiter ();
 175                         Assert.IsTrue (awaiter.IsCompleted, "#1");
 176
 177                         awaiter.OnCompleted(() => {
 178                                 if (mres.WaitOne (1000))
 179                                         mres2.Set ();
 180                         });
 181
 182                         mres.Set ();
 183                         // this will only terminate correctly if the test was not executed from the main thread
 184                         // e.g. Touch.Unit defaults to run tests on the main thread and this will return false
 185                         Assert.AreEqual (Thread.CurrentThread.IsBackground, mres2.WaitOne (2000), "#2");;
 186                 }
 187         }
 188 }
 189
 190 #endif