Working on Tokens Manager.

1 files changed, 64 insertions, 0 deletions

diff --git a/Software/Visual_Studio/Tango.UnitTesting/MachineService_TST.cs b/Software/Visual_Studio/Tango.UnitTesting/MachineService_TST.cs
new file mode 100644
index 000000000..ca1653e3e
--- /dev/null
+++ b/Software/Visual_Studio/Tango.UnitTesting/MachineService_TST.cs
@@ -0,0 +1,64 @@
+using System;
+using System.Threading;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Tango.Web.Authentication;
+
+namespace Tango.UnitTesting
+{
+    [TestClass]
+    [TestCategory("Machine Service")]
+    public class MachineService_TST
+    {
+        private class TokenObject : IEquatable<TokenObject>
+        {
+            public String Value { get; set; }
+
+            public bool Equals(TokenObject other)
+            {
+                return Value == other.Value;
+            }
+        }
+
+        [TestMethod]
+        public void Test_Tokens_Manager()
+        {
+            TokensManager<TokenObject> tokensManager = new TokensManager<TokenObject>();
+
+            TokenObject t1 = new TokenObject();
+            t1.Value = "Roy";
+
+            TokenObject t2 = new TokenObject();
+            t2.Value = "Sagi";
+
+            String token1 = tokensManager.GetOrCreate(t1);
+            String token2 = tokensManager.GetOrCreate(t2);
+
+            Assert.AreEqual(tokensManager.GetTokenObject(token1), t1);
+            Assert.AreEqual(tokensManager.GetTokenObject(token2), t2);
+
+            TokenObject t3 = new TokenObject();
+            t3.Value = "Roy";
+
+            String token3 = tokensManager.GetOrCreate(t3);
+            Assert.AreEqual(token3, token1);
+            Assert.AreEqual(tokensManager.GetTokenObject(token1), tokensManager.GetTokenObject(token3));
+
+            TokenObject tOld = new TokenObject()
+            {
+                Value = "Expired Token"
+            };
+
+            tokensManager.ExpirationTime = TimeSpan.FromSeconds(2);
+
+            Thread.Sleep(2000);
+
+            String recent_token = tokensManager.GetOrCreate(tOld);
+
+            Thread.Sleep(1000);
+
+            Assert.IsTrue(tokensManager.Exists(recent_token));
+
+            Assert.IsFalse(tokensManager.Exists(token1));
+        }
+    }
+}