IWF-472: Adding required parameter for panic utils

indeedeng · Jan 24, 2025 · 9640c0c · 9640c0c
1 parent 975a925
commit 9640c0c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/integ/set_search_attributes_test.go b/integ/set_search_attributes_test.go
@@ -94,7 +94,7 @@ func TestSetSearchAttributes(t *testing.T) {
 				ValueType: ptr.Any(iwfidl.KEYWORD_ARRAY),
 			},
 		}}).Execute()
-	panicAtHttpError(err, httpRespGet)
+	panicAtHttpError(err, httpRespGet, t)
 
 	assertions.ElementsMatch(signalVals, searchResult.SearchAttributes)
 }
diff --git a/integ/wait_until_search_attributes_test.go b/integ/wait_until_search_attributes_test.go
@@ -80,14 +80,14 @@ func doTestWaitUntilSearchAttributes(
 
 	switch mode := config.GetExecutingStateIdMode(); mode {
 	case iwfidl.ENABLED_FOR_ALL:
-		assertSearch(fmt.Sprintf("WorkflowId='%v'", wfId), 1, apiClient, assertions)
-		assertSearch(fmt.Sprintf("WorkflowId='%v' AND %v='%v'", wfId, wait_until_search_attributes.TestSearchAttributeExecutingStateIdsKey, wait_until_search_attributes.State2), 1, apiClient, assertions)
+		assertSearch(t, fmt.Sprintf("WorkflowId='%v'", wfId), 1, apiClient, assertions)
+		assertSearch(t, fmt.Sprintf("WorkflowId='%v' AND %v='%v'", wfId, wait_until_search_attributes.TestSearchAttributeExecutingStateIdsKey, wait_until_search_attributes.State2), 1, apiClient, assertions)
 	case iwfidl.ENABLED_FOR_STATES_WITH_WAIT_UNTIL:
-		assertSearch(fmt.Sprintf("WorkflowId='%v'", wfId), 1, apiClient, assertions)
-		assertSearch(fmt.Sprintf("WorkflowId='%v' AND %v='%v'", wfId, wait_until_search_attributes.TestSearchAttributeExecutingStateIdsKey, wait_until_search_attributes.State2), 0, apiClient, assertions)
+		assertSearch(t, fmt.Sprintf("WorkflowId='%v'", wfId), 1, apiClient, assertions)
+		assertSearch(t, fmt.Sprintf("WorkflowId='%v' AND %v='%v'", wfId, wait_until_search_attributes.TestSearchAttributeExecutingStateIdsKey, wait_until_search_attributes.State2), 0, apiClient, assertions)
 	case iwfidl.DISABLED:
-		assertSearch(fmt.Sprintf("WorkflowId='%v'", wfId), 1, apiClient, assertions)
-		assertSearch(fmt.Sprintf("WorkflowId='%v' AND %v='%v'", wfId, wait_until_search_attributes.TestSearchAttributeExecutingStateIdsKey, wait_until_search_attributes.State2), 0, apiClient, assertions)
+		assertSearch(t, fmt.Sprintf("WorkflowId='%v'", wfId), 1, apiClient, assertions)
+		assertSearch(t, fmt.Sprintf("WorkflowId='%v' AND %v='%v'", wfId, wait_until_search_attributes.TestSearchAttributeExecutingStateIdsKey, wait_until_search_attributes.State2), 0, apiClient, assertions)
 	}
 
 	reqWait := apiClient.DefaultApi.ApiV1WorkflowGetWithWaitPost(context.Background())