83440cf110
The legacy part for now is internal/errorsx. It will stay there until I figure out whether it also needs some extra bug fixing. The good part is now in internal/netxlite/errorsx and contains all the logic for mapping errors. We need to further improve upon this logic by writing more thorough integration tests for QUIC. We also need to copy the various dialer, conn, etc adapters that set errors. We will put them inside netxlite and we will generate errors in a way that is less crazy with respect to the major operation. (The idea is to always wrap, given that now we measure in an incremental way and we don't measure every operation together.) Part of https://github.com/ooni/probe/issues/1591
87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
package modelx
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"errors"
|
|
"math"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ooni/probe-cli/v3/internal/netxlite/errorsx"
|
|
)
|
|
|
|
func TestNewTLSConnectionState(t *testing.T) {
|
|
conn, err := tls.Dial("tcp", "www.google.com:443", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
state := NewTLSConnectionState(conn.ConnectionState())
|
|
if len(state.PeerCertificates) < 1 {
|
|
t.Fatal("too few certificates")
|
|
}
|
|
if state.Version < tls.VersionSSL30 || state.Version > 0x0304 /*tls.VersionTLS13*/ {
|
|
t.Fatal("unexpected TLS version")
|
|
}
|
|
}
|
|
|
|
func TestMeasurementRoot(t *testing.T) {
|
|
ctx := context.Background()
|
|
if ContextMeasurementRoot(ctx) != nil {
|
|
t.Fatal("unexpected value for ContextMeasurementRoot")
|
|
}
|
|
if ContextMeasurementRootOrDefault(ctx) == nil {
|
|
t.Fatal("unexpected value ContextMeasurementRootOrDefault")
|
|
}
|
|
handler := &dummyHandler{}
|
|
root := &MeasurementRoot{
|
|
Handler: handler,
|
|
Beginning: time.Time{},
|
|
}
|
|
ctx = WithMeasurementRoot(ctx, root)
|
|
v := ContextMeasurementRoot(ctx)
|
|
if v != root {
|
|
t.Fatal("unexpected ContextMeasurementRoot value")
|
|
}
|
|
v = ContextMeasurementRootOrDefault(ctx)
|
|
if v != root {
|
|
t.Fatal("unexpected ContextMeasurementRoot value")
|
|
}
|
|
}
|
|
|
|
func TestMeasurementRootWithMeasurementRootPanic(t *testing.T) {
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("expected panic")
|
|
}
|
|
}()
|
|
ctx := context.Background()
|
|
_ = WithMeasurementRoot(ctx, nil)
|
|
}
|
|
|
|
func TestErrWrapperPublicAPI(t *testing.T) {
|
|
child := errors.New("mocked error")
|
|
wrapper := &errorsx.ErrWrapper{
|
|
Failure: "moobar",
|
|
WrappedErr: child,
|
|
}
|
|
if wrapper.Error() != "moobar" {
|
|
t.Fatal("The Error() method is misbehaving")
|
|
}
|
|
if wrapper.Unwrap() != child {
|
|
t.Fatal("The Unwrap() method is misbehaving")
|
|
}
|
|
}
|
|
|
|
func TestComputeBodySnapSize(t *testing.T) {
|
|
if ComputeBodySnapSize(-1) != math.MaxInt64 {
|
|
t.Fatal("unexpected result")
|
|
}
|
|
if ComputeBodySnapSize(0) != defaultBodySnapSize {
|
|
t.Fatal("unexpected result")
|
|
}
|
|
if ComputeBodySnapSize(127) != 127 {
|
|
t.Fatal("unexpected result")
|
|
}
|
|
}
|