Add spec for download item.
This commit is contained in:
parent
22168bc6d8
commit
5ef9c7e1a1
1 changed files with 50 additions and 0 deletions
|
@ -2,6 +2,7 @@ assert = require 'assert'
|
||||||
remote = require 'remote'
|
remote = require 'remote'
|
||||||
http = require 'http'
|
http = require 'http'
|
||||||
path = require 'path'
|
path = require 'path'
|
||||||
|
fs = require 'fs'
|
||||||
app = remote.require 'app'
|
app = remote.require 'app'
|
||||||
BrowserWindow = remote.require 'browser-window'
|
BrowserWindow = remote.require 'browser-window'
|
||||||
|
|
||||||
|
@ -72,3 +73,52 @@ describe 'session module', ->
|
||||||
quotas: ['persistent'],
|
quotas: ['persistent'],
|
||||||
w.webContents.session.clearStorageData options, ->
|
w.webContents.session.clearStorageData options, ->
|
||||||
w.webContents.send 'getcount'
|
w.webContents.send 'getcount'
|
||||||
|
|
||||||
|
describe 'DownloadItem', ->
|
||||||
|
# A 5MB mock pdf.
|
||||||
|
mockPDF = new Buffer(1024*1024*5)
|
||||||
|
contentDisposition = 'inline; filename="mock.pdf"'
|
||||||
|
# TODO(hokein): Change the download directory to spec/fixtures directory.
|
||||||
|
# We have to use the # default download directory due to the broken
|
||||||
|
# session.setDownloadPath API is broken. Once the API is fixed, we can make
|
||||||
|
# this change.
|
||||||
|
defaultDownloadDir = path.join app.getPath('userData'), 'Downloads'
|
||||||
|
downloadFilePath = path.join defaultDownloadDir, "mock.pdf"
|
||||||
|
downloadServer = http.createServer (req, res) ->
|
||||||
|
res.writeHead 200, {
|
||||||
|
'Content-Length': mockPDF.length,
|
||||||
|
'Content-Type': 'application/pdf',
|
||||||
|
'Content-Disposition': contentDisposition
|
||||||
|
}
|
||||||
|
res.end mockPDF
|
||||||
|
downloadServer.close()
|
||||||
|
|
||||||
|
it 'can download successfully', (done) ->
|
||||||
|
downloadServer.listen 0, '127.0.0.1', ->
|
||||||
|
{port} = downloadServer.address()
|
||||||
|
w.loadUrl "#{url}:#{port}"
|
||||||
|
w.webContents.session.setOpenDownloadDialog false
|
||||||
|
|
||||||
|
w.webContents.session.once 'will-download', (e, item, webContents) ->
|
||||||
|
item.on 'done', (e, state) ->
|
||||||
|
assert.equal state, "completed"
|
||||||
|
assert.equal item.getContentDisposition(), contentDisposition
|
||||||
|
assert.equal item.getReceiveBytes(), mockPDF.length
|
||||||
|
assert.equal item.getTotalBytes(), mockPDF.length
|
||||||
|
assert fs.existsSync downloadFilePath
|
||||||
|
fs.unlinkSync downloadFilePath
|
||||||
|
done()
|
||||||
|
assert.equal item.getURL(), "#{url}:#{port}/"
|
||||||
|
assert.equal item.getMimeType(), "application/pdf"
|
||||||
|
|
||||||
|
it 'can cancel download', (done) ->
|
||||||
|
downloadServer.listen 0, '127.0.0.1', ->
|
||||||
|
{port} = downloadServer.address()
|
||||||
|
w.loadUrl "#{url}:#{port}/"
|
||||||
|
w.webContents.session.setOpenDownloadDialog false
|
||||||
|
w.webContents.session.once 'will-download', (e, item, webContents) ->
|
||||||
|
item.pause()
|
||||||
|
item.on 'done', (e, state) ->
|
||||||
|
assert.equal state, "cancelled"
|
||||||
|
done()
|
||||||
|
item.cancel()
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue