add method Disconnect

globocom · Aug 20, 2020 · 20ed533 · 20ed533
1 parent 0345f44
commit 20ed533
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git 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
@@ -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