Method replaceAssignmentRule

tests/versioning_test.go:4453–4480

replaceAssignmentRule replaces a new versioning assignment rule to the task queue user data. It checks the response and returns the conflict token.

(
	env *testcore.TestEnv,
	tq, newBuildID string,
	idx int32, conflictToken []byte, expectSuccess bool)

// replaceAssignmentRule replaces a new versioning assignment rule to the task queue user data.
// It checks the response and returns the conflict token.
func (s *VersioningIntegSuite) replaceAssignmentRule(
	env *testcore.TestEnv,
	tq, newBuildID string,
	idx int32, conflictToken []byte, expectSuccess bool) []byte {
	res, err := env.FrontendClient().UpdateWorkerVersioningRules(s.Context(), &workflowservice.UpdateWorkerVersioningRulesRequest{
		Namespace:     env.Namespace().String(),
		TaskQueue:     tq,
		ConflictToken: conflictToken,
		Operation: &workflowservice.UpdateWorkerVersioningRulesRequest_ReplaceAssignmentRule{
			ReplaceAssignmentRule: &workflowservice.UpdateWorkerVersioningRulesRequest_ReplaceBuildIdAssignmentRule{
				RuleIndex: idx,
				Rule: &taskqueuepb.BuildIdAssignmentRule{
					TargetBuildId: newBuildID,
				},
			},
		},
	})
	if expectSuccess {
		s.NoError(err)
		s.NotNil(res)
		s.Equal(newBuildID, res.GetAssignmentRules()[idx].GetRule().GetTargetBuildId())
		return res.GetConflictToken()
	} else {
		s.Error(err)
		s.Nil(res)
		return nil
	}
}

// deleteAssignmentRule deletes the versioning assignment rule at a given index.
// It checks the response and returns the conflict token.