85
}
86
87
>
func (s *historyExecutionActivitySuite) TestReplaceSelect_Single() {
history_execution_activity.go
88
>
shardID := rand.Int31()
89
>
namespaceID := primitives.NewUUID()
90
>
workflowID := shuffle.String(testHistoryExecutionWorkflowID)
91
>
runID := primitives.NewUUID()
92
>
scheduledEventID := rand.Int63()
93
>
94
>
activity := s.newRandomExecutionActivityRow(shardID, namespaceID, workflowID, runID, scheduledEventID)
95
>
result, err := s.store.ReplaceIntoActivityInfoMaps(newExecutionContext(), []sqlplugin.ActivityInfoMapsRow{activity})
96
>
s.NoError(err)
97
>
rowsAffected, err := result.RowsAffected()
98
>
s.NoError(err)
99
>
s.Equal(1, int(rowsAffected))
100
>
101
>
selectFilter := sqlplugin.ActivityInfoMapsAllFilter{
102
>
ShardID: shardID,
103
>
NamespaceID: namespaceID,
104
>
WorkflowID: workflowID,
105
>
RunID: runID,
106
>
}
107
>
rows, err := s.store.SelectAllFromActivityInfoMaps(newExecutionContext(), selectFilter)
108
>
s.NoError(err)
109
>
rowMap := map[int64]sqlplugin.ActivityInfoMapsRow{}
110
>
for _, activity := range rows {
111
>
rowMap[activity.ScheduleID] = activity
112
>
}
113
>
s.Equal(map[int64]sqlplugin.ActivityInfoMapsRow{
114
>
activity.ScheduleID: activity,
115
>
}, rowMap)
116
}
117