mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2025-10-29 21:41:02 +00:00
fed2d81c44
Co-authored-by: Renovate Bot <forgejo-renovate-action@forgejo.org> Co-committed-by: Renovate Bot <forgejo-renovate-action@forgejo.org>
92 lines
3.4 KiB
Go
92 lines
3.4 KiB
Go
// Copyright 2025 The Forgejo Authors. All rights reserved.
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
package process
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProcessStacktraces(t *testing.T) {
|
|
_, _, finish := GetManager().AddContext(t.Context(), "Normal process")
|
|
defer finish()
|
|
parentCtx, _, finish := GetManager().AddContext(t.Context(), "Children normal process")
|
|
defer finish()
|
|
_, _, finish = GetManager().AddContext(parentCtx, "Children process")
|
|
defer finish()
|
|
_, _, finish = GetManager().AddTypedContext(t.Context(), "System process", SystemProcessType, true)
|
|
defer finish()
|
|
|
|
t.Run("No flat with no system process", func(t *testing.T) {
|
|
processes, processCount, _, err := GetManager().ProcessStacktraces(false, true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 4, processCount)
|
|
assert.Len(t, processes, 2)
|
|
|
|
assert.Equal(t, "Children normal process", processes[0].Description)
|
|
assert.Equal(t, NormalProcessType, processes[0].Type)
|
|
assert.Empty(t, processes[0].ParentPID)
|
|
assert.Len(t, processes[0].Children, 1)
|
|
|
|
assert.Equal(t, "Children process", processes[0].Children[0].Description)
|
|
assert.Equal(t, processes[0].PID, processes[0].Children[0].ParentPID)
|
|
|
|
assert.Equal(t, "Normal process", processes[1].Description)
|
|
assert.Equal(t, NormalProcessType, processes[1].Type)
|
|
assert.Empty(t, processes[1].ParentPID)
|
|
assert.Empty(t, processes[1].Children)
|
|
})
|
|
|
|
t.Run("Flat with no system process", func(t *testing.T) {
|
|
processes, processCount, _, err := GetManager().ProcessStacktraces(true, true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 4, processCount)
|
|
assert.Len(t, processes, 3)
|
|
|
|
assert.Equal(t, "Children process", processes[0].Description)
|
|
assert.Equal(t, NormalProcessType, processes[0].Type)
|
|
assert.Equal(t, processes[1].PID, processes[0].ParentPID)
|
|
assert.Empty(t, processes[0].Children)
|
|
|
|
assert.Equal(t, "Children normal process", processes[1].Description)
|
|
assert.Equal(t, NormalProcessType, processes[1].Type)
|
|
assert.Empty(t, processes[1].ParentPID)
|
|
assert.Empty(t, processes[1].Children)
|
|
|
|
assert.Equal(t, "Normal process", processes[2].Description)
|
|
assert.Equal(t, NormalProcessType, processes[2].Type)
|
|
assert.Empty(t, processes[2].ParentPID)
|
|
assert.Empty(t, processes[2].Children)
|
|
})
|
|
|
|
t.Run("System process", func(t *testing.T) {
|
|
processes, processCount, _, err := GetManager().ProcessStacktraces(false, false)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 4, processCount)
|
|
assert.Len(t, processes, 4)
|
|
|
|
assert.Equal(t, "System process", processes[0].Description)
|
|
assert.Equal(t, SystemProcessType, processes[0].Type)
|
|
assert.Empty(t, processes[0].ParentPID)
|
|
assert.Empty(t, processes[0].Children)
|
|
|
|
assert.Equal(t, "Children normal process", processes[1].Description)
|
|
assert.Equal(t, NormalProcessType, processes[1].Type)
|
|
assert.Empty(t, processes[1].ParentPID)
|
|
assert.Len(t, processes[1].Children, 1)
|
|
|
|
assert.Equal(t, "Normal process", processes[2].Description)
|
|
assert.Equal(t, NormalProcessType, processes[2].Type)
|
|
assert.Empty(t, processes[2].ParentPID)
|
|
assert.Empty(t, processes[2].Children)
|
|
|
|
// This is the "main" pid, testing code always runs in a goroutine.
|
|
assert.Equal(t, "(unassociated)", processes[3].Description)
|
|
assert.Equal(t, NoneProcessType, processes[3].Type)
|
|
assert.Empty(t, processes[3].ParentPID)
|
|
assert.Empty(t, processes[3].Children)
|
|
})
|
|
}
|