Test suite expects static Config not depending on input
This commit is contained in:
parent
9783e3bb47
commit
c95fb894f0
|
@ -8,7 +8,6 @@ import (
|
|||
"net"
|
||||
"net/smtp"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/ooni/probe-cli/v3/internal/engine/experiment/urlgetter"
|
||||
|
@ -26,9 +25,6 @@ var (
|
|||
|
||||
// errInvalidScheme indicates that the scheme is invalid
|
||||
errInvalidScheme = errors.New("scheme must be smtp(s)")
|
||||
|
||||
// errInvalidPort indicates that the port provided could not be parsed as an int
|
||||
errInvalidPort = errors.New("Port number is not a valid integer")
|
||||
)
|
||||
|
||||
const (
|
||||
|
@ -37,14 +33,16 @@ const (
|
|||
)
|
||||
|
||||
// Config contains the experiment config.
|
||||
type Config struct {
|
||||
type Config struct{}
|
||||
|
||||
type RuntimeConfig struct {
|
||||
host string
|
||||
port string
|
||||
forced_tls bool
|
||||
noop_count uint8
|
||||
}
|
||||
|
||||
func config(input model.MeasurementTarget) (*Config, error) {
|
||||
func config(input model.MeasurementTarget) (*RuntimeConfig, error) {
|
||||
if input == "" {
|
||||
// TODO: static input data (eg. gmail/riseup..)
|
||||
return nil, errNoInputProvided
|
||||
|
@ -68,16 +66,11 @@ func config(input model.MeasurementTarget) (*Config, error) {
|
|||
port = "465"
|
||||
}
|
||||
} else {
|
||||
// Check that requested port is a valid integer
|
||||
_, err := strconv.Atoi(parsed.Port())
|
||||
if err != nil {
|
||||
return nil, errInvalidPort
|
||||
} else {
|
||||
port = parsed.Port()
|
||||
}
|
||||
// Valid port is checked by URL parsing
|
||||
port = parsed.Port()
|
||||
}
|
||||
|
||||
valid_config := Config{
|
||||
valid_config := RuntimeConfig{
|
||||
host: parsed.Hostname(),
|
||||
forced_tls: parsed.Scheme == "smtps",
|
||||
port: port,
|
||||
|
@ -321,7 +314,6 @@ func (m Measurer) Run(
|
|||
ctx context.Context, sess model.ExperimentSession,
|
||||
measurement *model.Measurement, callbacks model.ExperimentCallbacks,
|
||||
) error {
|
||||
|
||||
log := sess.Logger()
|
||||
trace := measurexlite.NewTrace(0, measurement.MeasurementStartTimeSaved)
|
||||
|
||||
|
@ -372,7 +364,7 @@ func (m Measurer) Run(
|
|||
}
|
||||
|
||||
// Try EHLO + NoOps
|
||||
if !tcp_session.smtp("localhost", 10) {
|
||||
if !tcp_session.smtp("localhost", config.noop_count) {
|
||||
continue
|
||||
}
|
||||
} else {
|
||||
|
@ -386,7 +378,7 @@ func (m Measurer) Run(
|
|||
continue
|
||||
}
|
||||
|
||||
if !tcp_session.smtp("localhost", 10) {
|
||||
if !tcp_session.smtp("localhost", config.noop_count) {
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue
Block a user