yanchengPowerSupply/node_modules/pdfjs-dist/lib/test/unit/network_spec.js

110 lines
3.6 KiB
JavaScript

/**
* @licstart The following is the entire license notice for the
* JavaScript code in this page
*
* Copyright 2022 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.
*
* @licend The above is the entire license notice for the
* JavaScript code in this page
*/
"use strict";
var _util = require("../../shared/util.js");
var _network = require("../../display/network.js");
describe("network", function () {
const pdf1 = new URL("../pdfs/tracemonkey.pdf", window.location).href;
const pdf1Length = 1016315;
it("read without stream and range", async function () {
const stream = new _network.PDFNetworkStream({
url: pdf1,
rangeChunkSize: 65536,
disableStream: true,
disableRange: true
});
const fullReader = stream.getFullReader();
let isStreamingSupported, isRangeSupported;
const promise = fullReader.headersReady.then(function () {
isStreamingSupported = fullReader.isStreamingSupported;
isRangeSupported = fullReader.isRangeSupported;
});
let len = 0,
count = 0;
const read = function () {
return fullReader.read().then(function (result) {
if (result.done) {
return undefined;
}
count++;
len += result.value.byteLength;
return read();
});
};
await Promise.all([read(), promise]);
expect(len).toEqual(pdf1Length);
expect(count).toEqual(1);
expect(isStreamingSupported).toEqual(false);
expect(isRangeSupported).toEqual(false);
});
it("read custom ranges", async function () {
const rangeSize = 32768;
const stream = new _network.PDFNetworkStream({
url: pdf1,
length: pdf1Length,
rangeChunkSize: rangeSize,
disableStream: true,
disableRange: false
});
const fullReader = stream.getFullReader();
let isStreamingSupported, isRangeSupported, fullReaderCancelled;
const promise = fullReader.headersReady.then(function () {
isStreamingSupported = fullReader.isStreamingSupported;
isRangeSupported = fullReader.isRangeSupported;
fullReader.cancel(new _util.AbortException("Don't need fullReader."));
fullReaderCancelled = true;
});
const tailSize = pdf1Length % rangeSize || rangeSize;
const range1Reader = stream.getRangeReader(pdf1Length - tailSize - rangeSize, pdf1Length - tailSize);
const range2Reader = stream.getRangeReader(pdf1Length - tailSize, pdf1Length);
const result1 = {
value: 0
},
result2 = {
value: 0
};
const read = function (reader, lenResult) {
return reader.read().then(function (result) {
if (result.done) {
return undefined;
}
lenResult.value += result.value.byteLength;
return read(reader, lenResult);
});
};
await Promise.all([read(range1Reader, result1), read(range2Reader, result2), promise]);
expect(result1.value).toEqual(rangeSize);
expect(result2.value).toEqual(tailSize);
expect(isStreamingSupported).toEqual(false);
expect(isRangeSupported).toEqual(true);
expect(fullReaderCancelled).toEqual(true);
});
});