gecko-dev/netwerk/test/unit/test_localstreams.js

88 строки
2.9 KiB
JavaScript

// Tests bug 304414
Cu.import("resource://gre/modules/NetUtil.jsm");
const PR_RDONLY = 0x1; // see prio.h
// 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);
// Check that the stream is not buffered
do_check_eq(Components.classes["@mozilla.org/io-util;1"]
.getService(Components.interfaces.nsIIOUtil)
.inputStreamIsBuffered(stream),
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 NetUtil.newChannel({
uri: uri,
loadUsingSystemPrincipal: true
}).open2();
}
function run_test() {
// Get a file and a directory in order to do some testing
var file = do_get_file("../unit/data/test_readline6.txt");
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
}