feat(oonirun): improve tests (#915)
See https://github.com/ooni/probe/issues/2184 While there, rename `runtimex.PanicIfFalse` to `runtimex.Assert` (it was about time...)
This commit is contained in:
Simone Basso
GitHub
parent
a8a29cc0dd
commit
d0da224a2a
32 changed files with 1837 additions and 112 deletions
357
internal/model/mocks/session_test.go
Normal file
357
internal/model/mocks/session_test.go
Normal file
|
|
@ -0,0 +1,357 @@
|
|||
package mocks
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"net/url"
|
||||
"testing"
|
||||
|
||||
"github.com/google/go-cmp/cmp"
|
||||
"github.com/ooni/probe-cli/v3/internal/model"
|
||||
"github.com/ooni/probe-cli/v3/internal/runtimex"
|
||||
"github.com/ooni/probe-cli/v3/internal/testingx"
|
||||
)
|
||||
|
||||
func TestSession(t *testing.T) {
|
||||
t.Run("GetTestHelpersByName", func(t *testing.T) {
|
||||
var expect []model.OOAPIService
|
||||
ff := &testingx.FakeFiller{}
|
||||
ff.Fill(&expect)
|
||||
runtimex.Assert(len(expect) > 0, "expected non-empty array")
|
||||
s := &Session{
|
||||
MockGetTestHelpersByName: func(name string) ([]model.OOAPIService, bool) {
|
||||
return expect, len(expect) > 0
|
||||
},
|
||||
}
|
||||
out, good := s.GetTestHelpersByName("xx")
|
||||
if !good {
|
||||
t.Fatal("not good")
|
||||
}
|
||||
if diff := cmp.Diff(expect, out); diff != "" {
|
||||
t.Fatal(diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("DefaultHTTPClient", func(t *testing.T) {
|
||||
expected := &HTTPClient{}
|
||||
s := &Session{
|
||||
MockDefaultHTTPClient: func() model.HTTPClient {
|
||||
return expected
|
||||
},
|
||||
}
|
||||
out := s.DefaultHTTPClient()
|
||||
if expected != out {
|
||||
t.Fatal("unexpected result")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("FetchPsiphonConfig", func(t *testing.T) {
|
||||
var expected []byte
|
||||
ff := &testingx.FakeFiller{}
|
||||
ff.Fill(&expected)
|
||||
runtimex.Assert(len(expected) > 0, "expected nonempty list")
|
||||
s := &Session{
|
||||
MockFetchPsiphonConfig: func(ctx context.Context) ([]byte, error) {
|
||||
return expected, nil
|
||||
},
|
||||
}
|
||||
out, err := s.FetchPsiphonConfig(context.Background())
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if diff := cmp.Diff(expected, out); diff != "" {
|
||||
t.Fatal(diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("FetchTorTargets", func(t *testing.T) {
|
||||
expected := errors.New("mocked err")
|
||||
s := &Session{
|
||||
MockFetchTorTargets: func(ctx context.Context, cc string) (map[string]model.OOAPITorTarget, error) {
|
||||
return nil, expected
|
||||
},
|
||||
}
|
||||
out, err := s.FetchTorTargets(context.Background(), "IT")
|
||||
if !errors.Is(err, expected) {
|
||||
t.Fatal("unexpected err", err)
|
||||
}
|
||||
if len(out) > 0 {
|
||||
t.Fatal("expected empty out")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("FetchURLList", func(t *testing.T) {
|
||||
expected := errors.New("mocked err")
|
||||
s := &Session{
|
||||
MockFetchURLList: func(ctx context.Context, config model.OOAPIURLListConfig) ([]model.OOAPIURLInfo, error) {
|
||||
return nil, expected
|
||||
},
|
||||
}
|
||||
out, err := s.FetchURLList(context.Background(), model.OOAPIURLListConfig{})
|
||||
if !errors.Is(err, expected) {
|
||||
t.Fatal("unexpected err", err)
|
||||
}
|
||||
if len(out) > 0 {
|
||||
t.Fatal("expected empty out")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("KeyValueStore", func(t *testing.T) {
|
||||
expect := &KeyValueStore{}
|
||||
s := &Session{
|
||||
MockKeyValueStore: func() model.KeyValueStore {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.KeyValueStore()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("Logger", func(t *testing.T) {
|
||||
expect := &Logger{}
|
||||
s := &Session{
|
||||
MockLogger: func() model.Logger {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.Logger()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("MaybeResolverIP", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockMaybeResolverIP: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.MaybeResolverIP()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ProbeASNString", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockProbeASNString: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.ProbeASNString()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ProbeCC", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockProbeCC: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.ProbeCC()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ProbeIP", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockProbeIP: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.ProbeIP()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ProbeNetworkName", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockProbeNetworkName: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.ProbeNetworkName()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ProxyURL", func(t *testing.T) {
|
||||
expect := &url.URL{Scheme: "xx"}
|
||||
s := &Session{
|
||||
MockProxyURL: func() *url.URL {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.ProxyURL()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ResolverIP", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockResolverIP: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.ResolverIP()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("SoftwareName", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockSoftwareName: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.SoftwareName()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("SoftwareVersion", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockSoftwareVersion: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.SoftwareVersion()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TempDir", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockTempDir: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.TempDir()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TorArgs", func(t *testing.T) {
|
||||
var expect []string
|
||||
ff := &testingx.FakeFiller{}
|
||||
ff.Fill(&expect)
|
||||
runtimex.Assert(len(expect) > 0, "expected non empty slice")
|
||||
s := &Session{
|
||||
MockTorArgs: func() []string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.TorArgs()
|
||||
if diff := cmp.Diff(expect, out); diff != "" {
|
||||
t.Fatal(diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TorBinary", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockTorBinary: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.TorBinary()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TunnelDir", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockTunnelDir: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.TunnelDir()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("UserAgent", func(t *testing.T) {
|
||||
expect := "xx"
|
||||
s := &Session{
|
||||
MockUserAgent: func() string {
|
||||
return expect
|
||||
},
|
||||
}
|
||||
out := s.UserAgent()
|
||||
if out != expect {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("NewExperimentBuilder", func(t *testing.T) {
|
||||
eb := &ExperimentBuilder{}
|
||||
s := &Session{
|
||||
MockNewExperimentBuilder: func(name string) (model.ExperimentBuilder, error) {
|
||||
return eb, nil
|
||||
},
|
||||
}
|
||||
out, err := s.NewExperimentBuilder("x")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if out != eb {
|
||||
t.Fatal("invalid output")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("NewSubmitter", func(t *testing.T) {
|
||||
expected := errors.New("mocked err")
|
||||
s := &Session{
|
||||
MockNewSubmitter: func(ctx context.Context) (model.Submitter, error) {
|
||||
return nil, expected
|
||||
},
|
||||
}
|
||||
out, err := s.NewSubmitter(context.Background())
|
||||
if !errors.Is(err, expected) {
|
||||
t.Fatal("unexpected err")
|
||||
}
|
||||
if out != nil {
|
||||
t.Fatal("unexpected out")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("CheckIn", func(t *testing.T) {
|
||||
expected := errors.New("mocked err")
|
||||
s := &Session{
|
||||
MockCheckIn: func(ctx context.Context, config *model.OOAPICheckInConfig) (*model.OOAPICheckInInfo, error) {
|
||||
return nil, expected
|
||||
},
|
||||
}
|
||||
out, err := s.CheckIn(context.Background(), &model.OOAPICheckInConfig{})
|
||||
if !errors.Is(err, expected) {
|
||||
t.Fatal("unexpected err")
|
||||
}
|
||||
if out != nil {
|
||||
t.Fatal("unexpected out")
|
||||
}
|
||||
})
|
||||
}
|
||||
Loading…
Reference in a new issue