go/oauth2-proxy
1
0
Fork 0
mirror of https://github.com/oauth2-proxy/oauth2-proxy.git synced 2025-05-23 22:40:31 +02:00
Code Issues Releases Activity
oauth2-proxy/pkg/app/pagewriter/pagewriter_test.go

139 lines
3.7 KiB
Go
Raw Normal View History

Move all pagewriter related code to dedicated pagewriter package
2021-02-13 11:38:33 +00:00
package pagewriter
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
import (
"io/ioutil"
Request ID Logging (#1087) * Add RequestID to the RequestScope * Expose RequestID to auth & request loggers * Use the RequestID in templated HTML pages * Allow customizing the RequestID header * Document new Request ID support * Add more cases to scope/requestID tests * Split Get vs Generate RequestID funtionality * Add {{.RequestID}} to the request logger tests * Move RequestID management to RequestScope * Use HTML escape instead of sanitization for Request ID rendering
2021-03-21 11:20:57 -07:00
"net/http"
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
"net/http/httptest"
"os"
"path/filepath"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
Move all pagewriter related code to dedicated pagewriter package
2021-02-13 11:38:33 +00:00
var _ = Describe("Writer", func() {
Context("NewWriter", func() {
var writer Writer
var opts Opts
Request ID Logging (#1087) * Add RequestID to the RequestScope * Expose RequestID to auth & request loggers * Use the RequestID in templated HTML pages * Allow customizing the RequestID header * Document new Request ID support * Add more cases to scope/requestID tests * Split Get vs Generate RequestID funtionality * Add {{.RequestID}} to the request logger tests * Move RequestID management to RequestScope * Use HTML escape instead of sanitization for Request ID rendering
2021-03-21 11:20:57 -07:00
var request *http.Request
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
BeforeEach(func() {
Move all pagewriter related code to dedicated pagewriter package
2021-02-13 11:38:33 +00:00
opts = Opts{
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
TemplatesPath: "",
ProxyPrefix: "/prefix",
Footer: "<Footer>",
Version: "<Version>",
Debug: false,
DisplayLoginForm: false,
ProviderName: "<ProviderName>",
SignInMessage: "<SignInMessage>",
}
Request ID Logging (#1087) * Add RequestID to the RequestScope * Expose RequestID to auth & request loggers * Use the RequestID in templated HTML pages * Allow customizing the RequestID header * Document new Request ID support * Add more cases to scope/requestID tests * Split Get vs Generate RequestID funtionality * Add {{.RequestID}} to the request logger tests * Move RequestID management to RequestScope * Use HTML escape instead of sanitization for Request ID rendering
2021-03-21 11:20:57 -07:00
request = httptest.NewRequest("", "http://127.0.0.1/", nil)
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
})
Context("With no custom templates", func() {
BeforeEach(func() {
var err error
Move all pagewriter related code to dedicated pagewriter package
2021-02-13 11:38:33 +00:00
writer, err = NewWriter(opts)
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
Expect(err).ToNot(HaveOccurred())
})
It("Writes the default error template", func() {
recorder := httptest.NewRecorder()
Request ID Logging (#1087) * Add RequestID to the RequestScope * Expose RequestID to auth & request loggers * Use the RequestID in templated HTML pages * Allow customizing the RequestID header * Document new Request ID support * Add more cases to scope/requestID tests * Split Get vs Generate RequestID funtionality * Add {{.RequestID}} to the request logger tests * Move RequestID management to RequestScope * Use HTML escape instead of sanitization for Request ID rendering
2021-03-21 11:20:57 -07:00
writer.WriteErrorPage(recorder, ErrorPageOpts{
Status: 500,
RedirectURL: "/redirect",
AppError: "Some debug error",
})
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
body, err := ioutil.ReadAll(recorder.Result().Body)
Expect(err).ToNot(HaveOccurred())
Expect(string(body)).To(HavePrefix("\n<!DOCTYPE html>"))
})
It("Writes the default sign in template", func() {
recorder := httptest.NewRecorder()
Request ID Logging (#1087) * Add RequestID to the RequestScope * Expose RequestID to auth & request loggers * Use the RequestID in templated HTML pages * Allow customizing the RequestID header * Document new Request ID support * Add more cases to scope/requestID tests * Split Get vs Generate RequestID funtionality * Add {{.RequestID}} to the request logger tests * Move RequestID management to RequestScope * Use HTML escape instead of sanitization for Request ID rendering
2021-03-21 11:20:57 -07:00
writer.WriteSignInPage(recorder, request, "/redirect")
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
body, err := ioutil.ReadAll(recorder.Result().Body)
Expect(err).ToNot(HaveOccurred())
Expect(string(body)).To(HavePrefix("\n<!DOCTYPE html>"))
})
})
Context("With custom templates", func() {
var customDir string
BeforeEach(func() {
var err error
customDir, err = ioutil.TempDir("", "oauth2-proxy-pagewriter-test")
Expect(err).ToNot(HaveOccurred())
templateHTML := `Custom Template`
signInFile := filepath.Join(customDir, signInTemplateName)
Expect(ioutil.WriteFile(signInFile, []byte(templateHTML), 0600)).To(Succeed())
errorFile := filepath.Join(customDir, errorTemplateName)
Expect(ioutil.WriteFile(errorFile, []byte(templateHTML), 0600)).To(Succeed())
opts.TemplatesPath = customDir
Move all pagewriter related code to dedicated pagewriter package
2021-02-13 11:38:33 +00:00
writer, err = NewWriter(opts)
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
Expect(err).ToNot(HaveOccurred())
})
AfterEach(func() {
Expect(os.RemoveAll(customDir)).To(Succeed())
})
It("Writes the custom error template", func() {
recorder := httptest.NewRecorder()
Request ID Logging (#1087) * Add RequestID to the RequestScope * Expose RequestID to auth & request loggers * Use the RequestID in templated HTML pages * Allow customizing the RequestID header * Document new Request ID support * Add more cases to scope/requestID tests * Split Get vs Generate RequestID funtionality * Add {{.RequestID}} to the request logger tests * Move RequestID management to RequestScope * Use HTML escape instead of sanitization for Request ID rendering
2021-03-21 11:20:57 -07:00
writer.WriteErrorPage(recorder, ErrorPageOpts{
Status: 500,
RedirectURL: "/redirect",
AppError: "Some debug error",
})
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
body, err := ioutil.ReadAll(recorder.Result().Body)
Expect(err).ToNot(HaveOccurred())
Expect(string(body)).To(Equal("Custom Template"))
})
It("Writes the custom sign in template", func() {
recorder := httptest.NewRecorder()
Request ID Logging (#1087) * Add RequestID to the RequestScope * Expose RequestID to auth & request loggers * Use the RequestID in templated HTML pages * Allow customizing the RequestID header * Document new Request ID support * Add more cases to scope/requestID tests * Split Get vs Generate RequestID funtionality * Add {{.RequestID}} to the request logger tests * Move RequestID management to RequestScope * Use HTML escape instead of sanitization for Request ID rendering
2021-03-21 11:20:57 -07:00
writer.WriteSignInPage(recorder, request, "/redirect")
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
body, err := ioutil.ReadAll(recorder.Result().Body)
Expect(err).ToNot(HaveOccurred())
Expect(string(body)).To(Equal("Custom Template"))
})
})
Context("With an invalid custom template", func() {
var customDir string
BeforeEach(func() {
var err error
customDir, err = ioutil.TempDir("", "oauth2-proxy-pagewriter-test")
Expect(err).ToNot(HaveOccurred())
templateHTML := `{{ Custom Broken Template`
signInFile := filepath.Join(customDir, signInTemplateName)
Expect(ioutil.WriteFile(signInFile, []byte(templateHTML), 0600)).To(Succeed())
opts.TemplatesPath = customDir
})
AfterEach(func() {
Expect(os.RemoveAll(customDir)).To(Succeed())
})
It("Should return an error", func() {
Move all pagewriter related code to dedicated pagewriter package
2021-02-13 11:38:33 +00:00
writer, err := NewWriter(opts)
Wrap templates and page rendering in PageWriter interface
2021-02-12 18:25:46 +00:00
Expect(err).To(MatchError(ContainSubstring("template: sign_in.html:1: function \"Custom\" not defined")))
Expect(writer).To(BeNil())
})
})
})
})
Reference in New Issue Copy Permalink