1 package intelrdt
2
3 import (
4 "os"
5 "path/filepath"
6 "strconv"
7 "strings"
8 "testing"
9 )
10
11 func TestParseMonFeatures(t *testing.T) {
12 t.Run("All features available", func(t *testing.T) {
13 parsedMonFeatures, err := parseMonFeatures(
14 strings.NewReader("mbm_total_bytes\nmbm_local_bytes\nllc_occupancy"))
15 if err != nil {
16 t.Errorf("Error while parsing mon features err = %v", err)
17 }
18
19 expectedMonFeatures := monFeatures{true, true, true}
20
21 if parsedMonFeatures != expectedMonFeatures {
22 t.Error("Cannot gather all features!")
23 }
24 })
25
26 t.Run("No features available", func(t *testing.T) {
27 parsedMonFeatures, err := parseMonFeatures(strings.NewReader(""))
28 if err != nil {
29 t.Errorf("Error while parsing mon features err = %v", err)
30 }
31
32 expectedMonFeatures := monFeatures{false, false, false}
33
34 if parsedMonFeatures != expectedMonFeatures {
35 t.Error("Expected no features available but there is any!")
36 }
37 })
38 }
39
40 func mockResctrlL3_MON(t *testing.T, NUMANodes []string, mocks map[string]uint64) string {
41 t.Helper()
42 testDir := t.TempDir()
43 monDataPath := filepath.Join(testDir, "mon_data")
44
45 for _, numa := range NUMANodes {
46 numaPath := filepath.Join(monDataPath, numa)
47 err := os.MkdirAll(numaPath, 0o700)
48 if err != nil {
49 t.Fatal(err)
50 }
51
52 for fileName, value := range mocks {
53 err := os.WriteFile(filepath.Join(numaPath, fileName), []byte(strconv.FormatUint(value, 10)), 0o644)
54 if err != nil {
55 t.Fatal(err)
56 }
57 }
58
59 }
60
61 return testDir
62 }
63
64 func TestGetMonitoringStats(t *testing.T) {
65 enabledMonFeatures.mbmTotalBytes = true
66 enabledMonFeatures.mbmLocalBytes = true
67 enabledMonFeatures.llcOccupancy = true
68 mbmEnabled = true
69 cmtEnabled = true
70
71 mocksNUMANodesToCreate := []string{"mon_l3_00", "mon_l3_01"}
72
73 mocksFilesToCreate := map[string]uint64{
74 "mbm_total_bytes": 9123911,
75 "mbm_local_bytes": 2361361,
76 "llc_occupancy": 123331,
77 }
78
79 mockedL3_MON := mockResctrlL3_MON(t, mocksNUMANodesToCreate, mocksFilesToCreate)
80
81 t.Run("Gather monitoring stats", func(t *testing.T) {
82 var stats Stats
83 err := getMonitoringStats(mockedL3_MON, &stats)
84 if err != nil {
85 t.Fatal(err)
86 }
87
88 expectedMBMStats := MBMNumaNodeStats{
89 MBMTotalBytes: mocksFilesToCreate["mbm_total_bytes"],
90 MBMLocalBytes: mocksFilesToCreate["mbm_local_bytes"],
91 }
92
93 expectedCMTStats := CMTNumaNodeStats{LLCOccupancy: mocksFilesToCreate["llc_occupancy"]}
94
95 for _, gotMBMStat := range *stats.MBMStats {
96 checkMBMStatCorrection(gotMBMStat, expectedMBMStats, t)
97 }
98
99 for _, gotCMTStat := range *stats.CMTStats {
100 checkCMTStatCorrection(gotCMTStat, expectedCMTStats, t)
101 }
102 })
103 }
104
View as plain text