Add unit test for client.js code coverage

cmdkoh · Nov 30, 2016 · 466b5b1 · 466b5b1
1 parent 2222ff9
commit 466b5b1
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/lib/client.js b/lib/client.js
@@ -60,7 +60,7 @@ module.exports = function (dependencies) {
       }, notification.headers);
 
       if (this.config.token) {
-        if (this.config.token.isExpired(this.config.token.current, "3300")) {
+        if (this.config.token.isExpired(this.config.token.current, 3300)) {
           this.config.token.regenerate(this.config.token.generation);
         }
         headers.authorization = "bearer " + this.config.token.current;

diff --git a/test/client.js b/test/client.js
@@ -642,6 +642,25 @@ describe("Client", function () {
 
           return expect(client.write(builtNotification(), "adfe5969")).to.eventually.have.property("status", "403");
         });
+
+        it("regenerate token", function () {
+          fakes.stream = new FakeStream("abcd1234", "200");
+          fakes.endpointManager.getStream.onCall(0).returns(fakes.stream);
+
+          fakes.token.isExpired = function (current, validSeconds) {
+            return true;
+          }
+
+          let client = new Client({
+            address: "testapi",
+            token: fakes.token
+          });
+
+          return client.write(builtNotification(), "abcd1234")
+            .then(function () {
+              expect(fakes.token.generation).to.equal(1);
+            });
+        });
       });
     });
   });