Something went wrong. Try again.
Something went wrong. Try again.
2.1 kB · 56 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657package config
import ( "context" "testing" "time"
"github.com/stretchr/testify/require" urfavecli "github.com/urfave/cli/v3")
// flagRun builds a command the way `streamplace sync` is built -- the command// an operator uses to warm an index -- and runs it with the given arguments,// handing back the CLI the flags landed in.func flagRun(t *testing.T, args ...string) *CLI { t.Helper() cli := &CLI{} cmd := cli.NewCommand("sync") cmd.Action = func(context.Context, *urfavecli.Command) error { return nil } require.NoError(t, cmd.Run(context.Background(), append([]string{"sync"}, args...))) return cli}
// TestSweepIntervalFlag: how often a node re-checks every repo it indexes is an// operator's decision -- and setting it to zero, which turns the periodic sweep// off entirely, has to be expressible.func TestSweepIntervalFlag(t *testing.T) { require.Equal(t, DefaultSweepInterval, flagRun(t).SweepInterval, "unset is the default") require.Equal(t, 90*time.Minute, flagRun(t, "--sweep-interval", "90m").SweepInterval) require.Equal(t, time.Duration(0), flagRun(t, "--sweep-interval=0").SweepInterval)
t.Setenv("SP_SWEEP_INTERVAL", "2h") require.Equal(t, 2*time.Hour, flagRun(t).SweepInterval)}
// TestSweepConcurrencyFlag: the sweep's host-lane cap is settable from the// command line and the environment, and every command built from NewCommand --// including `streamplace sync`, which is the one an operator uses to warm an// index -- gets it.func TestSweepConcurrencyFlag(t *testing.T) { run := func(t *testing.T, args ...string) *CLI { t.Helper() cli := &CLI{} cmd := cli.NewCommand("sync") cmd.Action = func(context.Context, *urfavecli.Command) error { return nil } require.NoError(t, cmd.Run(context.Background(), append([]string{"sync"}, args...))) return cli }
require.Equal(t, DefaultSweepConcurrency, run(t).SweepConcurrency, "unset is the default") require.Equal(t, 8, run(t, "--sweep-concurrency", "8").SweepConcurrency) require.Equal(t, 8, run(t, "--sweep-concurrency=8").SweepConcurrency)
t.Setenv("SP_SWEEP_CONCURRENCY", "12") require.Equal(t, 12, run(t).SweepConcurrency)}