278
}
279
281
>
numSignalRequests := 20
282
>
283
>
shardID := rand.Int31()
284
>
namespaceID := primitives.NewUUID()
285
>
workflowID := shuffle.String(testHistoryExecutionWorkflowID)
286
>
runID := primitives.NewUUID()
287
>
288
>
var signalRequests []sqlplugin.SignalsRequestedSetsRow
289
>
var signalRequestIDs []string
290
>
for range numSignalRequests {
291
>
signalRequestID := shuffle.String(testHistoryExecutionSignalID)
292
>
signalRequest := s.newRandomExecutionSignalRequestRow(shardID, namespaceID, workflowID, runID, signalRequestID)
293
>
signalRequestIDs = append(signalRequestIDs, signalRequestID)
294
>
signalRequests = append(signalRequests, signalRequest)
295
>
}
296
>
result, err := s.store.ReplaceIntoSignalsRequestedSets(newExecutionContext(), signalRequests)
297
>
s.NoError(err)
298
>
rowsAffected, err := result.RowsAffected()
299
>
s.NoError(err)
300
>
s.Equal(numSignalRequests, int(rowsAffected))
301
>
302
>
deleteFilter := sqlplugin.SignalsRequestedSetsFilter{
303
>
ShardID: shardID,
304
>
NamespaceID: namespaceID,
305
>
WorkflowID: workflowID,
306
>
RunID: runID,
307
>
SignalIDs: signalRequestIDs,
308
>
}
309
>
result, err = s.store.DeleteFromSignalsRequestedSets(newExecutionContext(), deleteFilter)
310
>
s.NoError(err)
311
>
rowsAffected, err = result.RowsAffected()
312
>
s.NoError(err)
313
>
s.Equal(numSignalRequests, int(rowsAffected))
314
>
315
>
selectFilter := sqlplugin.SignalsRequestedSetsAllFilter{
316
>
ShardID: shardID,
317
>
NamespaceID: namespaceID,
318
>
WorkflowID: workflowID,
319
>
RunID: runID,
320
>
}
321
>
rows, err := s.store.SelectAllFromSignalsRequestedSets(newExecutionContext(), selectFilter)
322
>
s.NoError(err)
323
>
s.Equal([]sqlplugin.SignalsRequestedSetsRow(nil), rows)
324
}
325