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.
86 lines
2.1 KiB
Go
86 lines
2.1 KiB
Go
package probeservices_test
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/ooni/probe-cli/v3/internal/engine/probeservices"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/probeservices/testorchestra"
|
|
)
|
|
|
|
func TestFetchTorTargets(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skip test in short mode")
|
|
}
|
|
clnt := newclient()
|
|
if err := clnt.MaybeRegister(context.Background(), testorchestra.MetadataFixture()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := clnt.MaybeLogin(context.Background()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data, err := clnt.FetchTorTargets(context.Background(), "ZZ")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if data == nil || len(data) <= 0 {
|
|
t.Fatal("invalid data")
|
|
}
|
|
}
|
|
|
|
func TestFetchTorTargetsNotRegistered(t *testing.T) {
|
|
clnt := newclient()
|
|
state := probeservices.State{
|
|
// Explicitly empty so the test is more clear
|
|
}
|
|
if err := clnt.StateFile.Set(state); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data, err := clnt.FetchTorTargets(context.Background(), "ZZ")
|
|
if err == nil {
|
|
t.Fatal("expected an error here")
|
|
}
|
|
if data != nil {
|
|
t.Fatal("expected nil data here")
|
|
}
|
|
}
|
|
|
|
type FetchTorTargetsHTTPTransport struct {
|
|
Response *http.Response
|
|
}
|
|
|
|
func (clnt *FetchTorTargetsHTTPTransport) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
resp, err := http.DefaultTransport.RoundTrip(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if req.URL.Path == "/api/v1/test-list/tor-targets" {
|
|
clnt.Response = resp
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func TestFetchTorTargetsSetsQueryString(t *testing.T) {
|
|
clnt := newclient()
|
|
txp := new(FetchTorTargetsHTTPTransport)
|
|
clnt.HTTPClient = &http.Client{Transport: txp}
|
|
if err := clnt.MaybeRegister(context.Background(), testorchestra.MetadataFixture()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := clnt.MaybeLogin(context.Background()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data, err := clnt.FetchTorTargets(context.Background(), "ZZ")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if data == nil || len(data) <= 0 {
|
|
t.Fatal("invalid data")
|
|
}
|
|
requestURL := txp.Response.Request.URL
|
|
if requestURL.Query().Get("country_code") != "ZZ" {
|
|
t.Fatal("invalid country code")
|
|
}
|
|
}
|