gecko-dev/xpcom/tests/TestStreamUtils.cpp
Chris Peterson 3f655a7340 Bug 1534878 - xpcom: Rename NS_InitXPCOM2() to NS_InitXPCOM(). r=froydnj
--HG--
extra : rebase_source : 6e7a46cf49f78e46e12d1e7fc76aba6f0c377be0
2019-03-14 23:38:09 -07:00

65 lines
1.6 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include <stdlib.h>
#include <stdio.h>
#include "nsIPipe.h"
#include "nsStreamUtils.h"
#include "nsString.h"
#include "nsCOMPtr.h"
//----
static bool test_consume_stream() {
const char kData[] =
"Get your facts first, and then you can distort them as much as you "
"please.";
nsCOMPtr<nsIInputStream> input;
nsCOMPtr<nsIOutputStream> output;
NS_NewPipe(getter_AddRefs(input), getter_AddRefs(output), 10, UINT32_MAX);
if (!input || !output) return false;
uint32_t n = 0;
output->Write(kData, sizeof(kData) - 1, &n);
if (n != (sizeof(kData) - 1)) return false;
output = nullptr; // close output
nsCString buf;
if (NS_FAILED(NS_ConsumeStream(input, UINT32_MAX, buf))) return false;
if (!buf.Equals(kData)) return false;
return true;
}
//----
typedef bool (*TestFunc)();
#define DECL_TEST(name) \
{ #name, name }
static const struct Test {
const char* name;
TestFunc func;
} tests[] = {DECL_TEST(test_consume_stream), {nullptr, nullptr}};
int main(int argc, char** argv) {
int count = 1;
if (argc > 1) count = atoi(argv[1]);
if (NS_FAILED(NS_InitXPCOM(nullptr, nullptr, nullptr))) return -1;
while (count--) {
for (const Test* t = tests; t->name != nullptr; ++t) {
printf("%25s : %s\n", t->name, t->func() ? "SUCCESS" : "FAILURE");
}
}
NS_ShutdownXPCOM(nullptr);
return 0;
}