xpcom/tests/TestStreamUtils.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent: */
   3 /* ***** BEGIN LICENSE BLOCK *****
   4  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version
   7  * 1.1 (the "License"); you may not use this file except in compliance with
   8  * the License. You may obtain a copy of the License at
   9  * http://www.mozilla.org/MPL/
  10  *
  11  * Software distributed under the License is distributed on an "AS IS" basis,
  12  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13  * for the specific language governing rights and limitations under the
  14  * License.
  15  *
  16  * The Original Code is C++ array template tests.
  17  *
  18  * The Initial Developer of the Original Code is Google Inc.
  19  * Portions created by the Initial Developer are Copyright (C) 2005
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *  Darin Fisher <darin@meer.net>
  24  *
  25  * Alternatively, the contents of this file may be used under the terms of
  26  * either the GNU General Public License Version 2 or later (the "GPL"), or
  27  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  28  * in which case the provisions of the GPL or the LGPL are applicable instead
  29  * of those above. If you wish to allow use of your version of this file only
  30  * under the terms of either the GPL or the LGPL, and not to allow others to
  31  * use your version of this file under the terms of the MPL, indicate your
  32  * decision by deleting the provisions above and replace them with the notice
  33  * and other provisions required by the GPL or the LGPL. If you do not delete
  34  * the provisions above, a recipient may use your version of this file under
  35  * the terms of any one of the MPL, the GPL or the LGPL.
  36  *
  37  * ***** END LICENSE BLOCK ***** */
  38
  39 #include <stdlib.h>
  40 #include <stdio.h>
  41 #include "nsIPipe.h"
  42 #include "nsStreamUtils.h"
  43 #include "nsString.h"
  44 #include "nsCOMPtr.h"
  45
  46 //----
  47
  48 static PRBool test_consume_stream() {
  49   const char kData[] =
  50       "Get your facts first, and then you can distort them as much as you "
  51       "please.";
  52
  53   nsCOMPtr<nsIInputStream> input;
  54   nsCOMPtr<nsIOutputStream> output;
  55   NS_NewPipe(getter_AddRefs(input),
  56              getter_AddRefs(output),
  57              10, PR_UINT32_MAX);
  58   if (!input || !output)
  59     return PR_FALSE;
  60
  61   PRUint32 n = 0;
  62   output->Write(kData, sizeof(kData) - 1, &n);
  63   if (n != (sizeof(kData) - 1))
  64     return PR_FALSE;
  65   output = nsnull;  // close output
  66
  67   nsCString buf;
  68   if (NS_FAILED(NS_ConsumeStream(input, PR_UINT32_MAX, buf)))
  69     return PR_FALSE;
  70
  71   if (!buf.Equals(kData))
  72     return PR_FALSE;
  73
  74   return PR_TRUE;
  75 }
  76
  77 //----
  78
  79 typedef PRBool (*TestFunc)();
  80 #define DECL_TEST(name) { #name, name }
  81
  82 static const struct Test {
  83   const char* name;
  84   TestFunc    func;
  85 } tests[] = {
  86   DECL_TEST(test_consume_stream),
  87   { nsnull, nsnull }
  88 };
  89
  90 int main(int argc, char **argv) {
  91   int count = 1;
  92   if (argc > 1)
  93     count = atoi(argv[1]);
  94
  95   if (NS_FAILED(NS_InitXPCOM2(nsnull, nsnull, nsnull)))
  96     return -1;
  97
  98   while (count--) {
  99     for (const Test* t = tests; t->name != nsnull; ++t) {
 100       printf("%25s : %s\n", t->name, t->func() ? "SUCCESS" : "FAILURE");
 101     }
 102   }
 103
 104   NS_ShutdownXPCOM(nsnull);
 105   return 0;
 106 }