mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
182 lines
4.0 KiB
Go
182 lines
4.0 KiB
Go
package metrics
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/uber-go/tally/v4"
|
|
"github.com/hanzoai/tasks/common/log"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
type MetricsSuite struct {
|
|
*require.Assertions
|
|
suite.Suite
|
|
controller *gomock.Controller
|
|
}
|
|
|
|
func TestMetricsSuite(t *testing.T) {
|
|
suite.Run(t, new(MetricsSuite))
|
|
}
|
|
|
|
func (s *MetricsSuite) SetupTest() {
|
|
s.Assertions = require.New(s.T())
|
|
s.controller = gomock.NewController(s.T())
|
|
}
|
|
|
|
func (s *MetricsSuite) TestStatsd() {
|
|
statsd := &StatsdConfig{
|
|
HostPort: "127.0.0.1:8125",
|
|
Prefix: "testStatsd",
|
|
}
|
|
|
|
config := new(Config)
|
|
config.Statsd = statsd
|
|
scope := NewScope(log.NewNoopLogger(), config)
|
|
s.NotNil(scope)
|
|
}
|
|
|
|
func (s *MetricsSuite) TestPrometheus() {
|
|
prom := &PrometheusConfig{
|
|
OnError: "panic",
|
|
TimerType: "histogram",
|
|
ListenAddress: "127.0.0.1:0",
|
|
}
|
|
config := new(Config)
|
|
config.Prometheus = prom
|
|
scope := NewScope(log.NewNoopLogger(), config)
|
|
s.NotNil(scope)
|
|
}
|
|
|
|
func (s *MetricsSuite) TestPrometheusWithSanitizeOptions() {
|
|
validChars := &ValidCharacters{
|
|
Ranges: []SanitizeRange{
|
|
{
|
|
StartRange: "a",
|
|
EndRange: "z",
|
|
},
|
|
{
|
|
StartRange: "A",
|
|
EndRange: "Z",
|
|
},
|
|
{
|
|
StartRange: "0",
|
|
EndRange: "9",
|
|
},
|
|
},
|
|
SafeCharacters: "-",
|
|
}
|
|
|
|
prom := &PrometheusConfig{
|
|
OnError: "panic",
|
|
TimerType: "histogram",
|
|
ListenAddress: "127.0.0.1:0",
|
|
SanitizeOptions: &SanitizeOptions{
|
|
NameCharacters: validChars,
|
|
KeyCharacters: validChars,
|
|
ValueCharacters: validChars,
|
|
ReplacementCharacter: "_",
|
|
},
|
|
}
|
|
config := new(Config)
|
|
config.Prometheus = prom
|
|
scope := NewScope(log.NewNoopLogger(), config)
|
|
s.NotNil(scope)
|
|
}
|
|
|
|
func (s *MetricsSuite) TestNoop() {
|
|
config := &Config{}
|
|
scope := NewScope(log.NewNoopLogger(), config)
|
|
s.Equal(tally.NoopScope, scope)
|
|
}
|
|
|
|
func (s *MetricsSuite) TestSetDefaultPerUnitHistogramBoundaries() {
|
|
type histogramTest struct {
|
|
input map[string][]float64
|
|
expectResult map[string][]float64
|
|
}
|
|
|
|
testCases := []histogramTest{
|
|
{
|
|
input: nil,
|
|
expectResult: map[string][]float64{
|
|
Dimensionless: defaultPerUnitHistogramBoundaries[Dimensionless],
|
|
Milliseconds: defaultPerUnitHistogramBoundaries[Milliseconds],
|
|
Seconds: {0.001, 0.002, 0.005, 0.010, 0.020, 0.050, 0.100, 0.200, 0.500, 1, 2, 5, 10, 20, 50, 100, 200, 500, 1_000},
|
|
Bytes: defaultPerUnitHistogramBoundaries[Bytes],
|
|
},
|
|
},
|
|
{
|
|
input: map[string][]float64{
|
|
UnitNameDimensionless: {1},
|
|
UnitNameMilliseconds: {10, 1000, 2000},
|
|
"notDefine": {1},
|
|
},
|
|
expectResult: map[string][]float64{
|
|
Dimensionless: {1},
|
|
Milliseconds: {10, 1000, 2000},
|
|
Seconds: {0.01, 1, 2},
|
|
Bytes: defaultPerUnitHistogramBoundaries[Bytes],
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
config := &ClientConfig{PerUnitHistogramBoundaries: test.input}
|
|
setDefaultPerUnitHistogramBoundaries(config)
|
|
s.Equal(test.expectResult, config.PerUnitHistogramBoundaries)
|
|
}
|
|
}
|
|
|
|
func TestMetricsHandlerFromConfig(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
logger := log.NewTestLogger()
|
|
|
|
for _, c := range []struct {
|
|
name string
|
|
cfg *Config
|
|
expectedType any
|
|
}{
|
|
{
|
|
name: "nil config",
|
|
cfg: nil,
|
|
expectedType: &noopMetricsHandler{},
|
|
},
|
|
{
|
|
name: "tally",
|
|
cfg: &Config{
|
|
Prometheus: &PrometheusConfig{
|
|
Framework: FrameworkTally,
|
|
ListenAddress: "localhost:0",
|
|
},
|
|
},
|
|
expectedType: &tallyMetricsHandler{},
|
|
},
|
|
{
|
|
name: "opentelemetry",
|
|
cfg: &Config{
|
|
Prometheus: &PrometheusConfig{
|
|
Framework: FrameworkOpentelemetry,
|
|
ListenAddress: "localhost:0",
|
|
},
|
|
},
|
|
expectedType: &otelMetricsHandler{},
|
|
},
|
|
} {
|
|
c := c
|
|
t.Run(c.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler, err := MetricsHandlerFromConfig(logger, c.cfg)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
handler.Stop(logger)
|
|
})
|
|
assert.IsType(t, c.expectedType, handler)
|
|
})
|
|
}
|
|
}
|