From 20ed53314e96dad6745b7f50888a97d83a3b28ad Mon Sep 17 00:00:00 2001
From: Janilton Maciel <janilton.maciel@g.globo>
Date: Thu, 20 Aug 2020 16:28:11 -0300
Subject: [PATCH] add method Disconnect

---
 enqueuestomp_test.go | 10 ++++++++++
 enqueustomp.go       |  4 ++++
 2 files changed, 14 insertions(+)

diff --git a/enqueuestomp_test.go b/enqueuestomp_test.go
index d3a3738..41ec155 100644
--- a/enqueuestomp_test.go
+++ b/enqueuestomp_test.go
@@ -379,3 +379,13 @@ func (s *EnqueueStompSuite) TestSendTopicWithWriteDisk(c *check.C) {
 	enqueueCount := s.j.StatTopic(topicName, "EnqueueCount")
 	c.Assert(enqueueCount, check.Equals, strconv.Itoa(total))
 }
+
+func (s *EnqueueStompSuite) TestDisconnect(c *check.C) {
+	enqueue, err := enqueuestomp.NewEnqueueStomp(
+		enqueuestomp.Config{},
+	)
+	c.Assert(err, check.IsNil)
+
+	err = enqueue.Disconnect()
+	c.Assert(err, check.IsNil)
+}
diff --git a/enqueustomp.go b/enqueustomp.go
index 9c17f69..1bac869 100644
--- a/enqueustomp.go
+++ b/enqueustomp.go
@@ -94,6 +94,10 @@ func (emq *EnqueueStomp) Config() Config {
 	return emq.config
 }
 
+func (emq *EnqueueStomp) Disconnect() error {
+	return emq.conn.Disconnect()
+}
+
 func (emq *EnqueueStomp) send(destinationType string, destinationName string, body []byte, sc SendConfig) error {
 	if len(body) == 0 {
 		return ErrEmptyBody