Added test

ludoviclefevre · May 18, 2015 · 6fb6ce5 · 6fb6ce5
1 parent c4940ab
commit 6fb6ce5
Show file tree

Hide file tree

Showing 4 changed files with 46 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -8,6 +8,21 @@ Generate SEO-friendly sitemap.
 
 Inspired by XML Sitemap in Yoast Wordpress SEO Plugin (https://yoast.com).
 
+## Install
+
+``` bash
+$ npm install hexo-generator-seo-friendly-sitemap --save
+```
+
+## Options
+
+You can configure this plugin in `_config.yml`.
+
+``` yaml
+sitemap:
+    path: sitemap.xml
+```
+
 ## License
 
 MIT © [Ludovic LEFEVRE](http://www.ludoviclefevre.fr)

diff --git a/lib/generator.js b/lib/generator.js
@@ -1,11 +1,12 @@
+'use strict';
+
 var path = require('path'),
     ejs = require('ejs'),
     _ = require('lodash'),
     Promise = require('bluebird'),
     fs = Promise.promisifyAll(require('fs')),
 
     googleFriendlySitemap = function (locals) {
-        'use strict';
 
         var config = this.config,
             viewPath = path.join(__dirname, '../views/'),
@@ -31,6 +32,10 @@ var path = require('path'),
 
             filePaths = [],
 
+            isPluginEnabled = function (config) {
+                return (config.sitemap && config.sitemap.path);
+            },
+
             addPosts = function () {
                 if (locals.posts.length === 0) {
                     return;
@@ -159,7 +164,7 @@ var path = require('path'),
                 filePaths.push(
                     {
                         template: 'index-sitemap.ejs',
-                        filename: 'index-sitemap.xml',
+                        filename: config.sitemap.path,
                         items: indexSitemapItems
                     });
             },
@@ -177,6 +182,9 @@ var path = require('path'),
                 );
             };
 
+        if (!isPluginEnabled(config)) {
+            return Promise.resolve([]);
+        }
         addPosts();
         addPages();
         addCategories();

diff --git a/test/allSitemaps.js b/test/allSitemaps.js
@@ -76,7 +76,7 @@ it('should generate all sitemap files if posts, pages, categories and tags are d
         checkAssertions = function (result) {
             result.should.be.a('array');
 
-            var indexSitemap = _.find(result, {path: 'index-sitemap.xml'}),
+            var indexSitemap = _.find(result, {path: 'sitemap.xml'}),
                 postSitemap = _.find(result, {path: 'post-sitemap.xml'}),
                 pageSitemap = _.find(result, {path: 'page-sitemap.xml'}),
                 categorySitemap = _.find(result, {path: 'category-sitemap.xml'}),

diff --git a/test/disabledPlugin.js b/test/disabledPlugin.js
@@ -0,0 +1,20 @@
+'use strict';
+
+var Hexo = require('hexo'),
+    path = require('path');
+
+describe('Hexo', function () {
+    it('should not generate sitemap files if plugin is not enabled in _config.yml.', function () {
+        var hexo = new Hexo(__dirname, {silent: true}),
+            generator = require(path.join(__dirname, '../lib/generator')).bind(hexo),
+            locals,
+            checkAssertions = function (result) {
+                return result.should.be.empty;
+            };
+
+        locals = hexo.locals.toObject();
+
+        return generator(locals)
+            .then(checkAssertions);
+    });
+});