From e9a9ead76264e314dfa15ab2332c98c63a9f5a4d Mon Sep 17 00:00:00 2001
From: Dhyey Thakore <dhyey35@gmail.com>
Date: Wed, 9 Jan 2019 19:32:11 +0530
Subject: [PATCH] [Integration-Test] Add test for number of downloads, fixes
 #810

---
 app/ui/expiryOptions.js                     |  6 ++++--
 app/ui/selectbox.js                         |  3 ++-
 test/integration/download-tests.js          | 23 +++++++++++++++++++++
 test/integration/pages/desktop/home_page.js |  2 ++
 4 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/app/ui/expiryOptions.js b/app/ui/expiryOptions.js
index b301e600..0a3268a6 100644
--- a/app/ui/expiryOptions.js
+++ b/app/ui/expiryOptions.js
@@ -42,7 +42,8 @@ module.exports = function(state, emit) {
         }
         state.downloadCount = value;
         emit('render');
-      }
+      },
+      'expire-after-dl-count-dropdown'
     ),
     dlCountSelect
   );
@@ -68,7 +69,8 @@ module.exports = function(state, emit) {
         }
         state.timeLimit = value;
         emit('render');
-      }
+      },
+      'expire-after-time-dropdown'
     ),
     timeSelect
   );
diff --git a/app/ui/selectbox.js b/app/ui/selectbox.js
index e1f7a040..3f66496b 100644
--- a/app/ui/selectbox.js
+++ b/app/ui/selectbox.js
@@ -1,10 +1,11 @@
 const html = require('choo/html');
 
-module.exports = function(selected, options, translate, changed) {
+module.exports = function(selected, options, translate, changed, htmlId) {
   let x = selected;
 
   return html`
     <select
+      id="${htmlId}"
       class="appearance-none cursor-pointer border rounded-sm bg-blue-lightest hover:border-blue focus:border-blue px-2 py-1 my-2 h-8"
       onchange="${choose}"
     >
diff --git a/test/integration/download-tests.js b/test/integration/download-tests.js
index aa23d994..3dfdee9e 100644
--- a/test/integration/download-tests.js
+++ b/test/integration/download-tests.js
@@ -34,4 +34,27 @@ describe('Firefox Send', function() {
       assert.ok(fs.existsSync(path.join(downloadDir, file)));
     });
   });
+
+  it('should update the download count on home page after 1 download', function() {
+    const expectedExpiresAfterText = 'Expires after 1 download';
+    browser.chooseFile(
+      homePage.uploadInput,
+      `${testFilesPath}/${testFiles[0]}`
+    );
+    browser.waitForExist(homePage.uploadButton);
+    browser.waitForExist(homePage.downloadCountDropdown);
+    browser.selectByIndex(homePage.downloadCountDropdown, 1);
+    browser.click(homePage.uploadButton);
+    browser.waitForExist(homePage.shareUrl);
+    const downloadPage = new DownloadPage(browser.getValue(homePage.shareUrl));
+    downloadPage.open();
+    downloadPage.download();
+    browser.waitForExist(downloadPage.downloadComplete);
+    browser.back();
+    browser.waitForExist(homePage.expiresAfterText);
+    assert.ok(
+      browser.getText(homePage.expiresAfterText).substring(0, 24) ===
+        expectedExpiresAfterText
+    );
+  });
 });
diff --git a/test/integration/pages/desktop/home_page.js b/test/integration/pages/desktop/home_page.js
index df8edf66..67cb66fc 100644
--- a/test/integration/pages/desktop/home_page.js
+++ b/test/integration/pages/desktop/home_page.js
@@ -9,6 +9,8 @@ class HomePage extends Page {
     this.uploadButton = '#upload-btn';
     this.progress = 'progress';
     this.shareUrl = '#share-url';
+    this.downloadCountDropdown = '#expire-after-dl-count-dropdown';
+    this.expiresAfterText = '.text-xs.text-grey-dark.w-full.mt-2.mb-2';
   }
 
   waitForPageToLoad() {