d3c5196474
This diff changes the software name used by unattended runs for which we did not override the default software name (`ooniprobe-cli`). It will become `ooniprobe-cli-unattended`. This software name is in line with the one we use for Android, iOS, and desktop unattended runs. While working in this diff, I introduced string constants for the run types and a string constant for the default software name. See https://github.com/ooni/probe/issues/2081.
65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package nettests
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/ooni/probe-cli/v3/cmd/ooniprobe/internal/database"
|
|
"github.com/ooni/probe-cli/v3/cmd/ooniprobe/internal/ooni"
|
|
"github.com/ooni/probe-cli/v3/internal/model"
|
|
)
|
|
|
|
func copyfile(source, dest string) error {
|
|
data, err := os.ReadFile(source)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return os.WriteFile(dest, data, 0600)
|
|
}
|
|
|
|
func newOONIProbe(t *testing.T) *ooni.Probe {
|
|
homePath, err := ioutil.TempDir("", "ooniprobetests")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
configPath := path.Join(homePath, "config.json")
|
|
testingConfig := path.Join("..", "..", "testdata", "testing-config.json")
|
|
if err := copyfile(testingConfig, configPath); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
probe := ooni.NewProbe(configPath, homePath)
|
|
swName := "ooniprobe-cli-tests"
|
|
swVersion := "3.0.0-alpha"
|
|
err = probe.Init(swName, swVersion)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return probe
|
|
}
|
|
|
|
func TestCreateContext(t *testing.T) {
|
|
newOONIProbe(t)
|
|
}
|
|
|
|
func TestRun(t *testing.T) {
|
|
probe := newOONIProbe(t)
|
|
sess, err := probe.NewSession(context.Background(), model.RunTypeManual)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
network, err := database.CreateNetwork(probe.DB(), sess)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
res, err := database.CreateResult(probe.DB(), probe.Home(), "middlebox", network.ID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
nt := HTTPInvalidRequestLine{}
|
|
ctl := NewController(nt, probe, res, sess)
|
|
nt.Run(ctl)
|
|
}
|