From b2e6b05b58d77dc2ea454e6d0b1aa6b81330417f Mon Sep 17 00:00:00 2001
From: mdevils <mdevils@yandex.ru>
Date: Fri, 21 Nov 2014 02:45:32 +0300
Subject: [PATCH] getLastNodeToken test

---
 test/js-file.js | 43 ++++++++++++++++++++++++++++++-------------
 1 file changed, 30 insertions(+), 13 deletions(-)

diff --git a/test/js-file.js b/test/js-file.js
index b336a72ad..15e2fdc83 100644
--- a/test/js-file.js
+++ b/test/js-file.js
@@ -51,19 +51,6 @@ describe('modules/js-file', function() {
         });
     });
 
-    it('should return token for specified node', function() {
-        var str = 'if (true) { while (true) x++; }';
-        var file = new JsFile(null, str, esprima.parse(str, {loc: true, range: true, tokens: true}));
-
-        var ifToken = file.getFirstNodeToken(file.getNodesByType('IfStatement')[0]);
-        assert(ifToken.type === 'Keyword');
-        assert(ifToken.value === 'if');
-
-        var incToken = file.getFirstNodeToken(file.getNodesByType('UpdateExpression')[0]);
-        assert(incToken.type === 'Identifier');
-        assert(incToken.value === 'x');
-    });
-
     describe('iterateTokenByValue',  function() {
         it('should find token by value', function() {
             var str = 'if (true);';
@@ -445,4 +432,34 @@ describe('modules/js-file', function() {
             assert(token === undefined);
         });
     });
+
+    describe('getFirstNodeToken', function() {
+        it('should return token for specified node', function() {
+            var str = 'if (true) { while (true) x++; }';
+            var file = new JsFile(null, str, esprima.parse(str, {loc: true, range: true, tokens: true}));
+
+            var ifToken = file.getFirstNodeToken(file.getNodesByType('IfStatement')[0]);
+            assert.equal(ifToken.type, 'Keyword');
+            assert.equal(ifToken.value, 'if');
+
+            var incToken = file.getFirstNodeToken(file.getNodesByType('UpdateExpression')[0]);
+            assert.equal(incToken.type, 'Identifier');
+            assert.equal(incToken.value, 'x');
+        });
+    });
+
+    describe('getLastNodeToken', function() {
+        it('should return token for specified node', function() {
+            var str = 'if (true) { while (true) x++; }';
+            var file = new JsFile(null, str, esprima.parse(str, {loc: true, range: true, tokens: true}));
+
+            var ifToken = file.getLastNodeToken(file.getNodesByType('IfStatement')[0]);
+            assert.equal(ifToken.type, 'Punctuator');
+            assert.equal(ifToken.value, '}');
+
+            var incToken = file.getLastNodeToken(file.getNodesByType('UpdateExpression')[0]);
+            assert.equal(incToken.type, 'Punctuator');
+            assert.equal(incToken.value, '++');
+        });
+    });
 });