added defered confirmation to connection mock

knative-extensions · Jul 27, 2023 · b4852da · b4852da
1 parent 2c761bb
commit b4852da
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/rabbit/rabbit_mocks.go b/pkg/rabbit/rabbit_mocks.go
@@ -61,6 +61,7 @@ func (rm *RabbitMQBadConnectionMock) NotifyClose(c chan *amqp.Error) chan *amqp.
 type RabbitMQChannelMock struct {
 	NotifyCloseChannel chan *amqp.Error
 	ConsumeChannel     <-chan amqp.Delivery
+	dc                 *amqp.DeferredConfirmation
 }
 
 func (rm *RabbitMQChannelMock) IsClosed() bool {
@@ -83,7 +84,7 @@ func (rm *RabbitMQChannelMock) Consume(a string, b string, c bool, d bool, e boo
 }
 
 func (rm *RabbitMQChannelMock) PublishWithDeferredConfirm(a string, b string, c bool, d bool, p amqp.Publishing) (*amqp.DeferredConfirmation, error) {
-	return &amqp.DeferredConfirmation{}, nil
+	return rm.dc, nil
 }
 
 func (rm *RabbitMQChannelMock) Confirm(a bool) error {