ooni-probe-cli/cmd/ooniprobe/internal/nettests/nettests_test.go

package nettests

import (
	"context"
	"io/ioutil"
	"path"
	"testing"

	"github.com/ooni/probe-cli/v3/cmd/ooniprobe/internal/database"
	"github.com/ooni/probe-cli/v3/cmd/ooniprobe/internal/ooni"
)

func copyfile(source, dest string) error {
	data, err := ioutil.ReadFile(source)
	if err != nil {
		return err
	}
	return ioutil.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())
	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)
}