using NUnit.Framework;
using System.Runtime.CompilerServices;
using System.Collections.Generic;
+using System.Collections;
namespace MonoTests.System.Runtime.CompilerServices
{
class Scheduler : TaskScheduler
{
string name;
+ int ic, qc;
public Scheduler (string name)
{
this.name = name;
}
- public int InlineCalls { get; set; }
- public int QueueCalls { get; set; }
+ public int InlineCalls { get { return ic; } }
+ public int QueueCalls { get { return qc; } }
protected override IEnumerable<Task> GetScheduledTasks ()
{
protected override void QueueTask (Task task)
{
- ++QueueCalls;
+ Interlocked.Increment (ref qc);
ThreadPool.QueueUserWorkItem (o => {
TryExecuteTask (task);
});
protected override bool TryExecuteTaskInline (Task task, bool taskWasPreviouslyQueued)
{
- ++InlineCalls;
+ 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]
Assert.IsTrue (t.Wait (3000), "#0");
Assert.AreEqual (0, t.Result, "#1");
Assert.AreEqual (0, b.InlineCalls, "#2b");
- Assert.AreEqual (2, a.QueueCalls, "#3a");
+ Assert.IsTrue (a.QueueCalls == 1 || a.QueueCalls == 2, "#3a");
Assert.AreEqual (1, b.QueueCalls, "#3b");
}
// 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);
+ }
}
}