Function TestCPUQuotaPeriodUSec

cgroup2/manager_test.go:404–447

(t *testing.T)

}

func TestCPUQuotaPeriodUSec(t *testing.T) {
	checkCgroupMode(t)
	requireSystemdVersion(t, cpuQuotaPeriodUSecSupportedVersion)

	pid := os.Getpid()
	group := fmt.Sprintf("testing-cpu-period-%d.scope", pid)

	tests := []struct {
		name           string
		cpuMax         CPUMax
		expectedCPUMax string
		expectedPeriod uint64
	}{
		{
			name:           "default cpu.max",
			cpuMax:         "max 100000",
			expectedCPUMax: "max 100000",
			expectedPeriod: 100000,
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			c, err := NewSystemd("", group, pid, &Resources{
				CPU: &CPU{
					Max: test.cpuMax,
				},
			})
			require.NoError(t, err, "failed to init new cgroup systemd manager")

			conn, err := systemdDbus.NewWithContext(context.TODO())
			require.NoError(t, err, "failed to connect to systemd")
			defer conn.Close()

			unitName := systemdUnitFromPath(c.path)
			props, err := conn.GetAllPropertiesContext(context.TODO(), unitName)
			require.NoError(t, err, "failed to get unit properties")

			periodUSec, ok := props["CPUQuotaPeriodUSec"]
			require.True(t, ok, "CPUQuotaPeriodUSec property not found")
			require.Equal(t, test.expectedPeriod, periodUSec.(uint64), "CPUQuotaPeriodUSec value doesn't match expected period")
		})
	}
}

func BenchmarkStat(b *testing.B) {
	checkCgroupMode(b)