mirror of
https://gitea.com/gitea/act_runner.git
synced 2026-10-29 11:26:31 +08:00
c3c647ef83
The test context is cancelled when the test ends, so anything still holding it stops instead of leaking into the next test. Uses inside `t.Cleanup`, `defer` and `go func` blocks are left alone, as the context is already cancelled there. Stacked on https://gitea.com/gitea/runner/pulls/1305. Assisted-by: Codet Co-authored-by: bircni <bircni@icloud.com> Reviewed-on: https://gitea.com/gitea/runner/pulls/1306 Reviewed-by: bircni <bircni@icloud.com>
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
// Copyright 2026 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package common
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"gitea.com/gitea/runner/act/common/logging"
|
|
"gitea.com/gitea/runner/act/common/logging/logtest"
|
|
)
|
|
|
|
func TestDryrunContext(t *testing.T) {
|
|
ctx := t.Context()
|
|
if Dryrun(ctx) {
|
|
t.Fatal("plain context should not be dryrun")
|
|
}
|
|
if !Dryrun(WithDryrun(ctx, true)) {
|
|
t.Fatal("WithDryrun(true) should set dryrun")
|
|
}
|
|
if Dryrun(WithDryrun(ctx, false)) {
|
|
t.Fatal("WithDryrun(false) should clear dryrun")
|
|
}
|
|
}
|
|
|
|
func TestJobErrorContainer(t *testing.T) {
|
|
ctx := t.Context()
|
|
err := errors.New("job failed")
|
|
|
|
SetJobError(ctx, err)
|
|
if got := JobError(ctx); got != nil {
|
|
t.Fatalf("JobError without container = %v, want nil", got)
|
|
}
|
|
|
|
ctx = WithJobErrorContainer(ctx)
|
|
SetJobError(ctx, err)
|
|
if got := JobError(ctx); !errors.Is(got, err) {
|
|
t.Fatalf("JobError = %v, want %v", got, err)
|
|
}
|
|
}
|
|
|
|
func TestLoggerAndHandlerContext(t *testing.T) {
|
|
ctx := t.Context()
|
|
if Logger(ctx) != logging.Default() {
|
|
t.Fatal("plain context should use the default logger")
|
|
}
|
|
if LoggerHandler(ctx) != nil {
|
|
t.Fatal("plain context should not have a logger handler")
|
|
}
|
|
if StepIDs(ctx) != nil {
|
|
t.Fatal("plain context should not have step IDs")
|
|
}
|
|
|
|
logger := logging.New(logtest.New())
|
|
ctx = WithLogger(ctx, logger)
|
|
if Logger(ctx) != logger {
|
|
t.Fatal("WithLogger should set logger")
|
|
}
|
|
|
|
handler := logtest.New()
|
|
ctx = WithLoggerHandler(ctx, handler)
|
|
if LoggerHandler(ctx) != handler {
|
|
t.Fatal("WithLoggerHandler should set handler")
|
|
}
|
|
}
|