SmartThreadPool/STPTests/TestThreadApartmentState.cs

using System.Threading;
using NUnit.Framework;
using Amib.Threading;

namespace SmartThreadPoolTests
{
	/// <summary>
	/// Summary description for TestThreadPriority.
	/// </summary>
	[TestFixture]
    [Category("TestThreadApartment")]
	public class TestThreadApartmentState
	{
		[Test]
		public void TestSTA()
		{
            CheckApartmentState(ApartmentState.STA);
		}

		[Test]
		public void TestMTA()
		{
            CheckApartmentState(ApartmentState.MTA);
		}

	    private static void CheckApartmentState(ApartmentState requestApartmentState)
	    {
	        STPStartInfo stpStartInfo = new STPStartInfo();
	        stpStartInfo.ApartmentState = requestApartmentState;

	        SmartThreadPool stp = new SmartThreadPool(stpStartInfo);

	        IWorkItemResult<ApartmentState> wir = stp.QueueWorkItem(() => GetCurrentThreadApartmentState());

	        ApartmentState resultApartmentState = wir.GetResult();

	        stp.WaitForIdle();

	        Assert.AreEqual(requestApartmentState, resultApartmentState);
	    }

	    private static ApartmentState GetCurrentThreadApartmentState()
	    {
	        return Thread.CurrentThread.GetApartmentState();
	    }
	}
}