5ebdeb56ca
## Checklist - [x] I have read the [contribution guidelines](https://github.com/ooni/probe-cli/blob/master/CONTRIBUTING.md) - [x] reference issue for this pull request: https://github.com/ooni/probe/issues/2158 - [x] if you changed anything related how experiments work and you need to reflect these changes in the ooni/spec repository, please link to the related ooni/spec pull request: https://github.com/ooni/spec/pull/250 ## Description This diff refactors the codebase to reimplement tlsping and tcpping to use the step-by-step measurements style. See docs/design/dd-003-step-by-step.md for more information on the step-by-step measurement style.
99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
package netxlite
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
func TestQuirkReduceErrors(t *testing.T) {
|
|
t.Run("no errors", func(t *testing.T) {
|
|
result := quirkReduceErrors(nil)
|
|
if !errors.Is(result, errReduceErrorsEmptyList) {
|
|
t.Fatal("wrong result")
|
|
}
|
|
})
|
|
|
|
t.Run("single error", func(t *testing.T) {
|
|
err := errors.New("mocked error")
|
|
result := quirkReduceErrors([]error{err})
|
|
if result != err {
|
|
t.Fatal("wrong result")
|
|
}
|
|
})
|
|
|
|
t.Run("multiple errors", func(t *testing.T) {
|
|
err1 := errors.New("mocked error #1")
|
|
err2 := errors.New("mocked error #2")
|
|
result := quirkReduceErrors([]error{err1, err2})
|
|
if result.Error() != "mocked error #1" {
|
|
t.Fatal("wrong result")
|
|
}
|
|
})
|
|
|
|
t.Run("multiple errors with meaningful ones", func(t *testing.T) {
|
|
err1 := errors.New("mocked error #1")
|
|
err2 := NewErrWrapper(
|
|
ClassifyGenericError,
|
|
CloseOperation,
|
|
errors.New("antani"),
|
|
)
|
|
err3 := NewErrWrapper(
|
|
ClassifyGenericError,
|
|
CloseOperation,
|
|
ECONNREFUSED,
|
|
)
|
|
err4 := errors.New("mocked error #3")
|
|
result := quirkReduceErrors([]error{err1, err2, err3, err4})
|
|
if result.Error() != FailureConnectionRefused {
|
|
t.Fatal("wrong result")
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestQuirkSortIPAddrs(t *testing.T) {
|
|
t.Run("with some addrs", func(t *testing.T) {
|
|
addrs := []string{
|
|
"::1",
|
|
"192.168.1.2",
|
|
"x.org", // ensure we skip non IP addrs
|
|
"2a00:1450:4002:404::2004",
|
|
"example.com", // ensure we skip non IP addrs
|
|
"142.250.184.36",
|
|
"2604:8800:5000:82:466:38ff:fecb:d46e",
|
|
"198.145.29.83",
|
|
"95.216.163.36",
|
|
}
|
|
expected := []string{
|
|
"192.168.1.2",
|
|
"142.250.184.36",
|
|
"198.145.29.83",
|
|
"95.216.163.36",
|
|
"::1",
|
|
"2a00:1450:4002:404::2004",
|
|
"2604:8800:5000:82:466:38ff:fecb:d46e",
|
|
}
|
|
out := quirkSortIPAddrs(addrs)
|
|
if diff := cmp.Diff(expected, out); diff != "" {
|
|
t.Fatal(diff)
|
|
}
|
|
})
|
|
|
|
t.Run("with an empty list", func(t *testing.T) {
|
|
if quirkSortIPAddrs(nil) != nil {
|
|
t.Fatal("expected nil output")
|
|
}
|
|
})
|
|
|
|
t.Run("with non-IP addrs", func(t *testing.T) {
|
|
addrs := []string{
|
|
"example.com",
|
|
"x.org",
|
|
}
|
|
if quirkSortIPAddrs(addrs) != nil {
|
|
t.Fatal("expected nil output")
|
|
}
|
|
})
|
|
}
|