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).
78 lines
2.4 KiB
Go
78 lines
2.4 KiB
Go
package tests
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
enumspb "go.temporal.io/api/enums/v1"
|
|
deploymentspb "github.com/hanzoai/tasks/api/deployment/v1"
|
|
"github.com/hanzoai/tasks/api/matchingservice/v1"
|
|
"github.com/hanzoai/tasks/common/namespace"
|
|
"github.com/hanzoai/tasks/common/primitives/timestamp"
|
|
"github.com/hanzoai/tasks/common/testing/testvars"
|
|
"github.com/hanzoai/tasks/tests/testcore"
|
|
)
|
|
|
|
// runWithMatchingBehaviors runs a test with all combinations of matching behaviors.
|
|
func runWithMatchingBehaviors(
|
|
t *testing.T,
|
|
baseOpts []testcore.TestOption,
|
|
subtest func(s *testcore.TestEnv, behavior testcore.MatchingBehavior),
|
|
) {
|
|
for _, behavior := range testcore.AllMatchingBehaviors() {
|
|
t.Run(behavior.Name(), func(t *testing.T) {
|
|
opts := append([]testcore.TestOption{}, baseOpts...)
|
|
opts = append(opts, behavior.Options()...)
|
|
|
|
env := testcore.NewEnv(t, opts...)
|
|
behavior.InjectHooks(env)
|
|
|
|
subtest(env, behavior)
|
|
})
|
|
}
|
|
}
|
|
|
|
// syncDeploymentVersionToTaskQueues sends a SyncDeploymentUserData request to the matching service
|
|
// to register a deployment version as current for the specified task queue types, then waits for
|
|
// the data to propagate to all partitions using CheckTaskQueueUserDataPropagation.
|
|
func syncDeploymentVersionToTaskQueues(
|
|
t testing.TB,
|
|
matchingClient matchingservice.MatchingServiceClient,
|
|
namespaceID namespace.ID,
|
|
tv *testvars.TestVars,
|
|
tqTypes ...enumspb.TaskQueueType,
|
|
) {
|
|
t.Helper()
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancel()
|
|
|
|
now := timestamp.TimePtr(time.Now())
|
|
resp, err := matchingClient.SyncDeploymentUserData(
|
|
ctx, &matchingservice.SyncDeploymentUserDataRequest{
|
|
NamespaceId: namespaceID.String(),
|
|
TaskQueue: tv.TaskQueue().GetName(),
|
|
TaskQueueTypes: tqTypes,
|
|
Operation: &matchingservice.SyncDeploymentUserDataRequest_UpdateVersionData{
|
|
UpdateVersionData: &deploymentspb.DeploymentVersionData{
|
|
Version: tv.DeploymentVersion(),
|
|
RoutingUpdateTime: now,
|
|
CurrentSinceTime: now,
|
|
},
|
|
},
|
|
},
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
// Wait for the data to propagate to all partitions.
|
|
_, err = matchingClient.CheckTaskQueueUserDataPropagation(
|
|
ctx, &matchingservice.CheckTaskQueueUserDataPropagationRequest{
|
|
NamespaceId: namespaceID.String(),
|
|
TaskQueue: tv.TaskQueue().GetName(),
|
|
Version: resp.GetVersion(),
|
|
},
|
|
)
|
|
require.NoError(t, err)
|
|
}
|