diff --git a/tests/slo/native/query/main.go b/tests/slo/native/query/main.go index fb7301bea..052106153 100644 --- a/tests/slo/native/query/main.go +++ b/tests/slo/native/query/main.go @@ -37,6 +37,11 @@ func main() { ctx, cancel = context.WithTimeout(ctx, time.Duration(cfg.Time)*time.Second) defer cancel() + go func() { + <-ctx.Done() + log.Println("exiting...") + }() + s, err := NewStorage(ctx, cfg, cfg.ReadRPS+cfg.WriteRPS, label) if err != nil { panic(fmt.Errorf("create storage failed: %w", err))