go/oauth2-proxy
1
0
Fork 0
mirror of https://github.com/oauth2-proxy/oauth2-proxy.git synced 2024-11-24 08:52:25 +02:00
Code Issues Releases Activity

Tranfser all cookies in tests

Browse Source
This commit is contained in:
Joel Speed 2019-05-16 17:13:14 +01:00 committed by Brian Van Klaveren
parent 93df7d9132
commit a7693cc72a
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkg/sessions/session_store_test.go
View File

@ -115,8 +115,9 @@ var _ = Describe("NewSessionStore", func() {
err := ss.Save(saveResp, req, session) err := ss.Save(saveResp, req, session)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
resultCookie := saveResp.Result().Cookies()[0] for _, c := range saveResp.Result().Cookies() {
request.AddCookie(resultCookie) request.AddCookie(c)
}
err = ss.Clear(response, request) err = ss.Clear(response, request)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
}) })
@ -177,14 +178,18 @@ var _ = Describe("NewSessionStore", func() {
err := ss.Save(saveResp, req, session) err := ss.Save(saveResp, req, session)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
resultCookie := saveResp.Result().Cookies()[0] resultCookies := saveResp.Result().Cookies()
request.AddCookie(resultCookie) for _, c := range resultCookies {
request.AddCookie(c)
}
err = ss.Clear(response, request) err = ss.Clear(response, request)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
// The following should only be for server stores // The following should only be for server stores
loadReq := httptest.NewRequest("GET", "http://example.com/", nil) loadReq := httptest.NewRequest("GET", "http://example.com/", nil)
loadReq.AddCookie(resultCookie) for _, c := range resultCookies {
loadReq.AddCookie(c)
}
loadedAfterClear, err = ss.Load(loadReq) loadedAfterClear, err = ss.Load(loadReq)
}) })