consul/lib/hoststats/collector_test.go
Melissa Kam c95380e3da
[CC-7434] Skip collecting data directory metrics in dev mode (#20521)
* Skip collecting data directory stats in dev mode

In dev mode, the data directory is not set, so this metrics collection would
always fail and logs errors.

* Log collection errors at DEBUG level

There isn't much action a user can take to fix these errors, so
logging them as DEBUG rather than ERROR.
2024-02-07 16:59:06 -06:00

49 lines
1.0 KiB
Go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1
package hoststats
import (
"testing"
"github.com/hashicorp/go-hclog"
"github.com/stretchr/testify/require"
"github.com/hashicorp/consul/sdk/testutil"
)
// TestCollector_collect validates that metrics for host resource usage
// are collected as expected.
func TestCollector_collect(t *testing.T) {
testcases := map[string]struct {
skipDataDir bool
}{
"WithDataDirectory": {},
"NoDataDirectory": {
skipDataDir: true,
},
}
for name, tc := range testcases {
t.Run(name, func(t *testing.T) {
dataDir := ""
if !tc.skipDataDir {
dataDir = testutil.TempDir(t, "consul-config")
}
collector := initCollector(hclog.NewNullLogger(), dataDir)
collector.collect()
hs := collector.hostStats
require.NotNil(t, hs)
require.Greater(t, hs.Uptime, uint64(0))
require.NotNil(t, hs.Memory)
require.NotNil(t, hs.CPU)
if !tc.skipDataDir {
require.NotNil(t, hs.DataDirStats)
} else {
require.Nil(t, hs.DataDirStats)
}
})
}
}