6aa2551c43
* urlgetter: fix tunnel test This diff fixes the urlgetter test suite to make sure we are correctly testing for tunnel creation. While there, improve the way in which we create a testing directory and add a test for that. Part of https://github.com/ooni/probe/issues/985. * fix comment * fix comment
101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
package tunnel
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/armon/go-socks5"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/internal/mockable"
|
|
)
|
|
|
|
func TestFakeWithCancelledContext(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel() // immediately fail
|
|
sess := &mockable.Session{}
|
|
tunnel, err := fakeStart(ctx, &Config{
|
|
Session: sess,
|
|
TunnelDir: "testdata",
|
|
})
|
|
if !errors.Is(err, context.Canceled) {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if tunnel != nil {
|
|
t.Fatal("expected nil tunnel here")
|
|
}
|
|
}
|
|
|
|
func TestFakeWithEmptyTunnelDir(t *testing.T) {
|
|
ctx := context.Background()
|
|
sess := &mockable.Session{}
|
|
tunnel, err := fakeStart(ctx, &Config{
|
|
Session: sess,
|
|
TunnelDir: "",
|
|
})
|
|
if !errors.Is(err, ErrEmptyTunnelDir) {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if tunnel != nil {
|
|
t.Fatal("expected nil tunnel here")
|
|
}
|
|
}
|
|
|
|
func TestFakeWithFailingMkdirAll(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
ctx := context.Background()
|
|
sess := &mockable.Session{}
|
|
tunnel, err := fakeStart(ctx, &Config{
|
|
Session: sess,
|
|
TunnelDir: "testdata",
|
|
testMkdirAll: func(dir string, mode os.FileMode) error {
|
|
return expected
|
|
},
|
|
})
|
|
if !errors.Is(err, expected) {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if tunnel != nil {
|
|
t.Fatal("expected nil tunnel here")
|
|
}
|
|
}
|
|
|
|
func TestFakeSocks5NewFails(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
ctx := context.Background()
|
|
sess := &mockable.Session{}
|
|
tunnel, err := fakeStart(ctx, &Config{
|
|
Session: sess,
|
|
TunnelDir: "testdata",
|
|
testSocks5New: func(conf *socks5.Config) (*socks5.Server, error) {
|
|
return nil, expected
|
|
},
|
|
})
|
|
if !errors.Is(err, expected) {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if tunnel != nil {
|
|
t.Fatal("expected nil tunnel here")
|
|
}
|
|
}
|
|
|
|
func TestFakeNetListenFails(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
ctx := context.Background()
|
|
sess := &mockable.Session{}
|
|
tunnel, err := fakeStart(ctx, &Config{
|
|
Session: sess,
|
|
TunnelDir: "testdata",
|
|
testNetListen: func(network, address string) (net.Listener, error) {
|
|
return nil, expected
|
|
},
|
|
})
|
|
if !errors.Is(err, expected) {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if tunnel != nil {
|
|
t.Fatal("expected nil tunnel here")
|
|
}
|
|
}
|