File tree 2 files changed +6
-0
lines changed
2 files changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -25,6 +25,7 @@ type AzServiceBusSender interface {
25
25
SendMessage (ctx context.Context , message * azservicebus.Message , options * azservicebus.SendMessageOptions ) error
26
26
SendMessageBatch (ctx context.Context , batch * azservicebus.MessageBatch , options * azservicebus.SendMessageBatchOptions ) error
27
27
NewMessageBatch (ctx context.Context , options * azservicebus.MessageBatchOptions ) (* azservicebus.MessageBatch , error )
28
+ Close (ctx context.Context ) error
28
29
}
29
30
30
31
// Sender contains an SBSender used to send the message to the ServiceBus queue and a Marshaller used to marshal any struct into a ServiceBus message
Original file line number Diff line number Diff line change @@ -330,6 +330,7 @@ type fakeAzSender struct {
330
330
NewMessageBatchReturnValue * azservicebus.MessageBatch
331
331
NewMessageBatchErr error
332
332
SendMessageBatchReceivedValue * azservicebus.MessageBatch
333
+ CloseErr error
333
334
}
334
335
335
336
func (f * fakeAzSender ) SendMessage (
@@ -370,3 +371,7 @@ func (f *fakeAzSender) NewMessageBatch(
370
371
options * azservicebus.MessageBatchOptions ) (* azservicebus.MessageBatch , error ) {
371
372
return f .NewMessageBatchReturnValue , f .NewMessageBatchErr
372
373
}
374
+
375
+ func (f * fakeAzSender ) Close (ctx context.Context ) error {
376
+ return f .CloseErr
377
+ }
You can’t perform that action at this time.
0 commit comments