test: fix schedule task test case (#29)

eggjs · Mar 22, 2019 · 6be344f · 6be344f
1 parent f266d34
commit 6be344f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/test/file-mode.test.js b/test/file-mode.test.js
@@ -327,10 +327,17 @@ describe('test/file-mode.test.js', () => {
       ];
       const shouldKeepDirs = [
         path.join(app.config.multipart.tmpdir, moment().subtract(2, 'years').format('YYYY/MM/DD/HH')),
-        path.join(app.config.multipart.tmpdir, moment().subtract(4, 'months').format('YYYY/MM/DD/HH')),
         path.join(app.config.multipart.tmpdir, moment().subtract(8, 'days').format('YYYY/MM/DD/HH')),
         path.join(app.config.multipart.tmpdir, moment().format('YYYY/MM/DD/HH')),
       ];
+      const currentMonth = new Date().getMonth();
+      const fourMonthBefore = path.join(app.config.multipart.tmpdir, moment().subtract(4, 'months').format('YYYY/MM/DD/HH'));
+      if (currentMonth < 4) {
+        // if current month is less than April, four months before shoule be last year.
+        oldDirs.push(fourMonthBefore);
+      } else {
+        shouldKeepDirs.push(fourMonthBefore);
+      }
       await Promise.all(oldDirs.map(dir => mkdirp(dir)));
       await Promise.all(shouldKeepDirs.map(dir => mkdirp(dir)));