fixture_test.go

package turtle

import (
	"context"
	"fmt"
	"net/url"
	"os"
	"os/signal"
	"strconv"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

func getTestDuration(t testing.TB) time.Duration {
	t.Helper()

	const envKey = "TURTLE_TEST_DURATION"

	if v := os.Getenv(envKey); v != "" {
		d, err := time.ParseDuration(v)
		require.NoError(t, err, "invalid value for %q", envKey)
		return d
	}

	return 10 * time.Second
}

func getTestConnections(t testing.TB) int {
	t.Helper()

	const envKey = "TURTLE_TEST_CONNECTIONS"

	if v := os.Getenv(envKey); v != "" {
		d, err := strconv.ParseInt(v, 10, 32)
		require.NoError(t, err, "invalid value for %q", envKey)
		return int(d)
	}

	return 10
}

func getTestTarget(t testing.TB, u url.URL) Target {
	t.Helper()

	return Target{
		Url:         u,
		Duration:    getTestDuration(t),
		Connections: getTestConnections(t),
	}
}

func getTestContext() (context.Context, context.CancelFunc) {
	return signal.NotifyContext(context.Background(), os.Interrupt)
}

func eventsSeqToString(events []Event) string {
	if len(events) < 1 {
		return "<empty events>"
	}

	firstEvent := events[0]
	rv := fmt.Sprintf("\n%s (0)", firstEvent.Name)
	for idx := 1; idx < len(events); idx++ {
		l, e := events[idx-1], events[idx]
		d := e.At.Sub(l.At).Round(time.Millisecond)
		rv += fmt.Sprintf(" -> %s (%s)", e.Name, d)
		if (idx+1)%3 == 0 {
			rv += "\n"
		}
	}
	return rv
}