Expose az sender Close() (#254)

* expose az sender close() * fix UT
Azure · Sep 11, 2024 · cae3c4d · cae3c4d
1 parent 0b93207
commit cae3c4d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/v2/sender.go b/v2/sender.go
@@ -25,6 +25,7 @@ type AzServiceBusSender interface {
 	SendMessage(ctx context.Context, message *azservicebus.Message, options *azservicebus.SendMessageOptions) error
 	SendMessageBatch(ctx context.Context, batch *azservicebus.MessageBatch, options *azservicebus.SendMessageBatchOptions) error
 	NewMessageBatch(ctx context.Context, options *azservicebus.MessageBatchOptions) (*azservicebus.MessageBatch, error)
+	Close(ctx context.Context) error
 }
 
 // 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 --git a/v2/sender_test.go b/v2/sender_test.go
@@ -330,6 +330,7 @@ type fakeAzSender struct {
 	NewMessageBatchReturnValue    *azservicebus.MessageBatch
 	NewMessageBatchErr            error
 	SendMessageBatchReceivedValue *azservicebus.MessageBatch
+	CloseErr                      error
 }
 
 func (f *fakeAzSender) SendMessage(
@@ -370,3 +371,7 @@ func (f *fakeAzSender) NewMessageBatch(
 	options *azservicebus.MessageBatchOptions) (*azservicebus.MessageBatch, error) {
 	return f.NewMessageBatchReturnValue, f.NewMessageBatchErr
 }
+
+func (f *fakeAzSender) Close(ctx context.Context) error {
+	return f.CloseErr
+}