86
}
87
89
>
shardID := rand.Int31()
90
>
namespaceID := primitives.NewUUID()
91
>
workflowID := shuffle.String(testHistoryExecutionWorkflowID)
92
>
runID := primitives.NewUUID()
93
>
timerID := shuffle.String(testHistoryExecutionTimerID)
94
>
95
>
timer := s.newRandomExecutionTimerRow(shardID, namespaceID, workflowID, runID, timerID)
96
>
result, err := s.store.ReplaceIntoTimerInfoMaps(newExecutionContext(), []sqlplugin.TimerInfoMapsRow{timer})
97
>
s.NoError(err)
98
>
rowsAffected, err := result.RowsAffected()
99
>
s.NoError(err)
100
>
s.Equal(1, int(rowsAffected))
101
>
102
>
selectFilter := sqlplugin.TimerInfoMapsAllFilter{
103
>
ShardID: shardID,
104
>
NamespaceID: namespaceID,
105
>
WorkflowID: workflowID,
106
>
RunID: runID,
107
>
}
108
>
rows, err := s.store.SelectAllFromTimerInfoMaps(newExecutionContext(), selectFilter)
109
>
s.NoError(err)
110
>
rowMap := map[string]sqlplugin.TimerInfoMapsRow{}
111
>
for _, timer := range rows {
112
>
rowMap[timer.TimerID] = timer
113
>
}
114
>
s.Equal(map[string]sqlplugin.TimerInfoMapsRow{
115
>
timer.TimerID: timer,
116
>
}, rowMap)
117
}
118