mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-30 14:56:19 +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`.
88 lines
3.2 KiB
Go
88 lines
3.2 KiB
Go
package metricstest
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
enumspb "go.temporal.io/api/enums/v1"
|
|
"go.temporal.io/server/common/metrics"
|
|
"go.temporal.io/server/common/tqid"
|
|
)
|
|
|
|
const (
|
|
omitted = "__omitted__"
|
|
normal = "__normal__"
|
|
sticky = "__sticky__"
|
|
)
|
|
|
|
func TestPerTaskQueueScope(t *testing.T) {
|
|
ns := "my_ns"
|
|
tq := tqid.UnsafeTaskQueueFamily("ns_id", "my_tq").TaskQueue(enumspb.TASK_QUEUE_TYPE_WORKFLOW)
|
|
verifyTags(t,
|
|
metrics.GetPerTaskQueueScope(NewCaptureHandler(), ns, tq, false),
|
|
map[string]string{"namespace": ns, "taskqueue": omitted, "task_type": "Workflow"},
|
|
)
|
|
|
|
verifyTags(t,
|
|
metrics.GetPerTaskQueueScope(NewCaptureHandler(), ns, tq, true),
|
|
map[string]string{"namespace": ns, "taskqueue": tq.Name(), "task_type": "Workflow"},
|
|
)
|
|
}
|
|
|
|
func TestPerTaskQueuePartitionIDScope_Normal(t *testing.T) {
|
|
ns := "my_ns"
|
|
p := tqid.UnsafeTaskQueueFamily("ns_id", "my_tq").TaskQueue(enumspb.TASK_QUEUE_TYPE_WORKFLOW).NormalPartition(1)
|
|
verifyTags(t,
|
|
metrics.GetPerTaskQueuePartitionIDScope(NewCaptureHandler(), ns, p, true, false),
|
|
map[string]string{"namespace": ns, "taskqueue": p.TaskQueue().Name(), "task_type": "Workflow", "partition": normal},
|
|
)
|
|
verifyTags(t,
|
|
metrics.GetPerTaskQueuePartitionIDScope(NewCaptureHandler(), ns, p, true, true),
|
|
map[string]string{"namespace": ns, "taskqueue": p.TaskQueue().Name(), "task_type": "Workflow", "partition": "1"},
|
|
)
|
|
}
|
|
|
|
func TestPerTaskQueuePartitionIDScope_Sticky(t *testing.T) {
|
|
ns := "my_ns"
|
|
p := tqid.UnsafeTaskQueueFamily("ns_id", "my_tq").TaskQueue(enumspb.TASK_QUEUE_TYPE_WORKFLOW).StickyPartition("abc")
|
|
verifyTags(t,
|
|
metrics.GetPerTaskQueuePartitionIDScope(NewCaptureHandler(), ns, p, true, false),
|
|
map[string]string{"namespace": ns, "taskqueue": p.TaskQueue().Name(), "task_type": "Workflow", "partition": sticky},
|
|
)
|
|
verifyTags(t,
|
|
metrics.GetPerTaskQueuePartitionIDScope(NewCaptureHandler(), ns, p, true, true),
|
|
map[string]string{"namespace": ns, "taskqueue": p.TaskQueue().Name(), "task_type": "Workflow", "partition": sticky},
|
|
)
|
|
}
|
|
|
|
func TestPerTaskQueuePartitionTypeScope_Normal(t *testing.T) {
|
|
ns := "my_ns"
|
|
p := tqid.UnsafeTaskQueueFamily("ns_id", "my_tq").TaskQueue(enumspb.TASK_QUEUE_TYPE_WORKFLOW).NormalPartition(1)
|
|
verifyTags(t,
|
|
metrics.GetPerTaskQueuePartitionTypeScope(NewCaptureHandler(), ns, p, true),
|
|
map[string]string{"namespace": ns, "taskqueue": p.TaskQueue().Name(), "task_type": "Workflow", "partition": normal},
|
|
)
|
|
}
|
|
|
|
func TestPerTaskQueuePartitionTypeScope_Sticky(t *testing.T) {
|
|
ns := "my_ns"
|
|
p := tqid.UnsafeTaskQueueFamily("ns_id", "my_tq").TaskQueue(enumspb.TASK_QUEUE_TYPE_WORKFLOW).StickyPartition("abc")
|
|
verifyTags(t,
|
|
metrics.GetPerTaskQueuePartitionTypeScope(NewCaptureHandler(), ns, p, true),
|
|
map[string]string{"namespace": ns, "taskqueue": p.TaskQueue().Name(), "task_type": "Workflow", "partition": sticky},
|
|
)
|
|
}
|
|
|
|
func verifyTags(t *testing.T, handler metrics.Handler, expectedTags map[string]string) {
|
|
a := assert.New(t)
|
|
h, ok := handler.(*CaptureHandler)
|
|
a.True(ok)
|
|
capture := h.StartCapture()
|
|
h.Counter("MyMetric").Record(1)
|
|
snap := capture.Snapshot()
|
|
h.StopCapture(capture)
|
|
a.Equal(1, len(snap))
|
|
a.Equal(1, len(snap["MyMetric"]))
|
|
a.Equal(expectedTags, snap["MyMetric"][0].Tags)
|
|
}
|