a5f4fc997c
* doc: merge the engine and the cli readmes Part of https://github.com/ooni/probe/issues/1335 * refactor: we don't wanna export pkg/oonimkall/tasks See https://github.com/ooni/probe/issues/1335
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package tasks
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"testing"
|
|
|
|
engine "github.com/ooni/probe-cli/v3/internal/engine"
|
|
)
|
|
|
|
func TestMeasurementSubmissionEventName(t *testing.T) {
|
|
if measurementSubmissionEventName(nil) != statusMeasurementSubmission {
|
|
t.Fatal("unexpected submission event name")
|
|
}
|
|
if measurementSubmissionEventName(errors.New("mocked error")) != failureMeasurementSubmission {
|
|
t.Fatal("unexpected submission event name")
|
|
}
|
|
}
|
|
|
|
func TestMeasurementSubmissionFailure(t *testing.T) {
|
|
if measurementSubmissionFailure(nil) != "" {
|
|
t.Fatal("unexpected submission failure")
|
|
}
|
|
if measurementSubmissionFailure(errors.New("mocked error")) != "mocked error" {
|
|
t.Fatal("unexpected submission failure")
|
|
}
|
|
}
|
|
|
|
func TestRunnerMaybeLookupLocationFailure(t *testing.T) {
|
|
out := make(chan *Event)
|
|
settings := &Settings{
|
|
AssetsDir: "../../testdata/oonimkall/assets",
|
|
Name: "Example",
|
|
Options: SettingsOptions{
|
|
SoftwareName: "oonimkall-test",
|
|
SoftwareVersion: "0.1.0",
|
|
},
|
|
StateDir: "../../testdata/oonimkall/state",
|
|
Version: 1,
|
|
}
|
|
seench := make(chan int64)
|
|
go func() {
|
|
var seen int64
|
|
for ev := range out {
|
|
switch ev.Key {
|
|
case "failure.ip_lookup", "failure.asn_lookup",
|
|
"failure.cc_lookup", "failure.resolver_lookup":
|
|
seen++
|
|
case "status.progress":
|
|
evv := ev.Value.(EventStatusProgress)
|
|
if evv.Percentage >= 0.2 {
|
|
panic(fmt.Sprintf("too much progress: %+v", ev))
|
|
}
|
|
case "status.queued", "status.started", "status.end":
|
|
default:
|
|
panic(fmt.Sprintf("unexpected key: %s", ev.Key))
|
|
}
|
|
}
|
|
seench <- seen
|
|
}()
|
|
expected := errors.New("mocked error")
|
|
r := NewRunner(settings, out)
|
|
r.maybeLookupLocation = func(*engine.Session) error {
|
|
return expected
|
|
}
|
|
r.Run(context.Background())
|
|
close(out)
|
|
if n := <-seench; n != 4 {
|
|
t.Fatal("unexpected number of events")
|
|
}
|
|
}
|