From 3ff80a2a395c15ce3a9b01148a3400527bf3b88f Mon Sep 17 00:00:00 2001
From: Chris Cameron <chris.cameron@wedoresi.com>
Date: Wed, 15 Nov 2017 11:45:00 -0500
Subject: [PATCH] Tidying

---
 ICD.Common.Utils.Tests/SafeCriticalSectionTest.cs | 8 +++-----
 ICD.Common.Utils.Tests/ThreadingUtilsTest.cs      | 6 ++----
 ICD.Common.Utils/ThreadingUtils.cs                | 2 +-
 3 files changed, 6 insertions(+), 10 deletions(-)

diff --git a/ICD.Common.Utils.Tests/SafeCriticalSectionTest.cs b/ICD.Common.Utils.Tests/SafeCriticalSectionTest.cs
index 3c764d0..4feda40 100644
--- a/ICD.Common.Utils.Tests/SafeCriticalSectionTest.cs
+++ b/ICD.Common.Utils.Tests/SafeCriticalSectionTest.cs
@@ -48,9 +48,8 @@ namespace ICD.Common.Utils.Tests
 
 			// ReSharper disable once NotAccessedVariable
 			object handle = ThreadingUtils.SafeInvoke(() => { result = section.TryEnter() ? 0 : 1; });
-			ThreadingUtils.Sleep(1000);
-
-			Assert.AreEqual(1, result);
+			
+			Assert.IsTrue(ThreadingUtils.Wait(() => result == 1, 1000));
 
 			section.Leave();
 
@@ -60,9 +59,8 @@ namespace ICD.Common.Utils.Tests
 				result = section.TryEnter() ? 2 : 0;
 				section.Leave();
 			});
-			ThreadingUtils.Sleep(1000);
 
-			Assert.AreEqual(2, result);
+			Assert.IsTrue(ThreadingUtils.Wait(() => result == 2, 1000));
 		}
 	}
 }
diff --git a/ICD.Common.Utils.Tests/ThreadingUtilsTest.cs b/ICD.Common.Utils.Tests/ThreadingUtilsTest.cs
index f0f8c0f..7a852e5 100644
--- a/ICD.Common.Utils.Tests/ThreadingUtilsTest.cs
+++ b/ICD.Common.Utils.Tests/ThreadingUtilsTest.cs
@@ -44,8 +44,7 @@ namespace ICD.Common.Utils.Tests
 			                          });
 
 			Assert.IsFalse(result);
-			ThreadingUtils.Sleep(1000);
-			Assert.IsTrue(result);
+			Assert.IsTrue(ThreadingUtils.Wait(() => result, 1000));
 		}
 
 		[Test]
@@ -59,8 +58,7 @@ namespace ICD.Common.Utils.Tests
 			                          }, true);
 
 			Assert.IsFalse(result);
-			ThreadingUtils.Sleep(1000);
-			Assert.IsTrue(result);
+			Assert.IsTrue(ThreadingUtils.Wait(() => result, 1000));
 		}
 	}
 }
diff --git a/ICD.Common.Utils/ThreadingUtils.cs b/ICD.Common.Utils/ThreadingUtils.cs
index 20e0882..f42fe82 100644
--- a/ICD.Common.Utils/ThreadingUtils.cs
+++ b/ICD.Common.Utils/ThreadingUtils.cs
@@ -17,7 +17,7 @@ namespace ICD.Common.Utils
 		/// </summary>
 		/// <param name="condition"></param>
 		/// <param name="timeout"></param>
-		/// <returns></returns>
+		/// <returns>False if the call times out</returns>
 		public static bool Wait(Func<bool> condition, long timeout)
 		{
 			if (condition == null)