mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-06-08 05:29:36 +00:00
134 lines
3.8 KiB
C++
134 lines
3.8 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* 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/. */
|
|
|
|
/* Helper routines for xpcom gtests. */
|
|
|
|
#include "Helpers.h"
|
|
|
|
#include <algorithm>
|
|
#include "gtest/gtest.h"
|
|
#include "nsIOutputStream.h"
|
|
#include "nsStreamUtils.h"
|
|
#include "nsTArray.h"
|
|
|
|
namespace testing {
|
|
|
|
// Populate an array with the given number of bytes. Data is lorem ipsum
|
|
// random text, but deterministic across multiple calls.
|
|
void
|
|
CreateData(uint32_t aNumBytes, nsTArray<char>& aDataOut)
|
|
{
|
|
static const char data[] =
|
|
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Donec egestas "
|
|
"purus eu condimentum iaculis. In accumsan leo eget odio porttitor, non "
|
|
"rhoncus nulla vestibulum. Etiam lacinia consectetur nisl nec "
|
|
"sollicitudin. Sed fringilla accumsan diam, pulvinar varius massa. Duis "
|
|
"mollis dignissim felis, eget tempus nisi tristique ut. Fusce euismod, "
|
|
"lectus non lacinia tempor, tellus diam suscipit quam, eget hendrerit "
|
|
"lacus nunc fringilla ante. Sed ultrices massa vitae risus molestie, ut "
|
|
"finibus quam laoreet nullam.";
|
|
static const uint32_t dataLength = sizeof(data) - 1;
|
|
|
|
aDataOut.SetCapacity(aNumBytes);
|
|
|
|
while (aNumBytes > 0) {
|
|
uint32_t amount = std::min(dataLength, aNumBytes);
|
|
aDataOut.AppendElements(data, amount);
|
|
aNumBytes -= amount;
|
|
}
|
|
}
|
|
|
|
// Write the given number of bytes out to the stream. Loop until expected
|
|
// bytes count is reached or an error occurs.
|
|
void
|
|
Write(nsIOutputStream* aStream, const nsTArray<char>& aData, uint32_t aOffset,
|
|
uint32_t aNumBytes)
|
|
{
|
|
uint32_t remaining =
|
|
std::min(aNumBytes, static_cast<uint32_t>(aData.Length() - aOffset));
|
|
|
|
while (remaining > 0) {
|
|
uint32_t numWritten;
|
|
nsresult rv = aStream->Write(aData.Elements() + aOffset, remaining,
|
|
&numWritten);
|
|
ASSERT_TRUE(NS_SUCCEEDED(rv));
|
|
if (numWritten < 1) {
|
|
break;
|
|
}
|
|
aOffset += numWritten;
|
|
remaining -= numWritten;
|
|
}
|
|
}
|
|
|
|
// Write the given number of bytes and then close the stream.
|
|
void
|
|
WriteAllAndClose(nsIOutputStream* aStream, const nsTArray<char>& aData)
|
|
{
|
|
Write(aStream, aData, 0, aData.Length());
|
|
aStream->Close();
|
|
}
|
|
|
|
// Synchronously consume the given input stream and validate the resulting data
|
|
// against the given array of expected values.
|
|
void
|
|
ConsumeAndValidateStream(nsIInputStream* aStream,
|
|
const nsTArray<char>& aExpectedData)
|
|
{
|
|
nsDependentCSubstring data(aExpectedData.Elements(), aExpectedData.Length());
|
|
ConsumeAndValidateStream(aStream, data);
|
|
}
|
|
|
|
// Synchronously consume the given input stream and validate the resulting data
|
|
// against the given string of expected values.
|
|
void
|
|
ConsumeAndValidateStream(nsIInputStream* aStream,
|
|
const nsACString& aExpectedData)
|
|
{
|
|
nsAutoCString outputData;
|
|
nsresult rv = NS_ConsumeStream(aStream, UINT32_MAX, outputData);
|
|
ASSERT_TRUE(NS_SUCCEEDED(rv));
|
|
ASSERT_EQ(aExpectedData.Length(), outputData.Length());
|
|
ASSERT_TRUE(aExpectedData.Equals(outputData));
|
|
}
|
|
|
|
NS_IMPL_ISUPPORTS(OutputStreamCallback, nsIOutputStreamCallback);
|
|
|
|
OutputStreamCallback::OutputStreamCallback()
|
|
: mCalled(false)
|
|
{
|
|
}
|
|
|
|
OutputStreamCallback::~OutputStreamCallback()
|
|
{
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
OutputStreamCallback::OnOutputStreamReady(nsIAsyncOutputStream* aStream)
|
|
{
|
|
mCalled = true;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMPL_ISUPPORTS(InputStreamCallback, nsIInputStreamCallback);
|
|
|
|
InputStreamCallback::InputStreamCallback()
|
|
: mCalled(false)
|
|
{
|
|
}
|
|
|
|
InputStreamCallback::~InputStreamCallback()
|
|
{
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
InputStreamCallback::OnInputStreamReady(nsIAsyncInputStream* aStream)
|
|
{
|
|
mCalled = true;
|
|
return NS_OK;
|
|
}
|
|
|
|
} // namespace testing
|