round_tripper_test.go

package cloudflarebp_test

import (
	"net/http"
	"testing"

	cloudflarebp "github.com/Gaz492/cloudflare-bp-go"

	"github.com/stretchr/testify/assert"
)

func TestApplyCloudFlareByPassDefaultClient(t *testing.T) {
	client := http.DefaultClient

	res, err := client.Get("https://www.curseforge.com")
	assert.New(t).NoError(err)
	assert.New(t).Equal(403, res.StatusCode)

	// apply our bypass for request headers and client TLS configurations
	http.DefaultClient.Transport = cloudflarebp.AddCloudFlareByPass(http.DefaultClient.Transport)

	res, err = client.Get("https://www.curseforge.com")
	assert.New(t).NoError(err)
	assert.New(t).Equal(200, res.StatusCode)
}

func TestApplyCloudFlareByPassDefinedTransport(t *testing.T) {
	client := &http.Client{
		Transport: &http.Transport{},
	}

	// if the client requests something before applying the fix some configurations are applied already
	// and our ByPass won't work anymore, so we have to apply our ByPass as the first thing
	client.Transport = cloudflarebp.AddCloudFlareByPass(client.Transport)

	res, err := client.Get("https://www.curseforge.com")
	assert.New(t).NoError(err)
	assert.New(t).Equal(200, res.StatusCode)
}

// TestAddCloudFlareByPassSocksProxy tests the CloudFlare bypass while we're using a SOCK5 proxy transport layer.
//func TestAddCloudFlareByPassSocksProxy(t *testing.T) {
//	auth := proxy.Auth{
//		User:     os.Getenv("PROXY_USER"),
//		Password: os.Getenv("PROXY_PASS"),
//	}
//
//	dialer, err := proxy.SOCKS5(
//		"tcp",
//		fmt.Sprintf("%s:1080", os.Getenv("PROXY_HOST_SOCKS5")),
//		&auth,
//		proxy.Direct,
//	)
//	assert.New(t).NoError(err)
//
//	dc := dialer.(interface {
//		DialContext(ctx context.Context, network, addr string) (net.Conn, error)
//	})
//
//	client := &http.Client{
//		Transport: &http.Transport{DialContext: dc.DialContext},
//	}
//
//	// if the client requests something before applying the fix some configurations are applied already
//	// and our ByPass won't work anymore, so we have to apply our ByPass as the first thing
//	client.Transport = cloudflarebp.AddCloudFlareByPass(client.Transport)
//
//	res, err := client.Get("https://www.curseforge.com")
//	assert.New(t).NoError(err)
//	assert.New(t).Equal(200, res.StatusCode)
//}
//
//// TestAddCloudFlareByPassHTTPProxy tests the CloudFlare bypass while we're using a HTTP proxy transport layer.
//func TestAddCloudFlareByPassHTTPProxy(t *testing.T) {
//	proxyURL, _ := url.Parse(
//		fmt.Sprintf(
//			"https://%s:%s@%s:%s",
//			url.QueryEscape(os.Getenv("PROXY_USER")), url.QueryEscape(os.Getenv("PROXY_PASS")),
//			url.QueryEscape(os.Getenv("PROXY_HOST_HTTPS")), url.QueryEscape(os.Getenv("PROXY_PORT_HTTPS")),
//		),
//	)
//
//	client := &http.Client{
//		Transport: &http.Transport{Proxy: http.ProxyURL(proxyURL)},
//	}
//
//	res, err := client.Get("https://www.curseforge.com")
//	assert.New(t).NoError(err)
//	assert.New(t).Equal(403, res.StatusCode)
//
//	// if the client requests something before applying the fix some configurations are applied already
//	// and our ByPass won't work anymore, so we have to apply our ByPass as the first thing
//	client.Transport = cloudflarebp.AddCloudFlareByPass(client.Transport)
//
//	res, err = client.Get("https://www.curseforge.com")
//	assert.New(t).NoError(err)
//	assert.New(t).Equal(200, res.StatusCode)
//}