redirects_example_test.go

package requests_test

import (
	"context"
	"fmt"
	"net/http"

	"github.com/carlmjohnson/requests"
)

func ExampleCheckRedirectPolicy() {
	cl := *http.DefaultClient
	cl.CheckRedirect = requests.NoFollow

	if err := requests.
		URL("https://httpbingo.org/redirect/1").
		Client(&cl).
		CheckStatus(http.StatusFound).
		Handle(func(res *http.Response) error {
			fmt.Println("Status", res.Status)
			fmt.Println("From", res.Request.URL)
			fmt.Println("To", res.Header.Get("Location"))
			return nil
		}).
		Fetch(context.Background()); err != nil {
		panic(err)
	}
	// Output:
	// Status 302 Found
	// From https://httpbingo.org/redirect/1
	// To /get
}

func ExampleMaxFollow() {
	cl := *http.DefaultClient
	cl.CheckRedirect = requests.MaxFollow(1)

	if err := requests.
		URL("https://httpbingo.org/redirect/2").
		Client(&cl).
		CheckStatus(http.StatusFound).
		Handle(func(res *http.Response) error {
			fmt.Println("Status", res.Status)
			fmt.Println("From", res.Request.URL)
			fmt.Println("To", res.Header.Get("Location"))
			return nil
		}).
		Fetch(context.Background()); err != nil {
		panic(err)
	}
	// Output:
	// Status 302 Found
	// From https://httpbingo.org/relative-redirect/1
	// To /get
}