Mukul Mishra
8 years ago
7 changed files with 417 additions and 170 deletions
@ -0,0 +1,234 @@
@@ -0,0 +1,234 @@
|
||||
/* Copyright 2017 Mozilla Foundation |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
/* globals __non_webpack_require__ */ |
||||
|
||||
import { assert, isNodeJS } from '../../src/shared/util'; |
||||
import { PDFNodeStream } from '../../src/display/node_stream'; |
||||
|
||||
// Make sure that we only running this script is Node.js environments.
|
||||
assert(isNodeJS()); |
||||
|
||||
let path = __non_webpack_require__('path'); |
||||
let url = __non_webpack_require__('url'); |
||||
let http = __non_webpack_require__('http'); |
||||
let fs = __non_webpack_require__('fs'); |
||||
|
||||
describe('node_stream', function() { |
||||
let server = null; |
||||
let port = null; |
||||
let pdf = url.parse(encodeURI('file://' + path.join(process.cwd(), |
||||
'./test/pdfs/tracemonkey.pdf'))).href; |
||||
let pdfLength = 1016315; |
||||
|
||||
beforeAll((done) => { |
||||
// Create http server to serve pdf data for tests.
|
||||
server = http.createServer((request, response) => { |
||||
let filePath = process.cwd() + '/test/pdfs' + request.url; |
||||
fs.lstat(filePath, (error, stat) => { |
||||
if (error) { |
||||
response.writeHead(404); |
||||
response.end(`File ${request.url} not found!`); |
||||
return; |
||||
} |
||||
if (!request.headers['range']) { |
||||
let contentLength = stat.size; |
||||
let stream = fs.createReadStream(filePath); |
||||
response.writeHead(200, { |
||||
'Content-Type': 'application/pdf', |
||||
'Content-Length': contentLength, |
||||
'Accept-Ranges': 'bytes', |
||||
}); |
||||
stream.pipe(response); |
||||
} else { |
||||
let [start, end] = |
||||
request.headers['range'].split('=')[1].split('-').map((x) => { |
||||
return Number(x); |
||||
}); |
||||
let stream = fs.createReadStream(filePath, { start, end, }); |
||||
response.writeHead(206, { |
||||
'Content-Type': 'application/pdf', |
||||
}); |
||||
stream.pipe(response); |
||||
} |
||||
}); |
||||
}).listen(0); /* Listen on a random free port */ |
||||
port = server.address().port; |
||||
done(); |
||||
}); |
||||
|
||||
afterAll((done) => { |
||||
// Close the server from accepting new connections after all test finishes.
|
||||
server.close(); |
||||
done(); |
||||
}); |
||||
|
||||
it('read both http(s) and filesystem pdf files', function(done) { |
||||
let stream1 = new PDFNodeStream({ |
||||
source: { |
||||
url: `http://127.0.0.1:${port}/tracemonkey.pdf`, |
||||
rangeChunkSize: 65536, |
||||
disableStream: true, |
||||
}, |
||||
disableRange: true, |
||||
}); |
||||
|
||||
let stream2 = new PDFNodeStream({ |
||||
source: { |
||||
url: pdf, |
||||
rangeChunkSize: 65536, |
||||
disableStream: true, |
||||
}, |
||||
disableRange: true, |
||||
}); |
||||
|
||||
let fullReader1 = stream1.getFullReader(); |
||||
let fullReader2 = stream2.getFullReader(); |
||||
|
||||
let isStreamingSupported1, isRangeSupported1; |
||||
let promise1 = fullReader1.headersReady.then(() => { |
||||
isStreamingSupported1 = fullReader1.isStreamingSupported; |
||||
isRangeSupported1 = fullReader1.isRangeSupported; |
||||
}); |
||||
|
||||
let isStreamingSupported2, isRangeSupported2; |
||||
let promise2 = fullReader2.headersReady.then(() => { |
||||
isStreamingSupported2 = fullReader2.isStreamingSupported; |
||||
isRangeSupported2 = fullReader2.isRangeSupported; |
||||
}); |
||||
|
||||
let len1 = 0, len2 = 0; |
||||
let read1 = function () { |
||||
return fullReader1.read().then(function (result) { |
||||
if (result.done) { |
||||
return; |
||||
} |
||||
len1 += result.value.byteLength; |
||||
return read1(); |
||||
}); |
||||
}; |
||||
let read2 = function () { |
||||
return fullReader2.read().then(function (result) { |
||||
if (result.done) { |
||||
return; |
||||
} |
||||
len2 += result.value.byteLength; |
||||
return read2(); |
||||
}); |
||||
}; |
||||
|
||||
let readPromise = Promise.all([read1(), read2(), promise1, promise2]); |
||||
readPromise.then((result) => { |
||||
expect(isStreamingSupported1).toEqual(false); |
||||
expect(isRangeSupported1).toEqual(false); |
||||
expect(isStreamingSupported2).toEqual(false); |
||||
expect(isRangeSupported2).toEqual(false); |
||||
expect(len1).toEqual(pdfLength); |
||||
expect(len1).toEqual(len2); |
||||
done(); |
||||
}).catch((reason) => { |
||||
done.fail(reason); |
||||
}); |
||||
}); |
||||
|
||||
it('read custom ranges for both http(s) and filesystem urls', |
||||
function(done) { |
||||
let rangeSize = 32768; |
||||
let stream1 = new PDFNodeStream({ |
||||
source: { |
||||
url: `http://127.0.0.1:${port}/tracemonkey.pdf`, |
||||
length: pdfLength, |
||||
rangeChunkSize: rangeSize, |
||||
disableStream: true, |
||||
}, |
||||
disableRange: false, |
||||
}); |
||||
let stream2 = new PDFNodeStream({ |
||||
source: { |
||||
url: pdf, |
||||
length: pdfLength, |
||||
rangeChunkSize: rangeSize, |
||||
disableStream: true, |
||||
}, |
||||
disableRange: false, |
||||
}); |
||||
|
||||
let fullReader1 = stream1.getFullReader(); |
||||
let fullReader2 = stream2.getFullReader(); |
||||
|
||||
let isStreamingSupported1, isRangeSupported1, fullReaderCancelled1; |
||||
let isStreamingSupported2, isRangeSupported2, fullReaderCancelled2; |
||||
|
||||
let promise1 = fullReader1.headersReady.then(function () { |
||||
isStreamingSupported1 = fullReader1.isStreamingSupported; |
||||
isRangeSupported1 = fullReader1.isRangeSupported; |
||||
// we shall be able to close the full reader without issues
|
||||
fullReader1.cancel('Don\'t need full reader'); |
||||
fullReaderCancelled1 = true; |
||||
}); |
||||
|
||||
let promise2 = fullReader2.headersReady.then(function () { |
||||
isStreamingSupported2 = fullReader2.isStreamingSupported; |
||||
isRangeSupported2 = fullReader2.isRangeSupported; |
||||
fullReader2.cancel('Don\'t need full reader'); |
||||
fullReaderCancelled2 = true; |
||||
}); |
||||
|
||||
// Skipping fullReader results, requesting something from the PDF end.
|
||||
let tailSize = (pdfLength % rangeSize) || rangeSize; |
||||
|
||||
let range11Reader = stream1.getRangeReader(pdfLength - tailSize - rangeSize, |
||||
pdfLength - tailSize); |
||||
let range12Reader = stream1.getRangeReader(pdfLength - tailSize, pdfLength); |
||||
|
||||
let range21Reader = stream2.getRangeReader(pdfLength - tailSize - rangeSize, |
||||
pdfLength - tailSize); |
||||
let range22Reader = stream2.getRangeReader(pdfLength - tailSize, pdfLength); |
||||
|
||||
let result11 = { value: 0, }, result12 = { value: 0, }; |
||||
let result21 = { value: 0, }, result22 = { value: 0, }; |
||||
|
||||
let read = function (reader, lenResult) { |
||||
return reader.read().then(function (result) { |
||||
if (result.done) { |
||||
return; |
||||
} |
||||
lenResult.value += result.value.byteLength; |
||||
return read(reader, lenResult); |
||||
}); |
||||
}; |
||||
|
||||
let readPromises = Promise.all([read(range11Reader, result11), |
||||
read(range12Reader, result12), |
||||
read(range21Reader, result21), |
||||
read(range22Reader, result22), |
||||
promise1, promise2]); |
||||
|
||||
readPromises.then(function () { |
||||
expect(result11.value).toEqual(rangeSize); |
||||
expect(result12.value).toEqual(tailSize); |
||||
expect(result21.value).toEqual(rangeSize); |
||||
expect(result22.value).toEqual(tailSize); |
||||
expect(isStreamingSupported1).toEqual(false); |
||||
expect(isRangeSupported1).toEqual(true); |
||||
expect(fullReaderCancelled1).toEqual(true); |
||||
expect(isStreamingSupported2).toEqual(false); |
||||
expect(isRangeSupported2).toEqual(true); |
||||
expect(fullReaderCancelled2).toEqual(true); |
||||
done(); |
||||
}).catch(function (reason) { |
||||
done.fail(reason); |
||||
}); |
||||
}); |
||||
}); |
||||
Loading…
Reference in new issue