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