Skip to content

Commit

Permalink
cleanup: only define ErrNoAvailableTestHelpers once (#1583)
Browse files Browse the repository at this point in the history
This is an easy cleanup I identified while working on
ooni/probe#2725.

Closes ooni/probe#2725.
  • Loading branch information
bassosimone authored May 2, 2024
1 parent cc25204 commit 1056c98
Show file tree
Hide file tree
Showing 8 changed files with 10 additions and 20 deletions.
5 changes: 1 addition & 4 deletions internal/experiment/hhfm/hhfm.go
Original file line number Diff line number Diff line change
Expand Up @@ -81,9 +81,6 @@ func (m Measurer) ExperimentVersion() string {
}

var (
// ErrNoAvailableTestHelpers is emitted when there are no available test helpers.
ErrNoAvailableTestHelpers = errors.New("no available helpers")

// ErrInvalidHelperType is emitted when the helper type is invalid.
ErrInvalidHelperType = errors.New("invalid helper type")
)
Expand All @@ -104,7 +101,7 @@ func (m Measurer) Run(ctx context.Context, args *model.ExperimentArgs) error {
const helperName = "http-return-json-headers"
helpers, ok := sess.GetTestHelpersByName(helperName)
if !ok || len(helpers) < 1 {
return ErrNoAvailableTestHelpers
return model.ErrNoAvailableTestHelpers
}
helper := helpers[0]
if helper.Type != "legacy" {
Expand Down
4 changes: 2 additions & 2 deletions internal/experiment/hhfm/hhfm_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -272,7 +272,7 @@ func TestNoHelpers(t *testing.T) {
Session: sess,
}
err := measurer.Run(ctx, args)
if !errors.Is(err, hhfm.ErrNoAvailableTestHelpers) {
if !errors.Is(err, model.ErrNoAvailableTestHelpers) {
t.Fatal("not the error we expected")
}
tk := measurement.TestKeys.(*hhfm.TestKeys)
Expand Down Expand Up @@ -327,7 +327,7 @@ func TestNoActualHelpersInList(t *testing.T) {
Session: sess,
}
err := measurer.Run(ctx, args)
if !errors.Is(err, hhfm.ErrNoAvailableTestHelpers) {
if !errors.Is(err, model.ErrNoAvailableTestHelpers) {
t.Fatal("not the error we expected")
}
tk := measurement.TestKeys.(*hhfm.TestKeys)
Expand Down
5 changes: 1 addition & 4 deletions internal/experiment/hirl/hirl.go
Original file line number Diff line number Diff line change
Expand Up @@ -68,9 +68,6 @@ func (m Measurer) ExperimentVersion() string {
}

var (
// ErrNoAvailableTestHelpers is emitted when there are no available test helpers.
ErrNoAvailableTestHelpers = errors.New("no available helpers")

// ErrInvalidHelperType is emitted when the helper type is invalid.
ErrInvalidHelperType = errors.New("invalid helper type")

Expand All @@ -91,7 +88,7 @@ func (m Measurer) Run(ctx context.Context, args *model.ExperimentArgs) error {
const helperName = "tcp-echo"
helpers, ok := sess.GetTestHelpersByName(helperName)
if !ok || len(helpers) < 1 {
return ErrNoAvailableTestHelpers
return model.ErrNoAvailableTestHelpers
}
helper := helpers[0]
if helper.Type != "legacy" {
Expand Down
4 changes: 2 additions & 2 deletions internal/experiment/hirl/hirl_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -302,7 +302,7 @@ func TestNoHelpers(t *testing.T) {
Session: sess,
}
err := measurer.Run(ctx, args)
if !errors.Is(err, hirl.ErrNoAvailableTestHelpers) {
if !errors.Is(err, model.ErrNoAvailableTestHelpers) {
t.Fatal("not the error we expected")
}
tk := measurement.TestKeys.(*hirl.TestKeys)
Expand Down Expand Up @@ -340,7 +340,7 @@ func TestNoActualHelperInList(t *testing.T) {
Session: sess,
}
err := measurer.Run(ctx, args)
if !errors.Is(err, hirl.ErrNoAvailableTestHelpers) {
if !errors.Is(err, model.ErrNoAvailableTestHelpers) {
t.Fatal("not the error we expected")
}
tk := measurement.TestKeys.(*hirl.TestKeys)
Expand Down
5 changes: 1 addition & 4 deletions internal/experiment/webconnectivity/webconnectivity.go
Original file line number Diff line number Diff line change
Expand Up @@ -94,9 +94,6 @@ func (m Measurer) ExperimentVersion() string {
}

var (
// ErrNoAvailableTestHelpers is emitted when there are no available test helpers.
ErrNoAvailableTestHelpers = errors.New("no available helpers")

// ErrNoInput indicates that no input was provided
ErrNoInput = errors.New("no input provided")

Expand Down Expand Up @@ -145,7 +142,7 @@ func (m Measurer) Run(ctx context.Context, args *model.ExperimentArgs) error {
// 1. find test helper
testhelpers, _ := sess.GetTestHelpersByName("web-connectivity")
if len(testhelpers) < 1 {
return ErrNoAvailableTestHelpers
return model.ErrNoAvailableTestHelpers
}
// 2. perform the DNS lookup step
dnsBegin := time.Now()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -206,7 +206,7 @@ func TestMeasureWithNoAvailableTestHelpers(t *testing.T) {
Session: sess,
}
err := measurer.Run(ctx, args)
if !errors.Is(err, webconnectivity.ErrNoAvailableTestHelpers) {
if !errors.Is(err, model.ErrNoAvailableTestHelpers) {
t.Fatal(err)
}
tk := measurement.TestKeys.(*webconnectivity.TestKeys)
Expand Down
3 changes: 1 addition & 2 deletions internal/experiment/webconnectivitylte/measurer.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@ import (
"net/http/cookiejar"
"sync"

"github.com/ooni/probe-cli/v3/internal/experiment/webconnectivity"
"github.com/ooni/probe-cli/v3/internal/inputparser"
"github.com/ooni/probe-cli/v3/internal/model"
"github.com/ooni/probe-cli/v3/internal/webconnectivityalgo"
Expand Down Expand Up @@ -108,7 +107,7 @@ func (m *Measurer) Run(ctx context.Context, args *model.ExperimentArgs) error {
testhelpers, _ := sess.GetTestHelpersByName("web-connectivity")
if len(testhelpers) < 1 {
sess.Logger().Warnf("continuing without a valid TH address")
tk.SetControlFailure(webconnectivity.ErrNoAvailableTestHelpers)
tk.SetControlFailure(model.ErrNoAvailableTestHelpers)
}

registerExtensions(measurement)
Expand Down
2 changes: 1 addition & 1 deletion internal/webconnectivityalgo/calltesthelpers.go
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ import (
// given request object, the given list of available test helpers, and the given session.
//
// If the list of test helpers is empty this function immediately returns nil, zero,
// and the [ErrNoAvailableTestHelpers] error to the caller.
// and the [model.ErrNoAvailableTestHelpers] error to the caller.
//
// In case of any other failure, this function returns nil, zero, and an error.
//
Expand Down

0 comments on commit 1056c98

Please sign in to comment.