зеркало из https://github.com/mozilla/pjs.git
84 строки
2.7 KiB
JavaScript
84 строки
2.7 KiB
JavaScript
// Tests bug 304414
|
|
|
|
const PR_RDONLY = 0x1; // see prio.h
|
|
|
|
function getDir(key) {
|
|
var dirSvc = Components.classes["@mozilla.org/file/directory_service;1"]
|
|
.getService(Components.interfaces.nsIProperties);
|
|
return dirSvc.get(key, Components.interfaces.nsILocalFile);
|
|
}
|
|
|
|
// Does some sanity checks on the stream and returns the number of bytes read
|
|
// when the checks passed.
|
|
function test_stream(stream) {
|
|
// This test only handles blocking streams; that's desired for file streams
|
|
// anyway.
|
|
do_check_eq(stream.isNonBlocking(), false);
|
|
|
|
// Wrap it in a binary stream (to avoid wrong results that
|
|
// scriptablestream would produce with binary content)
|
|
var binstream = Components.classes['@mozilla.org/binaryinputstream;1']
|
|
.createInstance(Components.interfaces.nsIBinaryInputStream);
|
|
binstream.setInputStream(stream);
|
|
|
|
var numread = 0;
|
|
for (;;) {
|
|
do_check_eq(stream.available(), binstream.available());
|
|
var avail = stream.available();
|
|
do_check_neq(avail, -1);
|
|
|
|
// PR_UINT32_MAX and PR_INT32_MAX; the files we're testing with aren't that
|
|
// large.
|
|
do_check_neq(avail, Math.pow(2, 32) - 1);
|
|
do_check_neq(avail, Math.pow(2, 31) - 1);
|
|
|
|
if (!avail) {
|
|
// For blocking streams, available() only returns 0 on EOF
|
|
// Make sure that there is really no data left
|
|
var could_read = false;
|
|
try {
|
|
binstream.readByteArray(1);
|
|
could_read = true;
|
|
} catch (e) {
|
|
// We expect the exception, so do nothing here
|
|
}
|
|
if (could_read)
|
|
do_throw("Data readable when available indicated EOF!");
|
|
return numread;
|
|
}
|
|
|
|
dump("Trying to read " + avail + " bytes\n");
|
|
// Note: Verification that this does return as much bytes as we asked for is
|
|
// done in the binarystream implementation
|
|
var data = binstream.readByteArray(avail);
|
|
|
|
numread += avail;
|
|
}
|
|
return numread;
|
|
}
|
|
|
|
function stream_for_file(file) {
|
|
var str = Components.classes["@mozilla.org/network/file-input-stream;1"]
|
|
.createInstance(Components.interfaces.nsIFileInputStream);
|
|
str.init(file, PR_RDONLY, 0, 0);
|
|
return str;
|
|
}
|
|
|
|
function stream_from_channel(file) {
|
|
var ios = Components.classes["@mozilla.org/network/io-service;1"]
|
|
.getService(Components.interfaces.nsIIOService);
|
|
var uri = ios.newFileURI(file);
|
|
return ios.newChannelFromURI(uri).open();
|
|
}
|
|
|
|
function run_test() {
|
|
// Get a file and a directory in order to do some testing
|
|
var file = getDir("XpcomLib");
|
|
var len = file.fileSize;
|
|
do_check_eq(test_stream(stream_for_file(file)), len);
|
|
do_check_eq(test_stream(stream_from_channel(file)), len);
|
|
var dir = file.parent;
|
|
test_stream(stream_from_channel(dir)); // Can't do size checking
|
|
}
|
|
|