Fix unit tests

treydock · May 18, 2024 · 7defe90 · 7defe90
1 parent c42eeba
commit 7defe90
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cgroup_exporter.go b/cgroup_exporter.go
@@ -24,6 +24,7 @@ import (
  "github.com/go-kit/log"
  "github.com/go-kit/log/level"
  "github.com/prometheus/client_golang/prometheus"
+ versionCollector "github.com/prometheus/client_golang/prometheus/collectors/version"
  "github.com/prometheus/client_golang/prometheus/promhttp"
  "github.com/prometheus/common/promlog"
  "github.com/prometheus/common/promlog/flag"
@@ -49,7 +50,7 @@ func metricsHandler(logger log.Logger) http.HandlerFunc {
  }
  cgroupCollector := collector.NewCgroupCollector(cgroupV2, paths, logger)
  registry.MustRegister(cgroupCollector)
- registry.MustRegister(version.NewCollector(fmt.Sprintf("%s_exporter", collector.Namespace)))
+ registry.MustRegister(versionCollector.NewCollector(fmt.Sprintf("%s_exporter", collector.Namespace)))
 
  gatherers := prometheus.Gatherers{registry}
  if !*disableExporterMetrics {

diff --git a/collector/cgroupv2.go b/collector/cgroupv2.go
@@ -87,7 +87,7 @@ func getInfov2(name string, pids []int, metric *CgroupMetric, logger log.Logger)
  }
  // effective UID
  uid := procStat.UIDs[1]
- metric.uid = uid
+ metric.uid = strconv.FormatUint(uid, 10)
  user, err := user.LookupId(metric.uid)
  if err != nil {
  level.Error(logger).Log("msg", "Error looking up slurm uid", "uid", metric.uid, "err", err)