mcs/class/corlib/Test/System.Threading.Tasks/TaskCompletionSourceTests.cs

   1 #if NET_4_0
   2 //
   3 // TaskCompletionSourceTests.cs
   4 //
   5 // Author:
   6 //       Jérémie "Garuma" Laval <jeremie.laval@gmail.com>
   7 //
   8 // Copyright (c) 2009 Jérémie "Garuma" Laval
   9 //
  10 // Permission is hereby granted, free of charge, to any person obtaining a copy
  11 // of this software and associated documentation files (the "Software"), to deal
  12 // in the Software without restriction, including without limitation the rights
  13 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  14 // copies of the Software, and to permit persons to whom the Software is
  15 // furnished to do so, subject to the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be included in
  18 // all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  21 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  22 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  23 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  24 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  25 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  26 // THE SOFTWARE.
  27
  28 using System;
  29 using System.Threading;
  30 using System.Threading.Tasks;
  31
  32 using NUnit.Framework;
  33
  34 namespace MonoTests.System.Threading.Tasks
  35 {
  36         [TestFixture]
  37         public class TaskCompletionSourceTests
  38         {
  39                 TaskCompletionSource<int> completionSource;
  40                 object state;
  41
  42                 [SetUp]
  43                 public void Setup ()
  44                 {
  45                         state = new object ();
  46                         completionSource = new TaskCompletionSource<int> (state, TaskCreationOptions.LongRunning);
  47                 }
  48
  49                 [Test]
  50                 public void CreationCheckTest ()
  51                 {
  52                         Assert.IsNotNull (completionSource.Task, "#1");
  53                         Assert.AreEqual (TaskCreationOptions.LongRunning, completionSource.Task.CreationOptions, "#2");
  54                 }
  55
  56                 [Test]
  57                 public void SetResultTest ()
  58                 {
  59                         Assert.IsNotNull (completionSource.Task, "#1");
  60                         Assert.IsTrue (completionSource.TrySetResult (42), "#2");
  61                         Assert.AreEqual (TaskStatus.RanToCompletion, completionSource.Task.Status, "#3");
  62                         Assert.AreEqual (42, completionSource.Task.Result, "#4");
  63                         Assert.IsFalse (completionSource.TrySetResult (43), "#5");
  64                         Assert.AreEqual (TaskStatus.RanToCompletion, completionSource.Task.Status, "#6");
  65                         Assert.AreEqual (42, completionSource.Task.Result, "#7");
  66                         Assert.IsFalse (completionSource.TrySetCanceled (), "#8");
  67                         Assert.AreEqual (TaskStatus.RanToCompletion, completionSource.Task.Status, "#9");
  68                 }
  69
  70                 [Test]
  71                 public void SetCanceledTest ()
  72                 {
  73                         Assert.IsNotNull (completionSource.Task, "#1");
  74                         Assert.IsTrue (completionSource.TrySetCanceled (), "#2");
  75                         Assert.AreEqual (TaskStatus.Canceled, completionSource.Task.Status, "#3");
  76                         Assert.IsFalse (completionSource.TrySetResult (42), "#4");
  77                         Assert.AreEqual (TaskStatus.Canceled, completionSource.Task.Status, "#5");
  78                 }
  79
  80                 [Test]
  81                 public void SetExceptionTest ()
  82                 {
  83                         Exception e = new Exception ("foo");
  84
  85                         Assert.IsNotNull (completionSource.Task, "#1");
  86                         Assert.IsTrue (completionSource.TrySetException (e), "#2");
  87                         Assert.AreEqual (TaskStatus.Faulted, completionSource.Task.Status, "#3");
  88                         Assert.IsInstanceOfType (typeof (AggregateException), completionSource.Task.Exception, "#4.1");
  89
  90                         AggregateException aggr = (AggregateException)completionSource.Task.Exception;
  91                         Assert.AreEqual (1, aggr.InnerExceptions.Count, "#4.2");
  92                         Assert.AreEqual (e, aggr.InnerExceptions[0], "#4.3");
  93
  94                         Assert.IsFalse (completionSource.TrySetResult (42), "#5");
  95                         Assert.AreEqual (TaskStatus.Faulted, completionSource.Task.Status, "#6");
  96                         Assert.IsFalse (completionSource.TrySetCanceled (), "#8");
  97                         Assert.AreEqual (TaskStatus.Faulted, completionSource.Task.Status, "#9");
  98                 }
  99
 100                 [Test, ExpectedException (typeof (InvalidOperationException))]
 101                 public void SetResultExceptionTest ()
 102                 {
 103                         Assert.IsNotNull (completionSource.Task, "#1");
 104                         Assert.IsTrue (completionSource.TrySetResult (42), "#2");
 105                         Assert.AreEqual (TaskStatus.RanToCompletion, completionSource.Task.Status, "#3");
 106                         Assert.AreEqual (42, completionSource.Task.Result, "#4");
 107
 108                         completionSource.SetResult (43);
 109                 }
 110         }
 111 }
 112 #endif