Bug 401781 - "Add tests for nsI*Stream" [p=WeirdAl r=bsmedberg]

/* ***** BEGIN LICENSE BLOCK *****

 * Version: MPL 1.1/GPL 2.0/LGPL 2.1

 *

 * The contents of this file are subject to the Mozilla Public License Version

 * 1.1 (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.mozilla.org/MPL/

 *

 * Software distributed under the License is distributed on an "AS IS" basis,

 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License

 * for the specific language governing rights and limitations under the

 * License.

 *

 * The Original Code is XPCOM Input/Output Stream tests.

 *

 * The Initial Developer of the Original Code is

 * Alexander J. Vincent.

 * Portions created by the Initial Developer are Copyright (C) 2007

 * the Initial Developer. All Rights Reserved.

 *

 * Contributor(s):

 *

 * Alternatively, the contents of this file may be used under the terms of

 * either the GNU General Public License Version 2 or later (the "GPL"), or

 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),

 * in which case the provisions of the GPL or the LGPL are applicable instead

 * of those above. If you wish to allow use of your version of this file only

 * under the terms of either the GPL or the LGPL, and not to allow others to

 * use your version of this file under the terms of the MPL, indicate your

 * decision by deleting the provisions above and replace them with the notice

 * and other provisions required by the GPL or the LGPL. If you do not delete

 * the provisions above, a recipient may use your version of this file under

 * the terms of any one of the MPL, the GPL or the LGPL.

 *

 * ***** END LICENSE BLOCK ***** */

const Ci = Components.interfaces;

const Cr = Components.results;

const CC = Components.Constructor;

var Pipe = CC("@mozilla.org/pipe;1",

              "nsIPipe",

              "init");

var BinaryOutput = CC("@mozilla.org/binaryoutputstream;1",

                      "nsIBinaryOutputStream",

                      "setOutputStream");

var BinaryInput = CC("@mozilla.org/binaryinputstream;1",

                     "nsIBinaryInputStream",

                     "setInputStream");

/**

 * Binary stream tests.

 */

function test_binary_streams() {

  var p, is, os;

  p = new Pipe(false, false, 1024, 1, null);

  is = new BinaryInput(p.inputStream);

  os = new BinaryOutput(p.outputStream);

  const LargeNum = Math.pow(2, 18) + Math.pow(2, 12) + 1;

  const HugeNum = Math.pow(2, 62);

  const HelloStr = "Hello World";

  const HelloArray = Array.map(HelloStr, function(c) {return c.charCodeAt(0)});

  var countObj = {};

  var msg = {};

  // Test reading immediately after writing.

  os.writeBoolean(true);

  do_check_eq(is.readBoolean(), true);

  os.write8(4);

  do_check_eq(is.read8(), 4);

  os.write16(4);

  do_check_eq(is.read16(), 4);

  os.write16(1024);

  do_check_eq(is.read16(), 1024);

  os.write32(7);

  do_check_eq(is.read32(), 7);

  os.write32(LargeNum);

  do_check_eq(is.read32(), LargeNum);

  os.write64(LargeNum);

  do_check_eq(is.read64(), LargeNum);

  os.write64(1024);

  do_check_eq(is.read64(), 1024);

  os.write64(HugeNum);

  do_check_eq(is.read64(), HugeNum);

  os.writeFloat(2.5);

  do_check_eq(is.readFloat(), 2.5);

  os.writeDouble(Math.SQRT2);

  do_check_eq(is.readDouble(), Math.SQRT2);

  os.writeStringZ("Mozilla");

  do_check_eq(is.readCString(), "Mozilla");

  os.writeWStringZ("Gecko");

  do_check_eq(is.readString(), "Gecko");

  os.writeBytes(HelloStr, HelloStr.length);

  do_check_eq(is.available(), HelloStr.length);

  msg = is.readBytes(HelloStr.length);

  do_check_eq(msg, HelloStr);

  msg = null;

  countObj.value = -1;

  os.writeByteArray(HelloArray, HelloArray.length);

  do_check_eq(is.available(), HelloStr.length);

  msg = is.readByteArray(HelloStr.length)

  do_check_eq(typeof msg, typeof HelloArray);

  do_check_eq(msg.toSource(), HelloArray.toSource());

  do_check_eq(is.available(), 0);

  // Test writing in one big chunk.

  os.writeBoolean(true);

  os.write8(4);

  os.write16(4);

  os.write16(1024);

  os.write32(7);

  os.write32(LargeNum);

  os.write64(LargeNum);

  os.write64(1024);

  os.write64(HugeNum);

  os.writeFloat(2.5);

  os.writeDouble(Math.SQRT2);

  os.writeStringZ("Mozilla");

  os.writeWStringZ("Gecko");

  os.writeBytes(HelloStr, HelloStr.length);

  os.writeByteArray(HelloArray, HelloArray.length);

  // Available should not be zero after a long write like this.

  do_check_neq(is.available(), 0);

  // Test reading in one big chunk.

  do_check_eq(is.readBoolean(), true);

  do_check_eq(is.read8(), 4);

  do_check_eq(is.read16(), 4);

  do_check_eq(is.read16(), 1024);

  do_check_eq(is.read32(), 7);

  do_check_eq(is.read32(), LargeNum);

  do_check_eq(is.read64(), LargeNum);

  do_check_eq(is.read64(), 1024);

  do_check_eq(is.read64(), HugeNum);

  do_check_eq(is.readFloat(), 2.5);

  do_check_eq(is.readDouble(), Math.SQRT2);

  do_check_eq(is.readCString(), "Mozilla");

  do_check_eq(is.readString(), "Gecko");

  // Remember, we wrote HelloStr twice - once as a string, and then as an array.

  do_check_eq(is.available(), HelloStr.length * 2);

  msg = is.readBytes(HelloStr.length);

  do_check_eq(msg, HelloStr);

  msg = null;

  countObj.value = -1;

  do_check_eq(is.available(), HelloStr.length);

  msg = is.readByteArray(HelloStr.length)

  do_check_eq(typeof msg, typeof HelloArray);

  do_check_eq(msg.toSource(), HelloArray.toSource());

  do_check_eq(is.available(), 0);

  // Test for invalid actions.

  os.close();

  is.close();

  try {

    os.writeBoolean(false);

    do_throw("Not reached!");

  } catch (e if (e instanceof Ci.nsIException &&

                 e.result == Cr.NS_BASE_STREAM_CLOSED)) {

    // do nothing

  }

  try {

    is.available();

    do_throw("Not reached!");

  } catch (e if (e instanceof Ci.nsIException &&

                 e.result == Cr.NS_BASE_STREAM_CLOSED)) {

    // do nothing

  }

  try {

    is.readBoolean();

    do_throw("Not reached!");

  } catch (e if (e instanceof Ci.nsIException &&

                 e.result == Cr.NS_ERROR_FAILURE)) {

    // do nothing

  }

}

function run_test() {

  test_binary_streams();

}