1 //
2 // AsyncVoidMethodBuilderTest.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 
30 using System;
31 using System.Threading;
32 using System.Threading.Tasks;
33 using NUnit.Framework;
34 using System.Runtime.CompilerServices;
35 using System.Runtime.ExceptionServices;
36 
37 namespace MonoTests.System.Runtime.CompilerServices
38 {
39 	[TestFixture]
40 	public class AsyncVoidMethodBuilderTest
41 	{
42 		class MyContext : SynchronizationContext
43 		{
44 			public int Started;
45 			public int Completed;
46 			public int PostCounter;
47 			public int SendCounter;
48 
OperationStarted()49 			public override void OperationStarted ()
50 			{
51 				++Started;
52 				base.OperationStarted ();
53 			}
54 
OperationCompleted()55 			public override void OperationCompleted ()
56 			{
57 				++Completed;
58 				base.OperationCompleted ();
59 			}
60 
Post(SendOrPostCallback d, object state)61 			public override void Post (SendOrPostCallback d, object state)
62 			{
63 				if (state is ExceptionDispatchInfo) {
64 					++PostCounter;
65 				}
66 			}
67 
Send(SendOrPostCallback d, object state)68 			public override void Send (SendOrPostCallback d, object state)
69 			{
70 				if (state is Exception) {
71 					++SendCounter;
72 					base.Send (d, state);
73 				}
74 			}
75 		}
76 
77 		[Test]
SetResult()78 		public void SetResult ()
79 		{
80 			var awaiter = AsyncVoidMethodBuilder.Create ();
81 			awaiter.SetResult ();
82 		}
83 
84 		[Test]
SetException()85 		public void SetException ()
86 		{
87 			var context = new MyContext ();
88 			var old = SynchronizationContext.Current;
89 
90 			try {
91 				SynchronizationContext.SetSynchronizationContext (context);
92 				var awaiter = AsyncVoidMethodBuilder.Create ();
93 
94 				Assert.AreEqual (1, context.Started, "#1");
95 				Assert.AreEqual (0, context.Completed, "#2");
96 				Assert.AreEqual (0, context.SendCounter, "#3");
97 				Assert.AreEqual (0, context.PostCounter, "#4");
98 
99 				awaiter.SetException (new ApplicationException ());
100 
101 				Assert.AreEqual (1, context.Started, "#5");
102 				Assert.AreEqual (1, context.Completed, "#6");
103 				Assert.AreEqual (0, context.SendCounter, "#7");
104 				Assert.AreEqual (1, context.PostCounter, "#8");
105 
106 				awaiter.SetResult ();
107 
108 				Assert.AreEqual (1, context.Started, "#9");
109 				Assert.AreEqual (2, context.Completed, "#10");
110 				Assert.AreEqual (0, context.SendCounter, "#11");
111 				Assert.AreEqual (1, context.PostCounter, "#12");
112 			} finally {
113 				SynchronizationContext.SetSynchronizationContext (old);
114 			}
115 		}
116 	}
117 }
118 
119