80 lines
2.4 KiB
Go
80 lines
2.4 KiB
Go
package rabbitmqamqp
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/Azure/go-amqp"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Management tests", func() {
|
|
It("AMQP Management should fail due to context cancellation", func() {
|
|
connection, err := Dial(context.Background(), "amqp://", nil)
|
|
Expect(err).To(BeNil())
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Millisecond)
|
|
cancel()
|
|
err = connection.Management().Open(ctx, connection)
|
|
Expect(err).NotTo(BeNil())
|
|
Expect(connection.Close(context.Background())).To(BeNil())
|
|
})
|
|
|
|
It("AMQP Management should receive events", func() {
|
|
ch := make(chan *StateChanged, 2)
|
|
connection, err := Dial(context.Background(), "amqp://", &AmqpConnOptions{
|
|
SASLType: amqp.SASLTypeAnonymous(),
|
|
RecoveryConfiguration: &RecoveryConfiguration{
|
|
ActiveRecovery: false,
|
|
},
|
|
})
|
|
Expect(err).To(BeNil())
|
|
connection.NotifyStatusChange(ch)
|
|
err = connection.Close(context.Background())
|
|
Expect(err).To(BeNil())
|
|
recv := <-ch
|
|
Expect(recv).NotTo(BeNil())
|
|
|
|
Expect(recv.From).To(Equal(&StateOpen{}))
|
|
Expect(recv.To).To(Equal(&StateClosed{}))
|
|
Expect(connection.Close(context.Background())).To(BeNil())
|
|
})
|
|
|
|
It("Request", func() {
|
|
|
|
connection, err := Dial(context.Background(), "amqp://", nil)
|
|
Expect(err).To(BeNil())
|
|
|
|
management := connection.Management()
|
|
kv := make(map[string]any)
|
|
kv["durable"] = true
|
|
kv["auto_delete"] = true
|
|
_queueArguments := make(map[string]any)
|
|
_queueArguments["x-queue-type"] = "classic"
|
|
kv["arguments"] = _queueArguments
|
|
path := "/queues/test"
|
|
result, err := management.Request(context.Background(), kv, path, "PUT", []int{200})
|
|
Expect(err).To(BeNil())
|
|
Expect(result).NotTo(BeNil())
|
|
result, err = management.Request(context.Background(), amqp.Null{}, path, "DELETE", []int{responseCode200})
|
|
Expect(err).To(BeNil())
|
|
Expect(result).NotTo(BeNil())
|
|
Expect(management.Close(context.Background())).To(BeNil())
|
|
|
|
Expect(connection.Close(context.Background())).To(BeNil())
|
|
})
|
|
|
|
It("GET on non-existing queue returns ErrDoesNotExist", func() {
|
|
|
|
connection, err := Dial(context.Background(), "amqp://", nil)
|
|
Expect(err).To(BeNil())
|
|
|
|
management := connection.Management()
|
|
path := "/queues/i-do-not-exist"
|
|
result, err := management.Request(context.Background(), amqp.Null{}, path, commandGet, []int{responseCode200, responseCode404})
|
|
Expect(err).To(Equal(ErrDoesNotExist))
|
|
Expect(result).To(BeNil())
|
|
})
|
|
})
|