Esempio n. 1
0
func StartProxy(wg *sync.WaitGroup) {
	log.Println("Starting proxy server")
	wg.Add(1)
	go func() {
		proxy := goproxy.NewProxyHttpServer(10)
		proxy.OnRequest().HandleConnect(goproxy.AlwaysMitm)
		proxy.Verbose = true

		err := http.ListenAndServe(":54321", proxy)
		if err != nil {
			log.Fatal(err.Error())
		}
		wg.Done()
	}()
}
Esempio n. 2
0
func TestLimiterHttp(t *testing.T) {

	v := limiter.NewRateLimiter(1, time.Second)

	proxy := goproxy.NewProxyHttpServer()
	proxy.OnRequest().Do(limiter.LimitHttp(v))

	client, s := oneShotProxy(proxy, t)
	defer s.Close()

	r1 := getStatus(srv.URL, client)
	assert.Equal(t, http.StatusNotFound, r1, "they must be equal")

	r2 := getStatus(srv.URL, client)
	assert.Equal(t, http.StatusTooManyRequests, r2, "they must be equal")

	time.Sleep(1 * time.Second)

	r3 := getStatus(srv.URL, client)
	assert.Equal(t, http.StatusNotFound, r3, "they must be equal")

}