mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
b16cb92b32
Profiling and tracing can now be enabled/disabled separately. Adds argument for trace file path. Support overriding profiling and tracing argument using environment variables. Update docs. Fixes #20576
74 lines
2.7 KiB
Go
74 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestProfilingDiagnostics(t *testing.T) {
|
|
tcs := []struct {
|
|
defaults *profilingDiagnostics
|
|
enabledEnv string
|
|
portEnv string
|
|
expected *profilingDiagnostics
|
|
}{
|
|
{defaults: newProfilingDiagnostics(false, 6060), enabledEnv: "", portEnv: "", expected: newProfilingDiagnostics(false, 6060)},
|
|
{defaults: newProfilingDiagnostics(true, 8080), enabledEnv: "", portEnv: "", expected: newProfilingDiagnostics(true, 8080)},
|
|
{defaults: newProfilingDiagnostics(false, 6060), enabledEnv: "false", portEnv: "8080", expected: newProfilingDiagnostics(false, 8080)},
|
|
{defaults: newProfilingDiagnostics(false, 6060), enabledEnv: "true", portEnv: "8080", expected: newProfilingDiagnostics(true, 8080)},
|
|
{defaults: newProfilingDiagnostics(false, 6060), enabledEnv: "true", portEnv: "", expected: newProfilingDiagnostics(true, 6060)},
|
|
}
|
|
|
|
for i, tc := range tcs {
|
|
t.Run(fmt.Sprintf("testcase %d", i), func(t *testing.T) {
|
|
os.Clearenv()
|
|
if tc.enabledEnv != "" {
|
|
err := os.Setenv(profilingEnabledEnvName, tc.enabledEnv)
|
|
assert.NoError(t, err)
|
|
}
|
|
if tc.portEnv != "" {
|
|
err := os.Setenv(profilingPortEnvName, tc.portEnv)
|
|
assert.NoError(t, err)
|
|
}
|
|
err := tc.defaults.overrideWithEnv()
|
|
assert.NoError(t, err)
|
|
assert.Exactly(t, tc.expected, tc.defaults)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestTracingDiagnostics(t *testing.T) {
|
|
tcs := []struct {
|
|
defaults *tracingDiagnostics
|
|
enabledEnv string
|
|
fileEnv string
|
|
expected *tracingDiagnostics
|
|
}{
|
|
{defaults: newTracingDiagnostics(false, "trace.out"), enabledEnv: "", fileEnv: "", expected: newTracingDiagnostics(false, "trace.out")},
|
|
{defaults: newTracingDiagnostics(true, "/tmp/trace.out"), enabledEnv: "", fileEnv: "", expected: newTracingDiagnostics(true, "/tmp/trace.out")},
|
|
{defaults: newTracingDiagnostics(false, "trace.out"), enabledEnv: "false", fileEnv: "/tmp/trace.out", expected: newTracingDiagnostics(false, "/tmp/trace.out")},
|
|
{defaults: newTracingDiagnostics(false, "trace.out"), enabledEnv: "true", fileEnv: "/tmp/trace.out", expected: newTracingDiagnostics(true, "/tmp/trace.out")},
|
|
{defaults: newTracingDiagnostics(false, "trace.out"), enabledEnv: "true", fileEnv: "", expected: newTracingDiagnostics(true, "trace.out")},
|
|
}
|
|
|
|
for i, tc := range tcs {
|
|
t.Run(fmt.Sprintf("testcase %d", i), func(t *testing.T) {
|
|
os.Clearenv()
|
|
if tc.enabledEnv != "" {
|
|
err := os.Setenv(tracingEnabledEnvName, tc.enabledEnv)
|
|
assert.NoError(t, err)
|
|
}
|
|
if tc.fileEnv != "" {
|
|
err := os.Setenv(tracingFileEnvName, tc.fileEnv)
|
|
assert.NoError(t, err)
|
|
}
|
|
err := tc.defaults.overrideWithEnv()
|
|
assert.NoError(t, err)
|
|
assert.Exactly(t, tc.expected, tc.defaults)
|
|
})
|
|
}
|
|
}
|