1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
package crit
import (
"testing"
)
func TestFindPs(t *testing.T) {
root := &PsTree{
PID: 1,
Children: []*PsTree{
{
PID: 2,
Children: []*PsTree{
{
PID: 3,
},
},
},
{
PID: 4,
Children: []*PsTree{
{
PID: 5,
},
},
},
},
}
// Test Case 1: Find an existing process with a valid PID
ps := root.FindPs(3)
if ps == nil {
t.Errorf("FindPs(3) returned nil, expected a valid process")
}
if ps != nil && ps.PID != 3 {
t.Errorf("FindPs(3) returned a process with PID %d, expected 3", ps.PID)
}
// Test Case 2: Find a non-existing process with an invalid PID
nonExistentPID := uint32(999)
notFoundProcess := root.FindPs(nonExistentPID)
if notFoundProcess != nil {
t.Errorf("FindPs(%d) returned a process, expected nil", nonExistentPID)
}
}
func TestCountBytes(t *testing.T) {
tests := []struct {
input int64
want string
}{
{1000, "1000 B"},
{5120, "5.0 KB"},
{100000, "97.7 KB"},
}
for _, test := range tests {
got := countBytes(test.input)
if got != test.want {
t.Errorf("want: %s, got: %s", test.want, got)
}
}
}
func TestProcessIP(t *testing.T) {
tests := []struct {
input []uint32
want string
}{
{[]uint32{}, ""},
{[]uint32{0}, "0.0.0.0"},
{[]uint32{16777343}, "127.0.0.1"},
{[]uint32{0, 0, 0, 0}, "::"},
{[]uint32{0, 0, 4294901760, 16777343}, "7f00:1::"},
}
for _, test := range tests {
got := processIP(test.input)
if got != test.want {
t.Errorf("want: %s, got: %s", test.want, got)
}
}
}
|
