Milo: Add themed page for errors

milo/appengine/frontend/milo_test.go

 // Copyright 2016 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package frontend
 
 import (
         "flag"
         "fmt"
         "io/ioutil"
         "net/http"
+        "net/url"
         "path/filepath"
         "reflect"
         "regexp"
         "strings"
         "testing"
         "time"
 
         "github.com/luci/gae/impl/memory"
         "github.com/luci/luci-go/common/clock/testclock"
         "github.com/luci/luci-go/milo/appengine/buildbot"
@@ skipping 49 matching lines @@
 }
 
 func (m fakeOAuthMethod) Authenticate(context.Context, *http.Request) (*auth.User, error) {
         return &auth.User{
                 Identity: identity.Identity("user:abc@example.com"),
                 Email:    "abc@example.com",
                 ClientID: m.clientID,
         }, nil
 }
 
-func (m fakeOAuthMethod) LoginURL(context.Context, string) (string, error) {
-        return "https://login.url/", nil
+func (m fakeOAuthMethod) LoginURL(c context.Context, target string) (string, error) {
+        return "https://login.url/?target=" + target, nil
 }
 
-func (m fakeOAuthMethod) LogoutURL(context.Context, string) (string, error) {
-        return "https://logout.url/", nil
+func (m fakeOAuthMethod) LogoutURL(c context.Context, target string) (string, error) {
+        return "https://logout.url/?target=" + target, nil
 }
 
 type analyticsSettings struct {
         AnalyticsID string `json:"analytics_id"`
 }
 
 func TestPages(t *testing.T) {
         fixZeroDurationRE := regexp.MustCompile(`(Running for:|waiting) 0s?`)
         fixZeroDuration := func(text string) string {
                 return fixZeroDurationRE.ReplaceAllLiteralString(text, "[ZERO DURATION]")
         }
 
         Convey("Testing basic rendering.", t, func() {
                 // Load all the bundles.
                 c := context.Background()
                 c = memory.Use(c)
+                c = settings.WithRequest(c, &http.Request{URL: &url.URL{Path: "/foobar"}})
                 c, _ = testclock.UseTime(c, testclock.TestTimeUTC)
                 a := auth.Authenticator{fakeOAuthMethod{"some_client_id"}}
                 c = auth.SetAuthenticator(c, a)
                 c = luciSettings.Use(c, luciSettings.New(&luciSettings.MemoryStorage{Expiration: time.Second}))
                 err := luciSettings.Set(c, "analytics", &analyticsSettings{"UA-12345-01"}, "", "")
                 So(err, ShouldBeNil)
                 for _, nb := range settings.GetTemplateBundles() {
                         Convey(fmt.Sprintf("Testing theme %q", nb.Name), func() {
                                 err := nb.Bundle.EnsureLoaded(c)
                                 So(err, ShouldBeNil)
@@ skipping 18 matching lines @@
                                                                         So(fixZeroDuration(string(buf)), ShouldEqual, fixZeroDuration(string(localBuf)))
                                                                 }
                                                         })
                                                 }
                                         })
                                 }
                         })
                 }
         })
 }