38 lines
755 B
Go
38 lines
755 B
Go
|
package mocks
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestHTTP3RoundTripper(t *testing.T) {
|
||
|
|
t.Run("RoundTrip", func(t *testing.T) {
|
||
|
|
expected := errors.New("mocked error")
|
||
|
|
txp := &HTTP3RoundTripper{
|
||
|
|
MockRoundTrip: func(req *http.Request) (*http.Response, error) {
|
||
|
|
return nil, expected
|
||
|
|
},
|
||
|
|
}
|
||
|
|
resp, err := txp.RoundTrip(&http.Request{})
|
||
|
|
if !errors.Is(err, expected) {
|
||
|
|
t.Fatal("unexpected err", err)
|
||
|
|
}
|
||
|
|
if resp != nil {
|
||
|
|
t.Fatal("unexpected resp")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("Close", func(t *testing.T) {
|
||
|
|
expected := errors.New("mocked error")
|
||
|
|
txp := &HTTP3RoundTripper{
|
||
|
|
MockClose: func() error {
|
||
|
|
return expected
|
||
|
|
},
|
||
|
|
}
|
||
|
|
if err := txp.Close(); !errors.Is(err, expected) {
|
||
|
|
t.Fatal("unexpected err", err)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|