112
}
113
114
>
func (s *historyExecutionSignalRequestSuite) TestReplaceSelect_Multiple() {
history_execution_signal_requested.go
115
>
numSignalRequests := 20
116
>
117
>
shardID := rand.Int31()
118
>
namespaceID := primitives.NewUUID()
119
>
workflowID := shuffle.String(testHistoryExecutionWorkflowID)
120
>
runID := primitives.NewUUID()
121
>
122
>
var signalRequests []sqlplugin.SignalsRequestedSetsRow
123
>
for range numSignalRequests {
124
>
signalRequest := s.newRandomExecutionSignalRequestRow(shardID, namespaceID, workflowID, runID, shuffle.String(testHistoryExecutionSignalID))
125
>
signalRequests = append(signalRequests, signalRequest)
126
>
}
127
>
result, err := s.store.ReplaceIntoSignalsRequestedSets(newExecutionContext(), signalRequests)
128
>
s.NoError(err)
129
>
rowsAffected, err := result.RowsAffected()
130
>
s.NoError(err)
131
>
s.Equal(numSignalRequests, int(rowsAffected))
132
>
133
>
selectFilter := sqlplugin.SignalsRequestedSetsAllFilter{
134
>
ShardID: shardID,
135
>
NamespaceID: namespaceID,
136
>
WorkflowID: workflowID,
137
>
RunID: runID,
138
>
}
139
>
rows, err := s.store.SelectAllFromSignalsRequestedSets(newExecutionContext(), selectFilter)
140
>
s.NoError(err)
141
>
signalRequestMap := map[string]sqlplugin.SignalsRequestedSetsRow{}
142
>
for _, signalRequest := range signalRequests {
143
>
signalRequestMap[signalRequest.SignalID] = signalRequest
144
>
}
145
>
rowMap := map[string]sqlplugin.SignalsRequestedSetsRow{}
146
>
for _, signalRequest := range rows {
147
>
rowMap[signalRequest.SignalID] = signalRequest
148
>
}
149
>
s.Equal(signalRequestMap, rowMap)
150
}
151