package main

import (
	"errors"
	"fmt"
	"log"
	"net/http"
	"time"
)

func main() {
	r, err := http.NewRequest("GET", "myproto://dummy", nil)
	if err != nil {
		log.Fatal(err)
	}
	var t http.Transport
	t.RegisterProtocol("myproto", &myRoundTripper{})
	c := &http.Client{
		Transport: &t,
		Timeout:   time.Second,
	}
	_, err = c.Do(r)
	if err == nil {
		log.Fatal("expected timeout")
	}
	fmt.Println(err)
}

type myRoundTripper struct{}

func (*myRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
	select {
	case <-req.Cancel:
		return nil, errors.New("round trip canceled as expected")
	case <-time.After(5 * time.Second):
		log.Fatal("round trip was not canceled")
		return nil, errors.New("round trip was not canceled")
	}
}