282
}
283
285
>
numRequestCancels := 20
286
>
287
>
shardID := rand.Int31()
288
>
namespaceID := primitives.NewUUID()
289
>
workflowID := shuffle.String(testHistoryExecutionWorkflowID)
290
>
runID := primitives.NewUUID()
291
>
292
>
var requestCancels []sqlplugin.RequestCancelInfoMapsRow
293
>
var requestCancelInitiatedIDs []int64
294
>
for range numRequestCancels {
295
>
requestCancelInitiatedID := rand.Int63()
296
>
requestCancel := s.newRandomExecutionRequestCancelRow(shardID, namespaceID, workflowID, runID, requestCancelInitiatedID)
297
>
requestCancelInitiatedIDs = append(requestCancelInitiatedIDs, requestCancelInitiatedID)
298
>
requestCancels = append(requestCancels, requestCancel)
299
>
}
300
>
result, err := s.store.ReplaceIntoRequestCancelInfoMaps(newExecutionContext(), requestCancels)
301
>
s.NoError(err)
302
>
rowsAffected, err := result.RowsAffected()
303
>
s.NoError(err)
304
>
s.Equal(numRequestCancels, int(rowsAffected))
305
>
306
>
deleteFilter := sqlplugin.RequestCancelInfoMapsFilter{
307
>
ShardID: shardID,
308
>
NamespaceID: namespaceID,
309
>
WorkflowID: workflowID,
310
>
RunID: runID,
311
>
InitiatedIDs: requestCancelInitiatedIDs,
312
>
}
313
>
result, err = s.store.DeleteFromRequestCancelInfoMaps(newExecutionContext(), deleteFilter)
314
>
s.NoError(err)
315
>
rowsAffected, err = result.RowsAffected()
316
>
s.NoError(err)
317
>
s.Equal(numRequestCancels, int(rowsAffected))
318
>
319
>
selectFilter := sqlplugin.RequestCancelInfoMapsAllFilter{
320
>
ShardID: shardID,
321
>
NamespaceID: namespaceID,
322
>
WorkflowID: workflowID,
323
>
RunID: runID,
324
>
}
325
>
rows, err := s.store.SelectAllFromRequestCancelInfoMaps(newExecutionContext(), selectFilter)
326
>
s.NoError(err)
327
>
s.Equal([]sqlplugin.RequestCancelInfoMapsRow(nil), rows)
328
}
329