Add basic tests for json generator

3 files changed, 3 insertions, 2 deletions

diff --git a/lib/modifiers/summary/indexArticleLevels.js b/lib/modifiers/summary/indexArticleLevels.js
index 02ca594..f311f74 100644
--- a/lib/modifiers/summary/indexArticleLevels.js
+++ b/lib/modifiers/summary/indexArticleLevels.js
@@ -8,7 +8,7 @@
 */
 function indexArticleLevels(article, baseLevel) {
     baseLevel = baseLevel || article.getLevel();
-    var articles = article.getArticle();
+    var articles = article.getArticles();
 
     articles = articles.map(function(inner, i) {
         return indexArticleLevels(inner, baseLevel + '.' + (i + 1));
diff --git a/lib/modifiers/summary/indexPartLevels.js b/lib/modifiers/summary/indexPartLevels.js
index 19ba089..d19c70a 100644
--- a/lib/modifiers/summary/indexPartLevels.js
+++ b/lib/modifiers/summary/indexPartLevels.js
@@ -9,7 +9,7 @@ var indexArticleLevels = require('./indexArticleLevels');
 */
 function indexPartLevels(part, index) {
     var baseLevel = String(index + 1);
-    var articles = part.getArticle();
+    var articles = part.getArticles();
 
     articles = articles.map(function(inner, i) {
         return indexArticleLevels(inner, baseLevel + '.' + (i + 1));
diff --git a/lib/modifiers/summary/unshiftArticle.js b/lib/modifiers/summary/unshiftArticle.js
index 6bb97c9..3f2ae4d 100644
--- a/lib/modifiers/summary/unshiftArticle.js
+++ b/lib/modifiers/summary/unshiftArticle.js
@@ -18,6 +18,7 @@ function unshiftArticle(summary, article) {
 
     var articles = part.getArticles();
     articles = articles.unshift(article);
+    part = part.set('articles', articles);
 
     parts = parts.set(0, part);
     summary = summary.set('parts', parts);