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

package nettests

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

	"github.com/ooni/probe-cli/internal/database"
	"github.com/ooni/probe-cli/internal/ooni"
	"github.com/ooni/probe-cli/internal/utils/shutil"
)

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")
	shutil.Copy(testingConfig, configPath, false)
	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()
	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)
}