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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
package intelrdt
import (
"io/ioutil"
"os"
"path/filepath"
"strconv"
"strings"
"testing"
)
func TestParseMonFeatures(t *testing.T) {
t.Run("All features available", func(t *testing.T) {
parsedMonFeatures, err := parseMonFeatures(
strings.NewReader("mbm_total_bytes\nmbm_local_bytes\nllc_occupancy"))
if err != nil {
t.Errorf("Error while parsing mon features err = %v", err)
}
expectedMonFeatures := monFeatures{true, true, true}
if parsedMonFeatures != expectedMonFeatures {
t.Error("Cannot gather all features!")
}
})
t.Run("No features available", func(t *testing.T) {
parsedMonFeatures, err := parseMonFeatures(strings.NewReader(""))
if err != nil {
t.Errorf("Error while parsing mon features err = %v", err)
}
expectedMonFeatures := monFeatures{false, false, false}
if parsedMonFeatures != expectedMonFeatures {
t.Error("Expected no features available but there is any!")
}
})
}
func mockResctrlL3_MON(NUMANodes []string, mocks map[string]uint64) (string, error) {
testDir, err := ioutil.TempDir("", "rdt_mbm_test")
if err != nil {
return "", err
}
monDataPath := filepath.Join(testDir, "mon_data")
for _, numa := range NUMANodes {
numaPath := filepath.Join(monDataPath, numa)
err = os.MkdirAll(numaPath, os.ModePerm)
if err != nil {
return "", err
}
for fileName, value := range mocks {
err := ioutil.WriteFile(filepath.Join(numaPath, fileName), []byte(strconv.FormatUint(value, 10)), 0o644)
if err != nil {
return "", err
}
}
}
return testDir, nil
}
func TestGetMonitoringStats(t *testing.T) {
enabledMonFeatures.mbmTotalBytes = true
enabledMonFeatures.mbmLocalBytes = true
enabledMonFeatures.llcOccupancy = true
mbmEnabled = true
cmtEnabled = true
mocksNUMANodesToCreate := []string{"mon_l3_00", "mon_l3_01"}
mocksFilesToCreate := map[string]uint64{
"mbm_total_bytes": 9123911,
"mbm_local_bytes": 2361361,
"llc_occupancy": 123331,
}
mockedL3_MON, err := mockResctrlL3_MON(mocksNUMANodesToCreate, mocksFilesToCreate)
defer func() {
err := os.RemoveAll(mockedL3_MON)
if err != nil {
t.Fatal(err)
}
}()
if err != nil {
t.Fatal(err)
}
t.Run("Gather monitoring stats", func(t *testing.T) {
var stats Stats
err := getMonitoringStats(mockedL3_MON, &stats)
if err != nil {
t.Fatal(err)
}
expectedMBMStats := MBMNumaNodeStats{
MBMTotalBytes: mocksFilesToCreate["mbm_total_bytes"],
MBMLocalBytes: mocksFilesToCreate["mbm_local_bytes"],
}
expectedCMTStats := CMTNumaNodeStats{LLCOccupancy: mocksFilesToCreate["llc_occupancy"]}
for _, gotMBMStat := range *stats.MBMStats {
checkMBMStatCorrection(gotMBMStat, expectedMBMStats, t)
}
for _, gotCMTStat := range *stats.CMTStats {
checkCMTStatCorrection(gotCMTStat, expectedCMTStats, t)
}
})
}
|
