f2b6a5972fWork part of: https://github.com/ooni/probe/issues/1733 Spec draft: https://github.com/ooni/spec/pull/219
117 lines
2.7 KiB
Go
117 lines
2.7 KiB
Go
package websteps
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/ooni/probe-cli/v3/internal/errorsx"
|
|
)
|
|
|
|
func TestMeasureSuccess(t *testing.T) {
|
|
req := &CtrlRequest{
|
|
URL: "https://example.com",
|
|
}
|
|
resp, err := Measure(context.Background(), req, &Config{})
|
|
if err != nil {
|
|
t.Fatal("unexpected error")
|
|
}
|
|
if resp == nil {
|
|
t.Fatal("unexpected nil response")
|
|
}
|
|
}
|
|
|
|
type MockChecker struct {
|
|
err error
|
|
}
|
|
|
|
func (c *MockChecker) InitialChecks(URL string) (*url.URL, error) {
|
|
return nil, c.err
|
|
}
|
|
|
|
type MockExplorer struct{}
|
|
|
|
func (c *MockExplorer) Explore(URL *url.URL, headers map[string][]string) ([]*RoundTrip, error) {
|
|
return nil, ErrExpectedExplore
|
|
}
|
|
|
|
type MockGenerator struct{}
|
|
|
|
func (c *MockGenerator) Generate(ctx context.Context, rts []*RoundTrip, clientResolutions []string) ([]*URLMeasurement, error) {
|
|
return nil, ErrExpectedGenerate
|
|
}
|
|
|
|
var ErrExpectedCheck error = errors.New("expected error checker")
|
|
var ErrExpectedExplore error = errors.New("expected error explorer")
|
|
var ErrExpectedGenerate error = errors.New("expected error generator")
|
|
|
|
func TestMeasureInitialChecksFail(t *testing.T) {
|
|
req := &CtrlRequest{
|
|
URL: "https://example.com",
|
|
}
|
|
resp, err := Measure(context.Background(), req, &Config{checker: &MockChecker{err: ErrExpectedCheck}})
|
|
if err == nil {
|
|
t.Fatal("expected an error here")
|
|
}
|
|
if err != ErrExpectedCheck {
|
|
t.Fatal("unexpected error type")
|
|
}
|
|
if resp != nil {
|
|
t.Fatal("resp should be nil")
|
|
}
|
|
}
|
|
|
|
func TestMeasureInitialChecksFailWithNXDOMAIN(t *testing.T) {
|
|
req := &CtrlRequest{
|
|
URL: "https://example.com",
|
|
}
|
|
resp, err := Measure(context.Background(), req, &Config{checker: &MockChecker{err: ErrNoSuchHost}})
|
|
if err != nil {
|
|
t.Fatal("unexpected error")
|
|
}
|
|
if resp == nil {
|
|
t.Fatal("resp should not be nil")
|
|
}
|
|
if len(resp.URLs) != 1 {
|
|
t.Fatal("unexpected number of measurements")
|
|
}
|
|
if resp.URLs[0].DNS == nil {
|
|
t.Fatal("DNS entry should not be nil")
|
|
}
|
|
if *resp.URLs[0].DNS.Failure != errorsx.FailureDNSNXDOMAINError {
|
|
t.Fatal("unexpected failure")
|
|
}
|
|
}
|
|
|
|
func TestMeasureExploreFails(t *testing.T) {
|
|
req := &CtrlRequest{
|
|
URL: "https://example.com",
|
|
}
|
|
resp, err := Measure(context.Background(), req, &Config{explorer: &MockExplorer{}})
|
|
if err == nil {
|
|
t.Fatal("expected an error here")
|
|
}
|
|
if err != ErrInternalServer {
|
|
t.Fatal("unexpected error type")
|
|
}
|
|
if resp != nil {
|
|
t.Fatal("resp should be nil")
|
|
}
|
|
}
|
|
|
|
func TestMeasureGenerateFails(t *testing.T) {
|
|
req := &CtrlRequest{
|
|
URL: "https://example.com",
|
|
}
|
|
resp, err := Measure(context.Background(), req, &Config{generator: &MockGenerator{}})
|
|
if err == nil {
|
|
t.Fatal("expected an error here")
|
|
}
|
|
if err != ErrExpectedGenerate {
|
|
t.Fatal("unexpected error type")
|
|
}
|
|
if resp != nil {
|
|
t.Fatal("resp should be nil")
|
|
}
|
|
}
|