[TEST] cancel all processes on PrepareTestEnv
(cherry picked from commit aba99ab8fc3b8bc8f55fa0b7d6261eea7b28d7c0)
This commit is contained in:
parent
2879c15858
commit
569b73e495
1 changed files with 38 additions and 0 deletions
|
@ -11,7 +11,9 @@ import (
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
"testing"
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
"code.gitea.io/gitea/models/db"
|
"code.gitea.io/gitea/models/db"
|
||||||
packages_model "code.gitea.io/gitea/models/packages"
|
packages_model "code.gitea.io/gitea/models/packages"
|
||||||
|
@ -20,6 +22,7 @@ import (
|
||||||
"code.gitea.io/gitea/modules/git"
|
"code.gitea.io/gitea/modules/git"
|
||||||
"code.gitea.io/gitea/modules/graceful"
|
"code.gitea.io/gitea/modules/graceful"
|
||||||
"code.gitea.io/gitea/modules/log"
|
"code.gitea.io/gitea/modules/log"
|
||||||
|
"code.gitea.io/gitea/modules/process"
|
||||||
repo_module "code.gitea.io/gitea/modules/repository"
|
repo_module "code.gitea.io/gitea/modules/repository"
|
||||||
"code.gitea.io/gitea/modules/setting"
|
"code.gitea.io/gitea/modules/setting"
|
||||||
"code.gitea.io/gitea/modules/storage"
|
"code.gitea.io/gitea/modules/storage"
|
||||||
|
@ -193,6 +196,36 @@ func PrepareAttachmentsStorage(t testing.TB) {
|
||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// cancelProcesses cancels all processes of the [process.Manager].
|
||||||
|
// Returns immediately if delay is 0, otherwise wait until all processes are done
|
||||||
|
// and fails the test if it takes longer that the given delay.
|
||||||
|
func cancelProcesses(t testing.TB, delay time.Duration) {
|
||||||
|
processManager := process.GetManager()
|
||||||
|
processes, _ := processManager.Processes(true, true)
|
||||||
|
for _, p := range processes {
|
||||||
|
processManager.Cancel(p.PID)
|
||||||
|
t.Logf("PrepareTestEnv:Process %q cancelled", p.Description)
|
||||||
|
}
|
||||||
|
if delay == 0 || len(processes) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
start := time.Now()
|
||||||
|
processes, _ = processManager.Processes(true, true)
|
||||||
|
for len(processes) > 0 {
|
||||||
|
if time.Since(start) > delay {
|
||||||
|
t.Errorf("ERROR PrepareTestEnv: could not cancel all processes within %s", delay)
|
||||||
|
for _, p := range processes {
|
||||||
|
t.Logf("PrepareTestEnv:Remaining Process: %q", p.Description)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
runtime.Gosched() // let the context cancellation propagate
|
||||||
|
processes, _ = processManager.Processes(true, true)
|
||||||
|
}
|
||||||
|
t.Logf("PrepareTestEnv: all processes cancelled within %s", time.Since(start))
|
||||||
|
}
|
||||||
|
|
||||||
func PrepareTestEnv(t testing.TB, skip ...int) func() {
|
func PrepareTestEnv(t testing.TB, skip ...int) func() {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
ourSkip := 1
|
ourSkip := 1
|
||||||
|
@ -201,6 +234,11 @@ func PrepareTestEnv(t testing.TB, skip ...int) func() {
|
||||||
}
|
}
|
||||||
deferFn := PrintCurrentTest(t, ourSkip)
|
deferFn := PrintCurrentTest(t, ourSkip)
|
||||||
|
|
||||||
|
// kill all background processes to prevent them from interfering with the fixture loading
|
||||||
|
// see https://codeberg.org/forgejo/forgejo/issues/2962
|
||||||
|
cancelProcesses(t, time.Second)
|
||||||
|
t.Cleanup(func() { cancelProcesses(t, 0) }) // cancel remaining processes in a non-blocking way
|
||||||
|
|
||||||
// load database fixtures
|
// load database fixtures
|
||||||
assert.NoError(t, unittest.LoadFixtures())
|
assert.NoError(t, unittest.LoadFixtures())
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue