d922bd9afc
The objective is to make PR checks run much faster. See https://github.com/ooni/probe/issues/2113 for context. Regarding netxlite's tests: Checking for every commit on master or on a release branch is good enough and makes pull requests faster than one minute since netxlite for windows is now 1m slower than coverage. We're losing some coverage but coverage from integration tests is not so good anyway, so I'm not super sad about this loss.
99 lines
1.8 KiB
Go
99 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/apex/log"
|
|
"github.com/ooni/probe-cli/v3/internal/cmd/jafar/iptables"
|
|
"github.com/ooni/probe-cli/v3/internal/shellx"
|
|
)
|
|
|
|
func ensureWeStartOverWithIPTables() {
|
|
iptables.NewCensoringPolicy().Waive()
|
|
}
|
|
|
|
func TestNoCommand(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skip test in short mode")
|
|
}
|
|
if runtime.GOOS != "linux" {
|
|
t.Skip("skip test on non Linux systems")
|
|
}
|
|
ensureWeStartOverWithIPTables()
|
|
*dnsProxyAddress = "127.0.0.1:0"
|
|
*httpProxyAddress = "127.0.0.1:0"
|
|
*tlsProxyAddress = "127.0.0.1:0"
|
|
go func() {
|
|
mainCh <- os.Interrupt
|
|
}()
|
|
main()
|
|
}
|
|
|
|
func TestWithCommand(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skip test in short mode")
|
|
}
|
|
if runtime.GOOS != "linux" {
|
|
t.Skip("skip test on non Linux systems")
|
|
}
|
|
ensureWeStartOverWithIPTables()
|
|
*dnsProxyAddress = "127.0.0.1:0"
|
|
*httpProxyAddress = "127.0.0.1:0"
|
|
*tlsProxyAddress = "127.0.0.1:0"
|
|
*mainCommand = "whoami"
|
|
defer func() {
|
|
*mainCommand = ""
|
|
}()
|
|
main()
|
|
}
|
|
|
|
func TestMustx(t *testing.T) {
|
|
t.Run("with no error", func(t *testing.T) {
|
|
var called int
|
|
mustx(nil, "", func(int) {
|
|
called++
|
|
})
|
|
if called != 0 {
|
|
t.Fatal("should not happen")
|
|
}
|
|
})
|
|
|
|
t.Run("with non-exit-code error", func(t *testing.T) {
|
|
var (
|
|
called int
|
|
exitcode int
|
|
)
|
|
mustx(errors.New("antani"), "", func(ec int) {
|
|
called++
|
|
exitcode = ec
|
|
})
|
|
if called != 1 {
|
|
t.Fatal("not called?!")
|
|
}
|
|
if exitcode != 1 {
|
|
t.Fatal("unexpected exitcode value")
|
|
}
|
|
})
|
|
|
|
t.Run("with exit-code error", func(t *testing.T) {
|
|
var (
|
|
called int
|
|
exitcode int
|
|
)
|
|
err := shellx.Run(log.Log, "curl", "-sf", "") // cause exitcode == 3
|
|
mustx(err, "", func(ec int) {
|
|
called++
|
|
exitcode = ec
|
|
})
|
|
if called != 1 {
|
|
t.Fatal("not called?!")
|
|
}
|
|
if exitcode != 3 {
|
|
t.Fatal("unexpected exitcode value")
|
|
}
|
|
})
|
|
}
|