From a9f2efecfeeb9a14d235e7623a1a1acee7a2eae1 Mon Sep 17 00:00:00 2001
From: Espen Hovlandsdal <espen@hovlandsdal.com>
Date: Wed, 17 May 2017 14:49:15 +0200
Subject: [PATCH] Make `close()` a prototype method. Closes #67.

---
 lib/eventsource.js       | 12 +++++++++++-
 test/eventsource_test.js | 18 ++++++++++++++++++
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/lib/eventsource.js b/lib/eventsource.js
index fd1d48c..a99dc06 100644
--- a/lib/eventsource.js
+++ b/lib/eventsource.js
@@ -214,7 +214,7 @@ function EventSource (url, eventSourceInitDict) {
     }
   }
 
-  this.close = function () {
+  this._close = function () {
     if (readyState === EventSource.CLOSED) return
     readyState = EventSource.CLOSED
     if (req.abort) req.abort()
@@ -309,6 +309,16 @@ EventSource.prototype.CONNECTING = 0
 EventSource.prototype.OPEN = 1
 EventSource.prototype.CLOSED = 2
 
+/**
+ * Closes the connection, if one is made, and sets the readyState attribute to 2 (closed)
+ *
+ * @see https://developer.mozilla.org/en-US/docs/Web/API/EventSource/close
+ * @api public
+ */
+EventSource.prototype.close = function () {
+  this._close()
+}
+
 /**
  * Emulates the W3C Browser based WebSocket interface using addEventListener.
  *
diff --git a/test/eventsource_test.js b/test/eventsource_test.js
index b6c9113..ecfec19 100644
--- a/test/eventsource_test.js
+++ b/test/eventsource_test.js
@@ -872,6 +872,24 @@ describe('readyState', function () {
   })
 })
 
+describe('Methods', function () {
+  it('close method exists and can be called to close an eventsource', function (done) {
+    createServer(function (err, server) {
+      if (err) return done(err)
+      var es = new EventSource(server.url)
+      server.on('request', writeEvents([]))
+      es.onopen = function () {
+        assert.equal(es.close(), undefined)
+        server.close(done)
+      }
+    })
+  })
+
+  it('close method is a prototype method', function () {
+    assert.equal(typeof EventSource.prototype.close, 'function')
+  })
+})
+
 describe('Properties', function () {
   it('url exposes original request url', function () {
     var url = 'http://localhost:' + _port