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