81
}
82
83
>
func (s *historyExecutionSignalRequestSuite) TestReplaceSelect_Single() {
history_execution_signal_requested.go
84
>
shardID := rand.Int31()
85
>
namespaceID := primitives.NewUUID()
86
>
workflowID := shuffle.String(testHistoryExecutionWorkflowID)
87
>
runID := primitives.NewUUID()
88
>
signalID := shuffle.String(testHistoryExecutionSignalID)
89
>
90
>
signalRequest := s.newRandomExecutionSignalRequestRow(shardID, namespaceID, workflowID, runID, signalID)
91
>
result, err := s.store.ReplaceIntoSignalsRequestedSets(newExecutionContext(), []sqlplugin.SignalsRequestedSetsRow{signalRequest})
92
>
s.NoError(err)
93
>
rowsAffected, err := result.RowsAffected()
94
>
s.NoError(err)
95
>
s.Equal(1, int(rowsAffected))
96
>
97
>
selectFilter := sqlplugin.SignalsRequestedSetsAllFilter{
98
>
ShardID: shardID,
99
>
NamespaceID: namespaceID,
100
>
WorkflowID: workflowID,
101
>
RunID: runID,
102
>
}
103
>
rows, err := s.store.SelectAllFromSignalsRequestedSets(newExecutionContext(), selectFilter)
104
>
s.NoError(err)
105
>
rowMap := map[string]sqlplugin.SignalsRequestedSetsRow{}
106
>
for _, signalRequest := range rows {
107
>
rowMap[signalRequest.SignalID] = signalRequest
108
>
}
109
>
s.Equal(map[string]sqlplugin.SignalsRequestedSetsRow{
110
>
signalRequest.SignalID: signalRequest,
111
>
}, rowMap)
112
}
113