michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: const Ci = Components.interfaces;
michael@0: const Cr = Components.results;
michael@0: const CC = Components.Constructor;
michael@0: const Cc = Components.classes;
michael@0: 
michael@0: // The string we use as data.
michael@0: const data = "0123456789";
michael@0: // Number of streams in the multiplex stream.
michael@0: const count = 10;
michael@0: 
michael@0: function test_multiplex_streams() {
michael@0:   var MultiplexStream = CC("@mozilla.org/io/multiplex-input-stream;1",
michael@0:                            "nsIMultiplexInputStream");
michael@0:   do_check_eq(1, 1);
michael@0: 
michael@0:   var BinaryInputStream = Components.Constructor("@mozilla.org/binaryinputstream;1",
michael@0:                                                  "nsIBinaryInputStream");
michael@0:   var BinaryOutputStream = Components.Constructor("@mozilla.org/binaryoutputstream;1",
michael@0:                                                   "nsIBinaryOutputStream",
michael@0:                                                   "setOutputStream");
michael@0:   var multiplex = new MultiplexStream();
michael@0:   for (var i = 0; i < count; ++i) {
michael@0:     let s = Cc["@mozilla.org/io/string-input-stream;1"]
michael@0:               .createInstance(Ci.nsIStringInputStream);
michael@0:     s.setData(data, data.length);
michael@0: 
michael@0:     multiplex.appendStream(s);
michael@0:   }
michael@0:   var seekable = multiplex.QueryInterface(Ci.nsISeekableStream);
michael@0:   var sis = Cc["@mozilla.org/scriptableinputstream;1"]
michael@0:               .createInstance(Ci.nsIScriptableInputStream);
michael@0:   sis.init(seekable);
michael@0:   // Read some data.
michael@0:   var readData = sis.read(20);
michael@0:   do_check_eq(readData, data + data);
michael@0:   // -- Tests for NS_SEEK_SET
michael@0:   // Seek to a non-zero, non-stream-boundary offset.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 2);
michael@0:   do_check_eq(seekable.tell(), 2);
michael@0:   do_check_eq(seekable.available(), 98);
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 9);
michael@0:   do_check_eq(seekable.tell(), 9);
michael@0:   do_check_eq(seekable.available(), 91);
michael@0:   // Seek across stream boundary.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 35);
michael@0:   do_check_eq(seekable.tell(), 35);
michael@0:   do_check_eq(seekable.available(), 65);
michael@0:   readData = sis.read(5);
michael@0:   do_check_eq(readData, data.slice(5));
michael@0:   do_check_eq(seekable.available(), 60);
michael@0:   // Seek at stream boundaries.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 40);
michael@0:   do_check_eq(seekable.tell(), 40);
michael@0:   do_check_eq(seekable.available(), 60);
michael@0:   readData = sis.read(10);
michael@0:   do_check_eq(readData, data);
michael@0:   do_check_eq(seekable.tell(), 50);
michael@0:   do_check_eq(seekable.available(), 50);
michael@0:   // Rewind and read across streams.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 39);
michael@0:   do_check_eq(seekable.tell(), 39);
michael@0:   do_check_eq(seekable.available(), 61);
michael@0:   readData = sis.read(11);
michael@0:   do_check_eq(readData, data.slice(9) + data);
michael@0:   do_check_eq(seekable.tell(), 50);
michael@0:   do_check_eq(seekable.available(), 50);
michael@0:   // Rewind to the beginning.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 0);
michael@0:   do_check_eq(seekable.tell(), 0);
michael@0:   do_check_eq(seekable.available(), 100);
michael@0:   // Seek to some random location
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 50);
michael@0:   // -- Tests for NS_SEEK_CUR
michael@0:   // Positive seek.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_CUR, 15);
michael@0:   do_check_eq(seekable.tell(), 65);
michael@0:   do_check_eq(seekable.available(), 35);
michael@0:   readData = sis.read(10);
michael@0:   do_check_eq(readData, data.slice(5) + data.slice(0, 5));
michael@0:   do_check_eq(seekable.tell(), 75);
michael@0:   do_check_eq(seekable.available(), 25);
michael@0:   // Negative seek.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_CUR, -15);
michael@0:   do_check_eq(seekable.tell(), 60);
michael@0:   do_check_eq(seekable.available(), 40);
michael@0:   readData = sis.read(10);
michael@0:   do_check_eq(readData, data);
michael@0:   do_check_eq(seekable.tell(), 70);
michael@0:   do_check_eq(seekable.available(), 30);
michael@0: 
michael@0:   // -- Tests for NS_SEEK_END
michael@0:   // Normal read.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_END, -5);
michael@0:   do_check_eq(seekable.tell(), data.length * count - 5);
michael@0:   readData = sis.read(5);
michael@0:   do_check_eq(readData, data.slice(5));
michael@0:   do_check_eq(seekable.tell(), data.length * count);
michael@0:   // Read across streams.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_END, -15);
michael@0:   do_check_eq(seekable.tell(), data.length * count - 15);
michael@0:   readData = sis.read(15);
michael@0:   do_check_eq(readData, data.slice(5) + data);
michael@0:   do_check_eq(seekable.tell(), data.length * count);
michael@0: 
michael@0:   // -- Try to do various edge cases
michael@0:   // Forward seek from the end, should throw.
michael@0:   var caught = false;
michael@0:   try {
michael@0:     seekable.seek(Ci.nsISeekableStream.NS_SEEK_END, 15);
michael@0:   } catch(e) {
michael@0:     caught = true;
michael@0:   }
michael@0:   do_check_eq(caught, true);
michael@0:   do_check_eq(seekable.tell(), data.length * count);
michael@0:   // Backward seek from the beginning, should be clamped.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 0);
michael@0:   do_check_eq(seekable.tell(), 0);
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_CUR, -15);
michael@0:   do_check_eq(seekable.tell(), 0);
michael@0:   // Seek too far: should be clamped.
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 0);
michael@0:   do_check_eq(seekable.tell(), 0);
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_CUR, 3 * data.length * count);
michael@0:   do_check_eq(seekable.tell(), 100);
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, data.length * count);
michael@0:   do_check_eq(seekable.tell(), 100);
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_CUR, -2 * data.length * count);
michael@0:   do_check_eq(seekable.tell(), 0);
michael@0: }
michael@0: 
michael@0: function test_multiplex_bug797871() {
michael@0: 
michael@0:   var data = "1234567890123456789012345678901234567890";
michael@0: 
michael@0:   var MultiplexStream = CC("@mozilla.org/io/multiplex-input-stream;1",
michael@0:                            "nsIMultiplexInputStream");
michael@0:   do_check_eq(1, 1);
michael@0: 
michael@0:   var BinaryInputStream = Components.Constructor("@mozilla.org/binaryinputstream;1",
michael@0:                                                  "nsIBinaryInputStream");
michael@0:   var BinaryOutputStream = Components.Constructor("@mozilla.org/binaryoutputstream;1",
michael@0:                                                   "nsIBinaryOutputStream",
michael@0:                                                   "setOutputStream");
michael@0:   var multiplex = new MultiplexStream();
michael@0:   let s = Cc["@mozilla.org/io/string-input-stream;1"]
michael@0:             .createInstance(Ci.nsIStringInputStream);
michael@0:   s.setData(data, data.length);
michael@0: 
michael@0:   multiplex.appendStream(s);
michael@0: 
michael@0:   var seekable = multiplex.QueryInterface(Ci.nsISeekableStream);
michael@0:   var sis = Cc["@mozilla.org/scriptableinputstream;1"]
michael@0:               .createInstance(Ci.nsIScriptableInputStream);
michael@0:   sis.init(seekable);
michael@0: 
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 8);
michael@0:   do_check_eq(seekable.tell(), 8);
michael@0:   readData = sis.read(2);
michael@0:   do_check_eq(seekable.tell(), 10);
michael@0: 
michael@0:   seekable.seek(Ci.nsISeekableStream.NS_SEEK_SET, 20);
michael@0:   do_check_eq(seekable.tell(), 20);
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   test_multiplex_streams();
michael@0:   test_multiplex_bug797871();
michael@0: }
michael@0: