196
}
197
198
>
func (c *MockMutableContext) SetRequestLinks(component Component, requestID string, links []*commonpb.Link) error {
context_mock.go
199
>
c.mu.Lock()
200
>
defer c.mu.Unlock()
201
>
if c.LinksByRequest == nil {
202
>
c.LinksByRequest = make(map[Component]map[string][]*commonpb.Link)
203
>
}
204
>
perRequest, ok := c.LinksByRequest[component]
205
>
if !ok {
206
>
perRequest = make(map[string][]*commonpb.Link)
207
>
c.LinksByRequest[component] = perRequest
208
>
}
209
>
if len(links) == 0 {
210
delete(perRequest, requestID)
212
>
perRequest[requestID] = links
213
>
}
214
>
return nil
215
}
216