116
}
117
118
>
func (s *historyExecutionActivitySuite) TestReplaceSelect_Multiple() {
history_execution_activity.go
119
>
numActivities := 20
120
>
121
>
shardID := rand.Int31()
122
>
namespaceID := primitives.NewUUID()
123
>
workflowID := shuffle.String(testHistoryExecutionWorkflowID)
124
>
runID := primitives.NewUUID()
125
>
126
>
var activities []sqlplugin.ActivityInfoMapsRow
127
>
for range numActivities {
128
>
activity := s.newRandomExecutionActivityRow(shardID, namespaceID, workflowID, runID, rand.Int63())
129
>
activities = append(activities, activity)
130
>
}
131
>
result, err := s.store.ReplaceIntoActivityInfoMaps(newExecutionContext(), activities)
132
>
s.NoError(err)
133
>
rowsAffected, err := result.RowsAffected()
134
>
s.NoError(err)
135
>
s.Equal(numActivities, int(rowsAffected))
136
>
137
>
selectFilter := sqlplugin.ActivityInfoMapsAllFilter{
138
>
ShardID: shardID,
139
>
NamespaceID: namespaceID,
140
>
WorkflowID: workflowID,
141
>
RunID: runID,
142
>
}
143
>
rows, err := s.store.SelectAllFromActivityInfoMaps(newExecutionContext(), selectFilter)
144
>
s.NoError(err)
145
>
activityMap := map[int64]sqlplugin.ActivityInfoMapsRow{}
146
>
for _, activity := range activities {
147
>
activityMap[activity.ScheduleID] = activity
148
>
}
149
>
rowMap := map[int64]sqlplugin.ActivityInfoMapsRow{}
150
>
for _, activity := range rows {
151
>
rowMap[activity.ScheduleID] = activity
152
>
}
153
>
s.Equal(activityMap, rowMap)
154
}
155