4eeadd06a5
* refactor: move more commands to internal/cmd Part of https://github.com/ooni/probe/issues/1335. We would like all commands to be at the same level of engine rather than inside engine (now that we can do it). * fix: update .gitignore * refactor: also move jafar outside engine * We should be good now?
62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
package probeservices_test
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/apex/log"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/atomicx"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/httpx"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/kvstore"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/probeservices"
|
|
)
|
|
|
|
func TestCheckReportIDWorkingAsIntended(t *testing.T) {
|
|
client := probeservices.Client{
|
|
Client: httpx.Client{
|
|
BaseURL: "https://ams-pg.ooni.org/",
|
|
HTTPClient: http.DefaultClient,
|
|
Logger: log.Log,
|
|
UserAgent: "miniooni/0.1.0-dev",
|
|
},
|
|
LoginCalls: atomicx.NewInt64(),
|
|
RegisterCalls: atomicx.NewInt64(),
|
|
StateFile: probeservices.NewStateFile(kvstore.NewMemoryKeyValueStore()),
|
|
}
|
|
reportID := `20201209T052225Z_urlgetter_IT_30722_n1_E1VUhMz08SEkgYFU`
|
|
ctx := context.Background()
|
|
found, err := client.CheckReportID(ctx, reportID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if found != true {
|
|
t.Fatal("unexpected found value")
|
|
}
|
|
}
|
|
|
|
func TestCheckReportIDWorkingWithCancelledContext(t *testing.T) {
|
|
client := probeservices.Client{
|
|
Client: httpx.Client{
|
|
BaseURL: "https://ams-pg.ooni.org/",
|
|
HTTPClient: http.DefaultClient,
|
|
Logger: log.Log,
|
|
UserAgent: "miniooni/0.1.0-dev",
|
|
},
|
|
LoginCalls: atomicx.NewInt64(),
|
|
RegisterCalls: atomicx.NewInt64(),
|
|
StateFile: probeservices.NewStateFile(kvstore.NewMemoryKeyValueStore()),
|
|
}
|
|
reportID := `20201209T052225Z_urlgetter_IT_30722_n1_E1VUhMz08SEkgYFU`
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel() // fail immediately
|
|
found, err := client.CheckReportID(ctx, reportID)
|
|
if !errors.Is(err, context.Canceled) {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
if found != false {
|
|
t.Fatal("unexpected found value")
|
|
}
|
|
}
|