[CLI] add new valid search attribute key cases for context header ind…

…exing in visibility
shijiesheng · Jun 24, 2024 · 4dfd72e · 4dfd72e
1 parent 83ebf7a
commit 4dfd72e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tools/cli/admin_cluster_commands.go b/tools/cli/admin_cluster_commands.go
@@ -36,7 +36,7 @@ import (
 
 // An indirection for the prompt function so that it can be mocked in the unit tests
 var promptFn = prompt
-var validSearchAttributeKey = regexp.MustCompile(`^[a-zA-Z][a-zA-Z_0-9]*$`)
+var validSearchAttributeKey = regexp.MustCompile(`^[a-zA-Z][a-zA-Z_.-0-9]*$`)
 
 // AdminAddSearchAttribute to whitelist search attribute
 func AdminAddSearchAttribute(c *cli.Context) {

diff --git a/tools/cli/admin_cluster_commands_test.go b/tools/cli/admin_cluster_commands_test.go
@@ -126,8 +126,8 @@ func TestValidSearchAttributeKey(t *testing.T) {
 	assert.NoError(t, validateSearchAttributeKey("cityId"))
 	assert.NoError(t, validateSearchAttributeKey("paymentProfileUUID"))
 	assert.NoError(t, validateSearchAttributeKey("job_type"))
+	assert.NoError(t, validateSearchAttributeKey("Header.ctx-tenancy"))
 
-	assert.Error(t, validateSearchAttributeKey("payments-biling-invoices-TransactionUUID"))
 	assert.Error(t, validateSearchAttributeKey("9lives"))
 	assert.Error(t, validateSearchAttributeKey("tax%"))
 }