fix download pages test and rearrange download spec files

igemuoftATG · Sep 27, 2016 · 33830eb · 33830eb
1 parent b62aede
commit 33830eb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 33 deletions.
diff --git a/test/download-all.spec.js b/test/download-all.spec.js
diff --git a/test/download-page.spec.js → test/download.spec.js b/test/download-page.spec.js → test/download.spec.js
@@ -10,25 +10,24 @@ chai.use(chaiAsPromised)
 const { assert, should } = chai
 should()
 
-const { downloadPage, getTeamPages } = require('../index.js')({ teamName: 'Toronto' })
+const { getTeamPages, downloadPage, downloadAll } = require('../index.js')({ teamName: 'Toronto' })
 
-describe('Download page', function() {
+describe('Download', function() {
   it('Should download all pages from getTeamPages', function() {
     this.timeout(15 * 1000)
 
     const dir = path.resolve(__dirname, 'download')
 
-    let expectedLength
 
     return getTeamPages()
-      .then((pages) => {
-        expectedLength = pages.length
-
-        return Promise.all(pages.map(page => downloadPage({ page, dir })))
-      })
-      .then(() => fs.readdirAsync(dir))
-      .then((files) => {
-        assert.isOk(files.length === expectedLength, 'Did not download all pages')
-      })
+      .then(pages => Promise.all(pages.map(page => downloadPage({ page, dir }))))
+  })
+
+  it('Should download all pages and templates', function() {
+    this.timeout(30 * 1000)
+
+    return downloadAll({ dir: path.resolve(__dirname, './download') }).then((results) => {
+      console.log('results: ', results)
+    })
   })
 })