mirror/runc
1
0
Fork 0
mirror of https://github.com/opencontainers/runc.git synced 2026-07-11 06:03:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

intelrdt: Add Cache Monitoring Technology stats

Browse Source 
Signed-off-by: Paweł Szulik <pawel.szulik@intel.com>
This commit is contained in:
Paweł Szulik
2020-04-15 21:18:41 +02:00
parent d1e4c7b803
commit 799d94818d
10 changed files with 335 additions and 170 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libcontainer/intelrdt/mbm_test.go
+26 -86
View File
@@ -3,71 +3,12 @@
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"))
if err != nil {
t.Errorf("Error while parsing mon features err = %v", err)
}
expectedMonFeatures := monFeatures{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}
if parsedMonFeatures != expectedMonFeatures {
t.Error("Expected no features available but there is any!")
}
})
}
func mockMBM(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)), 777)
if err != nil {
return "", err
}
}
}
return testDir, nil
}
func TestGetMbmStats(t *testing.T) {
func TestGetMBMNumaNodeStats(t *testing.T) {
mocksNUMANodesToCreate := []string{"mon_l3_00", "mon_l3_01"}
mocksFilesToCreate := map[string]uint64{
@@ -75,10 +16,10 @@ func TestGetMbmStats(t *testing.T) {
"mbm_local_bytes": 2361361,
}
mockedMBM, err := mockMBM(mocksNUMANodesToCreate, mocksFilesToCreate)
mockedL3_MON, err := mockResctrlL3_MON(mocksNUMANodesToCreate, mocksFilesToCreate)
defer func() {
err := os.RemoveAll(mockedMBM)
err := os.RemoveAll(mockedL3_MON)
if err != nil {
t.Fatal(err)
}
@@ -92,29 +33,13 @@ func TestGetMbmStats(t *testing.T) {
enabledMonFeatures.mbmTotalBytes = true
enabledMonFeatures.mbmLocalBytes = true
stats, err := getMBMStats(mockedMBM)
if err != nil {
t.Fatal(err)
}
if len(*stats) != len(mocksNUMANodesToCreate) {
t.Fatalf("Wrong number of stats slices from NUMA nodes. Expected: %v but got: %v",
len(mocksNUMANodesToCreate), len(*stats))
}
checkStatCorrection := func(got MBMNumaNodeStats, expected MBMNumaNodeStats, t *testing.T) {
if got.MBMTotalBytes != expected.MBMTotalBytes {
t.Fatalf("Wrong value of mbm_total_bytes. Expected: %v but got: %v",
expected.MBMTotalBytes,
got.MBMTotalBytes)
stats := make([]MBMNumaNodeStats, 0, len(mocksNUMANodesToCreate))
for _, numa := range mocksNUMANodesToCreate {
other, err := getMBMNumaNodeStats(filepath.Join(mockedL3_MON, "mon_data", numa))
if err != nil {
t.Fatal(err)
}
if got.MBMLocalBytes != expected.MBMLocalBytes {
t.Fatalf("Wrong value of mbm_local_bytes. Expected: %v but got: %v",
expected.MBMLocalBytes,
got.MBMLocalBytes)
}
stats = append(stats, *other)
}
expectedStats := MBMNumaNodeStats{
@@ -122,7 +47,22 @@ func TestGetMbmStats(t *testing.T) {
MBMLocalBytes: mocksFilesToCreate["mbm_local_bytes"],
}
checkStatCorrection((*stats)[0], expectedStats, t)
checkStatCorrection((*stats)[1], expectedStats, t)
checkMBMStatCorrection(stats[0], expectedStats, t)
checkMBMStatCorrection(stats[1], expectedStats, t)
})
}
func checkMBMStatCorrection(got MBMNumaNodeStats, expected MBMNumaNodeStats, t *testing.T) {
if got.MBMTotalBytes != expected.MBMTotalBytes {
t.Fatalf("Wrong value of mbm_total_bytes. Expected: %v but got: %v",
expected.MBMTotalBytes,
got.MBMTotalBytes)
}
if got.MBMLocalBytes != expected.MBMLocalBytes {
t.Fatalf("Wrong value of mbm_local_bytes. Expected: %v but got: %v",
expected.MBMLocalBytes,
got.MBMLocalBytes)
}
}