mirror of
https://github.com/telanflow/mps.git
synced 2025-09-26 20:41:25 +08:00
34 lines
656 B
Go
34 lines
656 B
Go
package mps
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewTunnelHandler(t *testing.T) {
|
|
srv := newTestServer()
|
|
defer srv.Close()
|
|
|
|
tunnel := NewTunnelHandler()
|
|
//tunnel.Transport().Proxy = func(r *http.Request) (*url.URL, error) {
|
|
// return url.Parse("http://127.0.0.1:7890")
|
|
//}
|
|
tunnelSrv := httptest.NewServer(tunnel)
|
|
defer tunnelSrv.Close()
|
|
|
|
resp, err := HttpGet(srv.URL, func(r *http.Request) (*url.URL, error) {
|
|
return url.Parse(tunnelSrv.URL)
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
resp.Body.Close()
|
|
|
|
asserts := assert.New(t)
|
|
asserts.Equal(resp.StatusCode, 200)
|
|
}
|