116
}
117
118
>
func (s *historyExecutionRequestCancelSuite) TestReplaceSelect_Multiple() {
history_execution_request_cancel.go
119
>
numRequestCancels := 20
120
>
121
>
shardID := rand.Int31()
122
>
namespaceID := primitives.NewUUID()
123
>
workflowID := shuffle.String(testHistoryExecutionWorkflowID)
124
>
runID := primitives.NewUUID()
125
>
126
>
var requestCancels []sqlplugin.RequestCancelInfoMapsRow
127
>
for range numRequestCancels {
128
>
requestCancel := s.newRandomExecutionRequestCancelRow(shardID, namespaceID, workflowID, runID, rand.Int63())
129
>
requestCancels = append(requestCancels, requestCancel)
130
>
}
131
>
result, err := s.store.ReplaceIntoRequestCancelInfoMaps(newExecutionContext(), requestCancels)
132
>
s.NoError(err)
133
>
rowsAffected, err := result.RowsAffected()
134
>
s.NoError(err)
135
>
s.Equal(numRequestCancels, int(rowsAffected))
136
>
137
>
selectFilter := sqlplugin.RequestCancelInfoMapsAllFilter{
138
>
ShardID: shardID,
139
>
NamespaceID: namespaceID,
140
>
WorkflowID: workflowID,
141
>
RunID: runID,
142
>
}
143
>
rows, err := s.store.SelectAllFromRequestCancelInfoMaps(newExecutionContext(), selectFilter)
144
>
s.NoError(err)
145
>
requestCancelMap := map[int64]sqlplugin.RequestCancelInfoMapsRow{}
146
>
for _, requestCancel := range requestCancels {
147
>
requestCancelMap[requestCancel.InitiatedID] = requestCancel
148
>
}
149
>
rowMap := map[int64]sqlplugin.RequestCancelInfoMapsRow{}
150
>
for _, requestCancel := range rows {
151
>
rowMap[requestCancel.InitiatedID] = requestCancel
152
>
}
153
>
s.Equal(requestCancelMap, rowMap)
154
}
155