using System.Threading.Tasks;
using NUnit.Framework;
using System.Runtime.CompilerServices;
+using System.Collections.Generic;
+using System.Collections;
namespace MonoTests.System.Runtime.CompilerServices
{
[TestFixture]
public class TaskAwaiterTest
{
+ class Scheduler : TaskScheduler
+ {
+ string name;
+ int ic, qc;
+
+ public Scheduler (string name)
+ {
+ this.name = name;
+ }
+
+ public int InlineCalls { get { return ic; } }
+ public int QueueCalls { get { return qc; } }
+
+ protected override IEnumerable<Task> GetScheduledTasks ()
+ {
+ throw new NotImplementedException ();
+ }
+
+ protected override void QueueTask (Task task)
+ {
+ Interlocked.Increment (ref qc);
+ ThreadPool.QueueUserWorkItem (o => {
+ TryExecuteTask (task);
+ });
+ }
+
+ protected override bool TryExecuteTaskInline (Task task, bool taskWasPreviouslyQueued)
+ {
+ Interlocked.Increment (ref ic);
+ return false;
+ }
+
+ public override string ToString ()
+ {
+ return "Scheduler-" + name;
+ }
+ }
+
+ class SingleThreadSynchronizationContext : SynchronizationContext
+ {
+ readonly Queue _queue = new Queue ();
+
+ public void RunOnCurrentThread ()
+ {
+ while (_queue.Count != 0) {
+ var workItem = (KeyValuePair<SendOrPostCallback, object>) _queue.Dequeue ();
+ workItem.Key (workItem.Value);
+ }
+ }
+
+ public override void Post (SendOrPostCallback d, object state)
+ {
+ if (d == null) {
+ throw new ArgumentNullException ("d");
+ }
+
+ _queue.Enqueue (new KeyValuePair<SendOrPostCallback, object> (d, state));
+ }
+
+ public override void Send (SendOrPostCallback d, object state)
+ {
+ throw new NotSupportedException ("Synchronously sending is not supported.");
+ }
+ }
+
+ string progress;
+ SynchronizationContext sc;
+ ManualResetEvent mre;
+
+ [SetUp]
+ public void Setup ()
+ {
+ sc = SynchronizationContext.Current;
+ }
+
+ [TearDown]
+ public void TearDown ()
+ {
+ SynchronizationContext.SetSynchronizationContext (sc);
+ }
+
[Test]
public void GetResultFaulted ()
{
awaiter.GetResult ();
Assert.AreEqual (TaskStatus.RanToCompletion, task.Status);
}
+
+ [Test]
+ public void CustomScheduler ()
+ {
+ // some test runners (e.g. Touch.Unit) will execute this on the main thread and that would lock them
+ if (!Thread.CurrentThread.IsBackground)
+ Assert.Ignore ("Current thread is not running in the background.");
+
+ var a = new Scheduler ("a");
+ var b = new Scheduler ("b");
+
+ var t = TestCS (a, b);
+ Assert.IsTrue (t.Wait (3000), "#0");
+ Assert.AreEqual (0, t.Result, "#1");
+ Assert.AreEqual (0, b.InlineCalls, "#2b");
+ Assert.IsTrue (a.QueueCalls == 1 || a.QueueCalls == 2, "#3a");
+ Assert.AreEqual (1, b.QueueCalls, "#3b");
+ }
+
+ static async Task<int> TestCS (TaskScheduler schedulerA, TaskScheduler schedulerB)
+ {
+ var res = await Task.Factory.StartNew (async () => {
+ if (TaskScheduler.Current != schedulerA)
+ return 1;
+
+ await Task.Factory.StartNew (
+ () => {
+ if (TaskScheduler.Current != schedulerB)
+ return 2;
+
+ return 0;
+ }, CancellationToken.None, TaskCreationOptions.None, schedulerB);
+
+ if (TaskScheduler.Current != schedulerA)
+ return 3;
+
+ return 0;
+ }, CancellationToken.None, TaskCreationOptions.None, schedulerA);
+
+ return res.Result;
+ }
+
+ [Test]
+ public void FinishedTaskOnCompleted ()
+ {
+ var mres = new ManualResetEvent (false);
+ var mres2 = new ManualResetEvent (false);
+
+ var tcs = new TaskCompletionSource<object> ();
+ tcs.SetResult (null);
+ var task = tcs.Task;
+
+ var awaiter = task.GetAwaiter ();
+ Assert.IsTrue (awaiter.IsCompleted, "#1");
+
+ awaiter.OnCompleted(() => {
+ if (mres.WaitOne (1000))
+ mres2.Set ();
+ });
+
+ mres.Set ();
+ // this will only terminate correctly if the test was not executed from the main thread
+ // e.g. Touch.Unit defaults to run tests on the main thread and this will return false
+ Assert.AreEqual (Thread.CurrentThread.IsBackground, mres2.WaitOne (2000), "#2");;
+ }
+
+ [Test]
+ public void CompletionOnSameCustomSynchronizationContext ()
+ {
+ progress = "";
+ var syncContext = new SingleThreadSynchronizationContext ();
+ SynchronizationContext.SetSynchronizationContext (syncContext);
+
+ syncContext.Post (delegate {
+ Go (syncContext);
+ }, null);
+
+ // Custom message loop
+ var cts = new CancellationTokenSource ();
+ cts.CancelAfter (5000);
+ while (progress.Length != 3 && !cts.IsCancellationRequested) {
+ syncContext.RunOnCurrentThread ();
+ Thread.Sleep (0);
+ }
+
+ Assert.AreEqual ("123", progress);
+ }
+
+ async void Go (SynchronizationContext ctx)
+ {
+ await Wait (ctx);
+
+ progress += "2";
+ }
+
+ async Task Wait (SynchronizationContext ctx)
+ {
+ await Task.Delay (10); // Force block suspend/return
+
+ ctx.Post (l => progress += "3", null);
+
+ progress += "1";
+
+ // Exiting same context - no need to post continuation
+ }
+
+ [Test]
+ public void CompletionOnDifferentCustomSynchronizationContext ()
+ {
+ mre = new ManualResetEvent (false);
+ progress = "";
+ var syncContext = new SingleThreadSynchronizationContext ();
+ SynchronizationContext.SetSynchronizationContext (syncContext);
+
+ syncContext.Post (delegate {
+ Task t = new Task (delegate() { });
+ Go2 (syncContext, t);
+ t.Start ();
+ }, null);
+
+ // Custom message loop
+ var cts = new CancellationTokenSource ();
+ cts.CancelAfter (5000);
+ while (progress.Length != 3 && !cts.IsCancellationRequested) {
+ syncContext.RunOnCurrentThread ();
+ Thread.Sleep (0);
+ }
+
+ Assert.AreEqual ("13xa2", progress);
+ }
+
+ async void Go2 (SynchronizationContext ctx, Task t)
+ {
+ await Wait2 (ctx, t);
+
+ progress += "a";
+
+ if (mre.WaitOne (5000))
+ progress += "2";
+ else
+ progress += "b";
+ }
+
+ async Task Wait2 (SynchronizationContext ctx, Task t)
+ {
+ await t; // Force block suspend/return
+
+ ctx.Post (l => {
+ progress += "3";
+ mre.Set ();
+ progress += "x";
+ }, null);
+
+ progress += "1";
+
+ SynchronizationContext.SetSynchronizationContext (null);
+ }
}
}
-#endif
\ No newline at end of file
+#endif
projects / mono.git / blobdiff