Skip to content

Commit cae3c4d

Browse files
authored
Expose az sender Close() (#254)
* expose az sender close() * fix UT
1 parent 0b93207 commit cae3c4d

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

Diff for: v2/sender.go

+1
Original file line numberDiff line numberDiff 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

Diff for: v2/sender_test.go

+5
Original file line numberDiff line numberDiff 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

335336
func (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+
}

0 commit comments

Comments
 (0)