1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 | 'use strict';
|
23 |
|
24 | var _slicedToArray = function () { function sliceIterator(arr, i) { var _arr = []; var _n = true; var _d = false; var _e = undefined; try { for (var _i = arr[Symbol.iterator](), _s; !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"]) _i["return"](); } finally { if (_d) throw _e; } } return _arr; } return function (arr, i) { if (Array.isArray(arr)) { return arr; } else if (Symbol.iterator in Object(arr)) { return sliceIterator(arr, i); } else { throw new TypeError("Invalid attempt to destructure non-iterable instance"); } }; }();
|
25 |
|
26 | var _util = require('../../shared/util');
|
27 |
|
28 | var _is_node = require('../../shared/is_node');
|
29 |
|
30 | var _is_node2 = _interopRequireDefault(_is_node);
|
31 |
|
32 | var _node_stream = require('../../display/node_stream');
|
33 |
|
34 | function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
35 |
|
36 | (0, _util.assert)((0, _is_node2.default)());
|
37 | var path = require('path');
|
38 | var url = require('url');
|
39 | var http = require('http');
|
40 | var fs = require('fs');
|
41 | describe('node_stream', function () {
|
42 | var server = null;
|
43 | var port = null;
|
44 | var pdf = url.parse(encodeURI('file://' + path.join(process.cwd(), './test/pdfs/tracemonkey.pdf'))).href;
|
45 | var pdfLength = 1016315;
|
46 | beforeAll(function (done) {
|
47 | server = http.createServer(function (request, response) {
|
48 | var filePath = process.cwd() + '/test/pdfs' + request.url;
|
49 | fs.lstat(filePath, function (error, stat) {
|
50 | if (error) {
|
51 | response.writeHead(404);
|
52 | response.end('File ' + request.url + ' not found!');
|
53 | return;
|
54 | }
|
55 | if (!request.headers['range']) {
|
56 | var contentLength = stat.size;
|
57 | var stream = fs.createReadStream(filePath);
|
58 | response.writeHead(200, {
|
59 | 'Content-Type': 'application/pdf',
|
60 | 'Content-Length': contentLength,
|
61 | 'Accept-Ranges': 'bytes'
|
62 | });
|
63 | stream.pipe(response);
|
64 | } else {
|
65 | var _request$headers$rang = request.headers['range'].split('=')[1].split('-').map(function (x) {
|
66 | return Number(x);
|
67 | }),
|
68 | _request$headers$rang2 = _slicedToArray(_request$headers$rang, 2),
|
69 | start = _request$headers$rang2[0],
|
70 | end = _request$headers$rang2[1];
|
71 |
|
72 | var _stream = fs.createReadStream(filePath, {
|
73 | start: start,
|
74 | end: end
|
75 | });
|
76 | response.writeHead(206, { 'Content-Type': 'application/pdf' });
|
77 | _stream.pipe(response);
|
78 | }
|
79 | });
|
80 | }).listen(0);
|
81 | port = server.address().port;
|
82 | done();
|
83 | });
|
84 | afterAll(function (done) {
|
85 | server.close();
|
86 | done();
|
87 | });
|
88 | it('read both http(s) and filesystem pdf files', function (done) {
|
89 | var stream1 = new _node_stream.PDFNodeStream({
|
90 | url: 'http://127.0.0.1:' + port + '/tracemonkey.pdf',
|
91 | rangeChunkSize: 65536,
|
92 | disableStream: true,
|
93 | disableRange: true
|
94 | });
|
95 | var stream2 = new _node_stream.PDFNodeStream({
|
96 | url: pdf,
|
97 | rangeChunkSize: 65536,
|
98 | disableStream: true,
|
99 | disableRange: true
|
100 | });
|
101 | var fullReader1 = stream1.getFullReader();
|
102 | var fullReader2 = stream2.getFullReader();
|
103 | var isStreamingSupported1 = void 0,
|
104 | isRangeSupported1 = void 0;
|
105 | var promise1 = fullReader1.headersReady.then(function () {
|
106 | isStreamingSupported1 = fullReader1.isStreamingSupported;
|
107 | isRangeSupported1 = fullReader1.isRangeSupported;
|
108 | });
|
109 | var isStreamingSupported2 = void 0,
|
110 | isRangeSupported2 = void 0;
|
111 | var promise2 = fullReader2.headersReady.then(function () {
|
112 | isStreamingSupported2 = fullReader2.isStreamingSupported;
|
113 | isRangeSupported2 = fullReader2.isRangeSupported;
|
114 | });
|
115 | var len1 = 0,
|
116 | len2 = 0;
|
117 | var read1 = function read1() {
|
118 | return fullReader1.read().then(function (result) {
|
119 | if (result.done) {
|
120 | return;
|
121 | }
|
122 | len1 += result.value.byteLength;
|
123 | return read1();
|
124 | });
|
125 | };
|
126 | var read2 = function read2() {
|
127 | return fullReader2.read().then(function (result) {
|
128 | if (result.done) {
|
129 | return;
|
130 | }
|
131 | len2 += result.value.byteLength;
|
132 | return read2();
|
133 | });
|
134 | };
|
135 | var readPromise = Promise.all([read1(), read2(), promise1, promise2]);
|
136 | readPromise.then(function (result) {
|
137 | expect(isStreamingSupported1).toEqual(false);
|
138 | expect(isRangeSupported1).toEqual(false);
|
139 | expect(isStreamingSupported2).toEqual(false);
|
140 | expect(isRangeSupported2).toEqual(false);
|
141 | expect(len1).toEqual(pdfLength);
|
142 | expect(len1).toEqual(len2);
|
143 | done();
|
144 | }).catch(function (reason) {
|
145 | done.fail(reason);
|
146 | });
|
147 | });
|
148 | it('read custom ranges for both http(s) and filesystem urls', function (done) {
|
149 | var rangeSize = 32768;
|
150 | var stream1 = new _node_stream.PDFNodeStream({
|
151 | url: 'http://127.0.0.1:' + port + '/tracemonkey.pdf',
|
152 | length: pdfLength,
|
153 | rangeChunkSize: rangeSize,
|
154 | disableStream: true,
|
155 | disableRange: false
|
156 | });
|
157 | var stream2 = new _node_stream.PDFNodeStream({
|
158 | url: pdf,
|
159 | length: pdfLength,
|
160 | rangeChunkSize: rangeSize,
|
161 | disableStream: true,
|
162 | disableRange: false
|
163 | });
|
164 | var fullReader1 = stream1.getFullReader();
|
165 | var fullReader2 = stream2.getFullReader();
|
166 | var isStreamingSupported1 = void 0,
|
167 | isRangeSupported1 = void 0,
|
168 | fullReaderCancelled1 = void 0;
|
169 | var isStreamingSupported2 = void 0,
|
170 | isRangeSupported2 = void 0,
|
171 | fullReaderCancelled2 = void 0;
|
172 | var promise1 = fullReader1.headersReady.then(function () {
|
173 | isStreamingSupported1 = fullReader1.isStreamingSupported;
|
174 | isRangeSupported1 = fullReader1.isRangeSupported;
|
175 | fullReader1.cancel('Don\'t need full reader');
|
176 | fullReaderCancelled1 = true;
|
177 | });
|
178 | var promise2 = fullReader2.headersReady.then(function () {
|
179 | isStreamingSupported2 = fullReader2.isStreamingSupported;
|
180 | isRangeSupported2 = fullReader2.isRangeSupported;
|
181 | fullReader2.cancel('Don\'t need full reader');
|
182 | fullReaderCancelled2 = true;
|
183 | });
|
184 | var tailSize = pdfLength % rangeSize || rangeSize;
|
185 | var range11Reader = stream1.getRangeReader(pdfLength - tailSize - rangeSize, pdfLength - tailSize);
|
186 | var range12Reader = stream1.getRangeReader(pdfLength - tailSize, pdfLength);
|
187 | var range21Reader = stream2.getRangeReader(pdfLength - tailSize - rangeSize, pdfLength - tailSize);
|
188 | var range22Reader = stream2.getRangeReader(pdfLength - tailSize, pdfLength);
|
189 | var result11 = { value: 0 },
|
190 | result12 = { value: 0 };
|
191 | var result21 = { value: 0 },
|
192 | result22 = { value: 0 };
|
193 | var read = function read(reader, lenResult) {
|
194 | return reader.read().then(function (result) {
|
195 | if (result.done) {
|
196 | return;
|
197 | }
|
198 | lenResult.value += result.value.byteLength;
|
199 | return read(reader, lenResult);
|
200 | });
|
201 | };
|
202 | var readPromises = Promise.all([read(range11Reader, result11), read(range12Reader, result12), read(range21Reader, result21), read(range22Reader, result22), promise1, promise2]);
|
203 | readPromises.then(function () {
|
204 | expect(result11.value).toEqual(rangeSize);
|
205 | expect(result12.value).toEqual(tailSize);
|
206 | expect(result21.value).toEqual(rangeSize);
|
207 | expect(result22.value).toEqual(tailSize);
|
208 | expect(isStreamingSupported1).toEqual(false);
|
209 | expect(isRangeSupported1).toEqual(true);
|
210 | expect(fullReaderCancelled1).toEqual(true);
|
211 | expect(isStreamingSupported2).toEqual(false);
|
212 | expect(isRangeSupported2).toEqual(true);
|
213 | expect(fullReaderCancelled2).toEqual(true);
|
214 | done();
|
215 | }).catch(function (reason) {
|
216 | done.fail(reason);
|
217 | });
|
218 | });
|
219 | }); |
\ | No newline at end of file |