Fixed cookie not created in owin host

rspacjer · Sep 29, 2011 · 4b91350 · 4b91350
1 parent 32110eb
commit 4b91350
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/src/Nancy.Hosting.Owin.Tests/NancyOwinHostFixture.cs b/src/Nancy.Hosting.Owin.Tests/NancyOwinHostFixture.cs
@@ -36,6 +36,7 @@ public NancyOwinHostFixture()
             this.fakeEngine = A.Fake<INancyEngine>();
 
             this.fakeBootstrapper = A.Fake<INancyBootstrapper>();
+
             A.CallTo(() => this.fakeBootstrapper.GetEngine()).Returns(this.fakeEngine);
 
             this.host = new NancyOwinHost(fakeBootstrapper);
@@ -302,6 +303,21 @@ public void Should_read_entire_request_body_when_there_is_a_continuation()
             output.ShouldEqual("This is some request body content");
         }
 
+        [Fact]
+        public void Should_set_cookie_with_valid_header()
+        {
+            var fakeResponse = new Response() { StatusCode = HttpStatusCode.OK };
+            fakeResponse.AddCookie("test", "testvalue");
+            var fakeContext = new NancyContext() { Response = fakeResponse };
+
+            this.SetupFakeNancyCompleteCallback(fakeContext);
+            var respHeaders = new Dictionary<string, string>();
+            ResponseCallBack callback = (status, headers, bodyDelegate) =>respHeaders=(Dictionary<string,string>)headers;
+
+            this.host.ProcessRequest(environment, callback, fakeErrorCallback);
+            respHeaders.ContainsKey("Set-Cookie").ShouldBeTrue();
+            (respHeaders["Set-Cookie"]=="test=testvalue; path=/").ShouldBeTrue();
+        }
         /// <summary>
         /// Sets the fake nancy engine to execute the complete callback with the given context
         /// </summary>

diff --git a/src/Nancy.Hosting.Owin/NancyOwinHost.cs b/src/Nancy.Hosting.Owin/NancyOwinHost.cs
@@ -143,7 +143,10 @@ private void InvokeNancy(NancyRequestParameters parameters, ResponseCallBack res
                     {
                         var returnCode = GetReturnCode(result);
                         var headers = result.Response.Headers;
-
+                        foreach (var cookie in result.Response.Cookies)
+                        {
+                            headers.Add("Set-Cookie", cookie.ToString());
+                        }
                         responseCallBack.Invoke(returnCode, headers, GetResponseBodyBuilder(result));
                     },
                     errorCallback);