From 731f6ed6f1d5fdbb159af726984fa462af3ba2da Mon Sep 17 00:00:00 2001
From: Masahiro Nagano <kazeburo@gmail.com>
Date: Thu, 9 Apr 2020 16:27:54 +0900
Subject: [PATCH] add rewrite test

---
 proxy/handler_test.go | 44 ++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 43 insertions(+), 1 deletion(-)

diff --git a/proxy/handler_test.go b/proxy/handler_test.go
index aa14905..2ed45b3 100644
--- a/proxy/handler_test.go
+++ b/proxy/handler_test.go
@@ -70,7 +70,7 @@ func TestCopyRequest(t *testing.T) {
 	assert.Equal(t, req.Header["User-Agent"][0], "dummy-client")
 	assert.Equal(t, req.Header["X-Chocon-Test-Value"][0], "6")
 
-	for k, _ := range req.Header {
+	for k := range req.Header {
 		if _, ok := ignoredHeaderNames[k]; ok {
 			assert.Fail(t, fmt.Sprintf("header filed: %s must be removed", k))
 		}
@@ -82,3 +82,45 @@ func BenchmarkCopyRequest(b *testing.B) {
 		_ = dummyProxy.copyRequest(dummyRequest)
 	}
 }
+
+func BenchmarkRewriteHost(b *testing.B) {
+	status := &Status{Code: http.StatusOK}
+	originalReq, _ := http.NewRequest("GET", "/dummy", nil)
+	originalReq.URL.Host = "example.com.ccnproxy:3000"
+	originalReq.Host = originalReq.URL.Host
+	req, _ := http.NewRequest("GET", "/dummy", nil)
+	req.URL.Scheme = "http"
+	for n := 0; n < b.N; n++ {
+		dummyProxy.rewriteProxyHost(originalReq, req, status)
+	}
+}
+
+func TestRewriteHost(t *testing.T) {
+	cases := []struct {
+		originalReqHost string
+		reqHost         string
+		scheme          string
+	}{
+		{"example.com.ccnproxy:3000", "example.com:3000", "http"},
+		{"example.com.ccnproxy", "example.com", "http"},
+		{"example.com.ccnproxy.local:3000", "example.com:3000", "http"},
+		{"example.com.ccnproxy.local", "example.com", "http"},
+		{"example.com.ccnproxy-ssl:3000", "example.com:3000", "https"},
+		{"example.com.ccnproxy-ssl", "example.com", "https"},
+	}
+
+	for _, c := range cases {
+		t.Run(c.originalReqHost, func(t *testing.T) {
+			status := &Status{Code: http.StatusOK}
+			originalReq, _ := http.NewRequest("GET", "/dummy", nil)
+			req, _ := http.NewRequest("GET", "/dummy", nil)
+			req.URL.Scheme = "http"
+			originalReq.URL.Host = c.originalReqHost
+			originalReq.Host = originalReq.URL.Host
+			dummyProxy.rewriteProxyHost(originalReq, req, status)
+			assert.Equal(t, status.Code, http.StatusOK)
+			assert.Equal(t, req.Host, c.reqHost)
+			assert.Equal(t, req.URL.Scheme, c.scheme)
+		})
+	}
+}