cgroup_exporter/collector/cgroupv2_test.go

// Copyright 2020 Trey Dockendorf
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package collector

import (
	"fmt"
	"os"
	"path/filepath"
	"testing"

	"github.com/go-kit/log"
)

func TestGetStatv2(t *testing.T) {
	w := log.NewSyncWriter(os.Stderr)
	logger := log.NewLogfmtLogger(w)
	_, err := getStatv2("swapcached", "/dne", logger)
	if err == nil {
		t.Errorf("Expected error with /dne but none given")
	}
	path := filepath.Join(*CgroupRoot, "system.slice")
	_, err = getStatv2("swapcached", path, logger)
	if err == nil {
		t.Errorf("Expected error with /dne but none given")
	}
	path = filepath.Join(*CgroupRoot, "user.slice/user-20821.slice/memory.max")
	_, err = getStatv2("swapcached", path, logger)
	if err == nil {
		t.Errorf("Expected error with single value file but none given")
	}
	path = filepath.Join(*CgroupRoot, "stat.invalid")
	_, err = getStatv2("nan", path, logger)
	if err == nil {
		t.Errorf("Expected error with stat.invalid but none given")
	}
	path = filepath.Join(*CgroupRoot, "user.slice/user-20821.slice/memory.stat")
	_, err = getStatv2("dne", path, logger)
	if err == nil {
		t.Errorf("Expected error when stat key missing but none given")
	}
	stat, err := getStatv2("swapcached", path, logger)
	if err != nil {
		t.Errorf("Unexpected error: %s", err)
	}
	if stat != 0 {
		t.Errorf("Unexpectd value: %v", stat)
	}
}

func TestCollectv2Error(t *testing.T) {
	w := log.NewSyncWriter(os.Stderr)
	logger := log.NewLogfmtLogger(w)
	exporter := NewExporter([]string{"/dne"}, logger, true)
	metrics, err := exporter.collectv2()
	if err != nil {
		t.Errorf("Unexpected error: %s", err.Error())
		return
	}
	if val := len(metrics); val != 1 {
		t.Errorf("Unexpected number of metrics, got %d expected 1", val)
		return
	}
	if val := metrics[0].err; val != true {
		t.Errorf("Unexpected value for err, got %v", val)
	}
}

func TestCollectv2UserSlice(t *testing.T) {
	varFalse := false
	collectProc = &varFalse
	pidGroupPath = func(pid int) (string, error) {
		if pid == 67998 {
			return "/user.slice/user-20821.slice/session-157.scope", nil
		}
		return "", fmt.Errorf("Could not find cgroup path for %d", pid)
	}
	w := log.NewSyncWriter(os.Stderr)
	logger := log.NewLogfmtLogger(w)
	exporter := NewExporter([]string{"/user.slice"}, logger, true)
	metrics, err := exporter.collectv2()
	if err != nil {
		t.Errorf("Unexpected error: %s", err.Error())
		return
	}
	if val := len(metrics); val != 1 {
		t.Errorf("Unexpected number of metrics, got %d expected 1", val)
		return
	}
	if val := metrics[0].name; val != "/user.slice/user-20821.slice" {
		t.Errorf("Unexpected value for name, got %v", val)
	}
	if val := metrics[0].cpuUser; val != 15.270449 {
		t.Errorf("Unexpected value for cpuUser, got %v", val)
	}
	if val := metrics[0].cpuSystem; val != 2.705424 {
		t.Errorf("Unexpected value for cpuSystem, got %v", val)
	}
	if val := metrics[0].cpuTotal; val != 17.975873 {
		t.Errorf("Unexpected value for cpuTotal, got %v", val)
	}
	if val := metrics[0].cpus; val != 0 {
		t.Errorf("Unexpected value for cpus, got %v", val)
	}
	if val := metrics[0].memoryRSS; val != 22048768 {
		t.Errorf("Unexpected value for memoryRSS, got %v", val)
	}
	if val := metrics[0].memoryUsed; val != 27115520 {
		t.Errorf("Unexpected value for memoryUsed, got %v", val)
	}
	if val := metrics[0].memoryTotal; val != 2147483648 {
		t.Errorf("Unexpected value for memoryTotal, got %v", val)
	}
	if val := metrics[0].memoryFailCount; val != 0 {
		t.Errorf("Unexpected value for memoryFailCount, got %v", val)
	}
	if val := metrics[0].memswUsed; val != 0 {
		t.Errorf("Unexpected value for swapUsed, got %v", val)
	}
	if val := metrics[0].memswTotal; val != 1.8446744073709552e+19 {
		t.Errorf("Unexpected value for swapTotal, got %v", val)
	}
	if val := metrics[0].uid; val != "20821" {
		t.Errorf("Unexpected value for uid, got %v", val)
	}
}

func TestCollectv2SLURM(t *testing.T) {
	varTrue := true
	collectProc = &varTrue
	varLen := 100
	collectProcMaxExec = &varLen
	pidGroupPath = func(pid int) (string, error) {
		if pid == 49276 {
			return "/system.slice/slurmstepd.scope/job_4/step_0/user/task_0", nil
		}
		if pid == 43310 {
			return "/system.slice/slurmstepd.scope/system", nil
		}
		return "", fmt.Errorf("Could not find cgroup path for %d", pid)
	}
	w := log.NewSyncWriter(os.Stderr)
	logger := log.NewLogfmtLogger(w)
	exporter := NewExporter([]string{"/slurm"}, logger, true)
	metrics, err := exporter.collectv2()
	if err != nil {
		t.Errorf("Unexpected error: %s", err.Error())
		return
	}
	if val := len(metrics); val != 1 {
		t.Errorf("Unexpected number of metrics, got %d expected 1", val)
		return
	}
	var m CgroupMetric
	for _, metric := range metrics {
		if metric.jobid == "4" {
			m = metric
		}
	}
	if m.jobid == "" {
		t.Errorf("Metrics with jobid=4 not found")
		return
	}
	if val := m.name; val != "/system.slice/slurmstepd.scope/job_4" {
		t.Errorf("Unexpected value for name, got %v", val)
	}
	if val := m.cpuUser; val != 0.049043 {
		t.Errorf("Unexpected value for cpuUser, got %v", val)
	}
	if val := m.cpuSystem; val != 0.077642 {
		t.Errorf("Unexpected value for cpuSystem, got %v", val)
	}
	if val := m.cpuTotal; val != 0.126686 {
		t.Errorf("Unexpected value for cpuTotal, got %v", val)
	}
	if val := m.cpus; val != 1 {
		t.Errorf("Unexpected value for cpus, got %v", val)
	}
	if val := m.memoryRSS; val != 2260992 {
		t.Errorf("Unexpected value for memoryRSS, got %v", val)
	}
	if val := m.memoryUsed; val != 5660672 {
		t.Errorf("Unexpected value for memoryUsed, got %v", val)
	}
	if val := m.memoryTotal; val != 1835008000 {
		t.Errorf("Unexpected value for memoryTotal, got %v", val)
	}
	if val := m.memoryFailCount; val != 0 {
		t.Errorf("Unexpected value for memoryFailCount, got %v", val)
	}
	if val := m.memswUsed; val != 0 {
		t.Errorf("Unexpected value for swapUsed, got %v", val)
	}
	if val := m.memswTotal; val != 1835008000 {
		t.Errorf("Unexpected value for swapTotal, got %v", val)
	}
	if val := m.uid; val != "20821" {
		t.Errorf("Unexpected value for uid, got %v", val)
	}
	if val := m.jobid; val != "4" {
		t.Errorf("Unexpected value for jobid, got %v", val)
	}
	if val, ok := m.processExec["/usr/bin/bash"]; !ok {
		t.Errorf("processExec does not contain /bin/bash")
	} else {
		if val != 1 {
			t.Errorf("Unexpected 1 values for processExec /usr/bin/bash, got %v", val)
		}
	}
}