File tree Expand file tree Collapse file tree 2 files changed +6
-0
lines changed Expand file tree Collapse file tree 2 files changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -25,6 +25,7 @@ type AzServiceBusSender interface {
2525 SendMessage (ctx context.Context , message * azservicebus.Message , options * azservicebus.SendMessageOptions ) error
2626 SendMessageBatch (ctx context.Context , batch * azservicebus.MessageBatch , options * azservicebus.SendMessageBatchOptions ) error
2727 NewMessageBatch (ctx context.Context , options * azservicebus.MessageBatchOptions ) (* azservicebus.MessageBatch , error )
28+ Close (ctx context.Context ) error
2829}
2930
3031// 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 {
330330 NewMessageBatchReturnValue * azservicebus.MessageBatch
331331 NewMessageBatchErr error
332332 SendMessageBatchReceivedValue * azservicebus.MessageBatch
333+ CloseErr error
333334}
334335
335336func (f * fakeAzSender ) SendMessage (
@@ -370,3 +371,7 @@ func (f *fakeAzSender) NewMessageBatch(
370371 options * azservicebus.MessageBatchOptions ) (* azservicebus.MessageBatch , error ) {
371372 return f .NewMessageBatchReturnValue , f .NewMessageBatchErr
372373}
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