ooni-probe-cli/internal/netxlite/quirks_test.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")
		}
	})
}