mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
133 lines
3.3 KiB
Go
133 lines
3.3 KiB
Go
package telemetry_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.temporal.io/server/common/telemetry"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
var basicOTLPTraceOnlyConfig = `
|
|
exporters:
|
|
- kind:
|
|
signal: traces
|
|
model: otlp
|
|
protocol: grpc
|
|
spec:
|
|
headers:
|
|
a: b
|
|
c: d
|
|
timeout: 10s
|
|
retry:
|
|
enabled: true
|
|
initial_interval: 1s
|
|
max_interval: 1s
|
|
max_elapsed_time: 1s
|
|
connection:
|
|
block: false
|
|
insecure: true
|
|
endpoint: localhost:4317
|
|
`
|
|
|
|
var sharedConnOTLPConfig = `
|
|
otel:
|
|
connections:
|
|
- kind: grpc
|
|
metadata:
|
|
name: conn1
|
|
spec:
|
|
endpoint: localhost:4317
|
|
exporters:
|
|
- kind:
|
|
signal: traces
|
|
model: otlp
|
|
protocol: grpc
|
|
spec:
|
|
connection_name: conn1
|
|
- kind:
|
|
signal: metrics
|
|
model: otlp
|
|
protocol: grpc
|
|
spec:
|
|
connection_name: conn1
|
|
`
|
|
|
|
func TestEmptyConfig(t *testing.T) {
|
|
cfg := telemetry.ExportConfig{}
|
|
exporters, err := cfg.SpanExporters()
|
|
require.NoError(t, err)
|
|
require.Len(t, exporters, 0)
|
|
}
|
|
|
|
func TestExportersWithSharedConn(t *testing.T) {
|
|
root := struct{ Otel telemetry.PrivateExportConfig }{}
|
|
err := yaml.Unmarshal([]byte(sharedConnOTLPConfig), &root)
|
|
require.NoError(t, err)
|
|
cfg := &root.Otel
|
|
|
|
spanExporters, err := cfg.SpanExporters()
|
|
require.NoError(t, err)
|
|
require.Len(t, spanExporters, 1)
|
|
|
|
metricExporters, err := cfg.MetricExporters()
|
|
require.NoError(t, err)
|
|
require.Len(t, metricExporters, 1)
|
|
}
|
|
|
|
func TestSharedConn(t *testing.T) {
|
|
root := struct{ Otel telemetry.PrivateExportConfig }{}
|
|
err := yaml.Unmarshal([]byte(sharedConnOTLPConfig), &root)
|
|
require.NoError(t, err)
|
|
cfg := &root.Otel
|
|
require.Len(t, cfg.Connections, 1)
|
|
require.Len(t, cfg.Exporters, 2)
|
|
|
|
exp := cfg.Exporters[0]
|
|
require.Equal(t, exp.Kind.Signal, "traces")
|
|
require.Equal(t, exp.Kind.Model, "otlp")
|
|
require.Equal(t, exp.Kind.Protocol, "grpc")
|
|
require.NotNil(t, exp.Spec)
|
|
sspec, ok := exp.Spec.(*telemetry.OTLPGRPCSpanExporter)
|
|
require.True(t, ok)
|
|
require.Equal(t, "conn1", sspec.ConnectionName)
|
|
|
|
exp = cfg.Exporters[1]
|
|
require.Equal(t, exp.Kind.Signal, "metrics")
|
|
require.Equal(t, exp.Kind.Model, "otlp")
|
|
require.Equal(t, exp.Kind.Protocol, "grpc")
|
|
require.NotNil(t, exp.Spec)
|
|
mspec, ok := exp.Spec.(*telemetry.OTLPGRPCMetricExporter)
|
|
require.True(t, ok)
|
|
require.Equal(t, "conn1", mspec.ConnectionName)
|
|
}
|
|
|
|
func TestOTLPTraceGRPC(t *testing.T) {
|
|
cfg := telemetry.PrivateExportConfig{}
|
|
err := yaml.Unmarshal([]byte(basicOTLPTraceOnlyConfig), &cfg)
|
|
require.NoError(t, err)
|
|
require.Len(t, cfg.Connections, 0)
|
|
require.Len(t, cfg.Exporters, 1)
|
|
|
|
exp := cfg.Exporters[0]
|
|
require.Equal(t, exp.Kind.Signal, "traces")
|
|
require.Equal(t, exp.Kind.Model, "otlp")
|
|
require.Equal(t, exp.Kind.Protocol, "grpc")
|
|
require.NotNil(t, exp.Spec)
|
|
|
|
spec, ok := exp.Spec.(*telemetry.OTLPGRPCSpanExporter)
|
|
require.True(t, ok)
|
|
require.Equal(t, map[string]string{"a": "b", "c": "d"}, spec.Headers)
|
|
require.Equal(t, 10*time.Second, spec.Timeout)
|
|
require.True(t, spec.Retry.Enabled)
|
|
require.Equal(t, time.Second, spec.Retry.InitialInterval)
|
|
require.Equal(t, time.Second, spec.Retry.MaxInterval)
|
|
require.Equal(t, time.Second, spec.Retry.MaxElapsedTime)
|
|
|
|
conn := spec.Connection
|
|
require.True(t, conn.Insecure)
|
|
require.Equal(t, "localhost:4317", conn.Endpoint)
|
|
require.False(t, conn.Block)
|
|
}
|