mirror of
https://github.com/datarhei/core.git
synced 2025-10-03 15:16:39 +08:00
42 lines
918 B
Go
42 lines
918 B
Go
package psutil
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestChildren(t *testing.T) {
|
|
p := &procfs{
|
|
children: map[int32][]int32{
|
|
0: {1},
|
|
1: {2, 3},
|
|
2: {4, 5, 6},
|
|
3: {7, 8, 9},
|
|
},
|
|
}
|
|
|
|
require.Equal(t, []int32{1}, p.Children(0))
|
|
require.Equal(t, []int32{2, 3}, p.Children(1))
|
|
require.Equal(t, []int32{4, 5, 6}, p.Children(2))
|
|
require.Equal(t, []int32{7, 8, 9}, p.Children(3))
|
|
require.Equal(t, []int32{}, p.Children(4))
|
|
}
|
|
|
|
func TestAllChildren(t *testing.T) {
|
|
p := &procfs{
|
|
children: map[int32][]int32{
|
|
0: {1},
|
|
1: {2, 3},
|
|
2: {4, 5, 6},
|
|
3: {7, 8, 9},
|
|
},
|
|
}
|
|
|
|
require.Equal(t, []int32{1, 2, 3, 4, 5, 6, 7, 8, 9}, p.AllChildren(0))
|
|
require.Equal(t, []int32{2, 3, 4, 5, 6, 7, 8, 9}, p.AllChildren(1))
|
|
require.Equal(t, []int32{4, 5, 6}, p.AllChildren(2))
|
|
require.Equal(t, []int32{7, 8, 9}, p.AllChildren(3))
|
|
require.Equal(t, []int32{}, p.AllChildren(4))
|
|
}
|