Saving a document doesn't show correct origin Fix brave#8698. Auditors …

…@bsclifton @diracdeltas
kumarrishav · May 10, 2017 · c2de975 · c2de975
1 parent 9b85662
commit c2de975
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 1 deletion.
diff --git a/app/renderer/components/downloadItem.js b/app/renderer/components/downloadItem.js
@@ -73,6 +73,8 @@ class DownloadItem extends ImmutableComponent {
   }
   render () {
     const origin = getOrigin(this.props.download.get('url'))
+    const localFileOrigins = ['file:', 'blob:', 'data:', 'chrome-extension:', 'chrome:']
+    const localFile = localFileOrigins.some((localFileOrigin) => origin.startsWith(localFileOrigin))
     const progressStyle = {
       width: downloadUtil.getPercentageComplete(this.props.download)
     }
@@ -160,7 +162,7 @@ class DownloadItem extends ImmutableComponent {
                     ? <span className='fa fa-unlock isInsecure' />
                     : null
                 }
-                <span title={origin}>{origin}</span>
+                <span title={origin}>{localFile ? 'Local file' : origin}</span>
               </div>
               : null
           }

diff --git a/test/unit/app/renderer/downloadItemTest.js b/test/unit/app/renderer/downloadItemTest.js
@@ -18,6 +18,7 @@ require('../../braveUnit')
 
 const savePath = path.join(require('os').tmpdir(), 'mostHatedPrimes.txt')
 const downloadUrl = 'http://www.bradrichter.com/mostHatedPrimes.txt'
+const localFileDownloadUrl = 'file:///Users/foobar/Library/abc.txt'
 const newDownload = (state) => Immutable.fromJS({
   startTime: new Date().getTime(),
   filename: 'mostHatedPrimes.txt',
@@ -28,6 +29,16 @@ const newDownload = (state) => Immutable.fromJS({
   deleteConfirmationVisible: false,
   state
 })
+const newDownloadLocalFile = (state) => Immutable.fromJS({
+  startTime: new Date().getTime(),
+  filename: 'abc.txt',
+  savePath,
+  url: localFileDownloadUrl,
+  totalBytes: 104729,
+  receivedBytes: 96931,
+  deleteConfirmationVisible: false,
+  state
+})
 
 describe('downloadItem component', function () {
   before(function () {
@@ -45,6 +56,29 @@ describe('downloadItem component', function () {
     mockery.disable()
   })
 
+  Object.values(downloadStates).forEach(function (state) {
+    describe(`${state} download local item`, function () {
+      before(function () {
+        this.downloadId = uuid.v4()
+        this.download = newDownloadLocalFile(state)
+        this.result = mount(
+          <DownloadItem
+            downloadId={this.downloadId}
+            download={this.download}
+            deleteConfirmationVisible={this.download.get('deleteConfirmationVisible')} />
+        )
+      })
+
+      it('filename exists and matches download filename', function () {
+        assert.equal(this.result.find('.downloadFilename').text(), this.download.get('filename'))
+      })
+
+      it('has local origin i.e file: and matches to "Local file" origin', function () {
+        assert.equal(this.result.find('.downloadOrigin').text(), 'Local file')
+      })
+    })
+  })
+
   Object.values(downloadStates).forEach(function (state) {
     describe(`${state} download item`, function () {
       before(function () {