10a2055163
* #1682: RFC9000 as main QUIC version * removed extra ALPN values from the TLSConfig * updated to quic-go v0.21.0 * only use h3
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package quicdialer_test
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"errors"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/lucas-clemente/quic-go"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/dialid"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/errorx"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/quicdialer"
|
|
)
|
|
|
|
func TestErrorWrapperFailure(t *testing.T) {
|
|
ctx := dialid.WithDialID(context.Background())
|
|
d := quicdialer.ErrorWrapperDialer{
|
|
Dialer: MockDialer{Sess: nil, Err: io.EOF}}
|
|
sess, err := d.DialContext(
|
|
ctx, "udp", "www.google.com:443", &tls.Config{}, &quic.Config{})
|
|
if sess != nil {
|
|
t.Fatal("expected a nil sess here")
|
|
}
|
|
errorWrapperCheckErr(t, err, errorx.QUICHandshakeOperation)
|
|
}
|
|
|
|
func errorWrapperCheckErr(t *testing.T, err error, op string) {
|
|
if !errors.Is(err, io.EOF) {
|
|
t.Fatal("expected another error here")
|
|
}
|
|
var errWrapper *errorx.ErrWrapper
|
|
if !errors.As(err, &errWrapper) {
|
|
t.Fatal("cannot cast to ErrWrapper")
|
|
}
|
|
if errWrapper.DialID == 0 {
|
|
t.Fatal("unexpected DialID")
|
|
}
|
|
if errWrapper.Operation != op {
|
|
t.Fatal("unexpected Operation")
|
|
}
|
|
if errWrapper.Failure != errorx.FailureEOFError {
|
|
t.Fatal("unexpected failure")
|
|
}
|
|
}
|
|
|
|
func TestErrorWrapperSuccess(t *testing.T) {
|
|
ctx := dialid.WithDialID(context.Background())
|
|
tlsConf := &tls.Config{
|
|
NextProtos: []string{"h3"},
|
|
ServerName: "www.google.com",
|
|
}
|
|
d := quicdialer.ErrorWrapperDialer{Dialer: quicdialer.SystemDialer{}}
|
|
sess, err := d.DialContext(ctx, "udp", "216.58.212.164:443", tlsConf, &quic.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if sess == nil {
|
|
t.Fatal("expected non-nil sess here")
|
|
}
|
|
}
|