tests: add check for sitemaps count in result

vladkosinov · May 19, 2015 · f33e8ac · f33e8ac
1 parent 65407b1
commit f33e8ac
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/test/test.js b/test/test.js
@@ -189,7 +189,12 @@ describe('toXML()', function () {
             sm.addUrl(path.join(hostName, i + ''));
         }
 
-        sm.toXML().forEach(function eachShouldBeEqual(sm) {
+        var files = sm.toXML();
+
+        var filesExpectedCount = Object.keys(dataExpected['multiple-files']['splited-by-limit']).length;
+        files.length.should.be.eql(filesExpectedCount);
+
+        files.forEach(function eachShouldBeEqual(sm) {
             sm.xml.should.equal(dataExpected['multiple-files']['splited-by-limit'][sm.fileName]);
         });
 
@@ -215,7 +220,12 @@ describe('toXML()', function () {
         sm1.addSitemap(sm3);
         sm2.addSitemap(sm4);
 
-        sm1.toXML().forEach(function eachShouldBeEqual(sm) {
+        var files = sm1.toXML();
+
+        var filesExpectedCount = Object.keys(dataExpected['multiple-files']['horizontal-structure']).length;
+        files.length.should.be.eql(filesExpectedCount);
+
+        files.forEach(function eachShouldBeEqual(sm) {
             sm.xml.should.equal(dataExpected['multiple-files']['horizontal-structure'][sm.fileName]);
         });
 
@@ -242,6 +252,9 @@ describe('example from README', function () {
 
         var files = sm.toXML();
 
+        var filesExpectedCount = Object.keys(dataExpected['multiple-files']['from-readme']).length;
+        files.length.should.be.eql(filesExpectedCount);
+
         files.forEach(function eachShouldBeEqual(sm) {
             sm.xml.should.equal(dataExpected['multiple-files']['from-readme'][sm.fileName]);
         });