26 lines
		
	
	
		
			720 B
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			26 lines
		
	
	
		
			720 B
		
	
	
	
		
			Go
		
	
	
	
// Copyright 2017 The Go Authors. All rights reserved.
 | 
						|
// Use of this source code is governed by the Apache 2.0
 | 
						|
// license that can be found in the LICENSE file.
 | 
						|
 | 
						|
package main
 | 
						|
 | 
						|
import (
 | 
						|
	"net/http/httptest"
 | 
						|
	"testing"
 | 
						|
)
 | 
						|
 | 
						|
func TestTipRedirects(t *testing.T) {
 | 
						|
	mux := newServeMux(&Proxy{builder: &godocBuilder{}})
 | 
						|
	req := httptest.NewRequest("GET", "http://example.com/foo?bar=baz", nil)
 | 
						|
	req.Header.Set("X-Forwarded-Proto", "http")
 | 
						|
	w := httptest.NewRecorder()
 | 
						|
	mux.ServeHTTP(w, req)
 | 
						|
	if w.Code != 302 {
 | 
						|
		t.Errorf("expected Code to be 302, got %d", w.Code)
 | 
						|
	}
 | 
						|
	want := "https://example.com/foo?bar=baz"
 | 
						|
	if loc := w.Header().Get("Location"); loc != want {
 | 
						|
		t.Errorf("Location header: got %s, want %s", loc, want)
 | 
						|
	}
 | 
						|
}
 |