adding Session Delete test

unosquare · Apr 14, 2017 · 82fbd5a · 82fbd5a
1 parent febb97f
commit 82fbd5a
Show file tree

Hide file tree

Showing 4 changed files with 113 additions and 2 deletions.
diff --git a/test/Unosquare.Labs.EmbedIO.Tests/FluentTest.cs b/test/Unosquare.Labs.EmbedIO.Tests/FluentTest.cs
@@ -40,7 +40,7 @@ public void FluentWithWebApi()
 
             Assert.AreEqual(webServer.Modules.Count, 1, "It has 1 modules loaded");
             Assert.IsNotNull(webServer.Module<WebApiModule>(), "It has WebApiModule");
-            Assert.AreEqual(webServer.Module<WebApiModule>().ControllersCount, 3, "It has three controllers");
+            Assert.AreEqual(webServer.Module<WebApiModule>().ControllersCount, 4, "It has four controllers");
 
             webServer.Dispose();
         }
@@ -66,7 +66,7 @@ public void FluentLoadWebApiControllers()
 
             Assert.AreEqual(webServer.Modules.Count, 1, "It has 1 modules loaded");
             Assert.IsNotNull(webServer.Module<WebApiModule>(), "It has WebApiModule");
-            Assert.AreEqual(webServer.Module<WebApiModule>().ControllersCount, 3, "It has three controllers");
+            Assert.AreEqual(webServer.Module<WebApiModule>().ControllersCount, 4, "It has four controllers");
 
             webServer.Dispose();
         }

diff --git a/test/Unosquare.Labs.EmbedIO.Tests/LocalSessionModuleTest.cs b/test/Unosquare.Labs.EmbedIO.Tests/LocalSessionModuleTest.cs
@@ -6,6 +6,7 @@
     using System.Threading.Tasks;
     using Modules;
     using TestObjects;
+    using System.IO;
 
     [TestFixture]
     public class LocalSessionModuleTest
@@ -28,6 +29,8 @@ public void Init()
             WebServer = new WebServer(WebServerUrl);
             WebServer.RegisterModule(new LocalSessionModule() { Expiration = WaitTimeSpan });
             WebServer.RegisterModule(new StaticFilesModule(RootPath));
+            WebServer.RegisterModule(new WebApiModule());
+            WebServer.Module<WebApiModule>().RegisterController<TestLocalSessionController>();
             WebServer.RunAsync();
         }
 
@@ -101,6 +104,65 @@ protected async Task GetFile(string content)
             }
         }
 
+        [Test]
+        public async Task DeleteSession()
+        {
+            var request = (HttpWebRequest)WebRequest.Create(WebServerUrl);
+            CookieContainer container = new CookieContainer();
+            request.CookieContainer = container;
+
+            request = (HttpWebRequest)WebRequest.Create(WebServerUrl + TestLocalSessionController.PutData);
+
+            request.CookieContainer = container;
+
+            using (var response = (HttpWebResponse)await request.GetResponseAsync())
+            {
+                Assert.AreEqual(response.StatusCode, HttpStatusCode.OK, "Status Code OK");
+
+                var body = new StreamReader(response.GetResponseStream()).ReadToEnd();
+
+                Assert.AreEqual(body, TestLocalSessionController.MyData);
+            }
+
+            request = (HttpWebRequest)WebRequest.Create(WebServerUrl + TestLocalSessionController.GetData);
+
+            request.CookieContainer = container;
+
+            using (var response = (HttpWebResponse)await request.GetResponseAsync())
+            {
+                Assert.AreEqual(response.StatusCode, HttpStatusCode.OK, "Status Code OK");
+
+                var body = new StreamReader(response.GetResponseStream()).ReadToEnd();
+
+                Assert.AreEqual(body, TestLocalSessionController.MyData);
+            }
+
+            request = (HttpWebRequest)WebRequest.Create(WebServerUrl + TestLocalSessionController.DeleteSession);
+            request.CookieContainer = container;
+
+            using (var response = (HttpWebResponse)await request.GetResponseAsync())
+            {
+                Assert.AreEqual(response.StatusCode, HttpStatusCode.OK, "Status Code OK");
+
+                var body = new StreamReader(response.GetResponseStream()).ReadToEnd();
+
+                Assert.AreEqual(body, "Deleted");
+            }
+
+            request = (HttpWebRequest)WebRequest.Create(WebServerUrl + TestLocalSessionController.GetData);
+
+            request.CookieContainer = container;
+
+            using (var response = (HttpWebResponse)await request.GetResponseAsync())
+            {
+                Assert.AreEqual(response.StatusCode, HttpStatusCode.OK, "Status Code OK");
+
+                var body = new StreamReader(response.GetResponseStream()).ReadToEnd();
+
+                Assert.AreEqual("", body);
+            }
+        }
+
         [TearDown]
         public void Kill()
         {

diff --git a/test/Unosquare.Labs.EmbedIO.Tests/TestObjects/TestLocalSessionController.cs b/test/Unosquare.Labs.EmbedIO.Tests/TestObjects/TestLocalSessionController.cs
@@ -0,0 +1,48 @@
+namespace Unosquare.Labs.EmbedIO.Tests.TestObjects
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Linq;
+    using System.Threading.Tasks;
+    using Unosquare.Labs.EmbedIO.Modules;
+#if NET46
+    using System.Net;
+#else
+    using Net;
+#endif
+
+    public class TestLocalSessionController : WebApiController
+    {
+        public const string DeleteSession = "deletesession";
+        public const string PutData = "putdata";
+        public const string GetData = "getdata";
+        public const string MyData = "MyData";
+
+        [WebApiHandler(HttpVerbs.Get, "/deletesession")]
+        public bool DeleteSessionC(WebServer server, HttpListenerContext context)
+        {
+            server.DeleteSession(context);
+
+            return context.JsonResponse("Deleted");
+        }
+
+        [WebApiHandler(HttpVerbs.Get, "/putdata")]
+        public bool PutDataSession(WebServer server, HttpListenerContext context)
+        {
+            server.GetSession(context).Data.TryAdd("sessionData", MyData);
+
+            return context.JsonResponse(server.GetSession(context).Data["sessionData"].ToString());
+        }
+
+        [WebApiHandler(HttpVerbs.Get, "/getdata")]
+        public bool GetDataSession(WebServer server, HttpListenerContext context)
+        {
+            object _data = null;
+            if (server.GetSession(context).Data.TryGetValue("sessionData", out _data))
+                return context.JsonResponse(server.GetSession(context).Data["sessionData"].ToString());
+            else
+                return context.JsonResponse("");
+
+        }
+    }
+}
diff --git a/test/Unosquare.Labs.EmbedIO.Tests/Unosquare.Labs.EmbedIO.Tests.csproj b/test/Unosquare.Labs.EmbedIO.Tests/Unosquare.Labs.EmbedIO.Tests.csproj
@@ -63,6 +63,7 @@
     <Compile Include="TestObjects\Resources.cs" />
     <Compile Include="TestObjects\TestController.cs" />
     <Compile Include="TestObjects\TestHelper.cs" />
+    <Compile Include="TestObjects\TestLocalSessionController.cs" />
     <Compile Include="TestObjects\TestRegexController.cs" />
     <Compile Include="TestObjects\TestWebModule.cs" />
     <Compile Include="TestObjects\TestWebSocket.cs" />