Quelle TestStringStream.cpp Sprache: 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/. */

#include "gtest/gtest.h"
#include "Helpers.h"
#include "mozilla/gtest/MozAssertions.h"
#include "nsICloneableInputStream.h"
#include "nsStringStream.h"
#include "nsTArray.h"
#include "nsIInputStream.h"
#include "nsCOMPtr.h"
#include "nsStreamUtils.h"
#include "mozilla/Span.h"
#include "nsISeekableStream.h"

namespace {

static void TestStringStream(uint32_t aNumBytes) {
  nsTArray<char> inputData;
  testing::CreateData(aNumBytes, inputData);
  nsDependentCSubstring inputString(inputData.Elements(), inputData.Length());

  nsCOMPtr<nsIInputStream> stream;
  nsresult rv = NS_NewCStringInputStream(getter_AddRefs(stream), inputString);
  ASSERT_NS_SUCCEEDED(rv);

  testing::ConsumeAndValidateStream(stream, inputString);
}

static void TestStringStreamClone(uint32_t aNumBytes) {
  nsTArray<char> inputData;
  testing::CreateData(aNumBytes, inputData);
  nsDependentCSubstring inputString(inputData.Elements(), inputData.Length());

  nsCOMPtr<nsIInputStream> stream;
  nsresult rv = NS_NewCStringInputStream(getter_AddRefs(stream), inputString);
  ASSERT_NS_SUCCEEDED(rv);

  nsCOMPtr<nsICloneableInputStream> cloneable = do_QueryInterface(stream);
  ASSERT_TRUE(cloneable != nullptr);
  ASSERT_TRUE(cloneable->GetCloneable());

  nsCOMPtr<nsIInputStream> clone;
  rv = cloneable->Clone(getter_AddRefs(clone));

  testing::ConsumeAndValidateStream(stream, inputString);

  // Release the stream to verify that the clone's string survives correctly.
  stream = nullptr;

  testing::ConsumeAndValidateStream(clone, inputString);
}

}  // namespace

TEST(StringStream, Simple_4k)
{ TestStringStream(1024 * 4); }

TEST(StringStream, Clone_4k)
{ TestStringStreamClone(1024 * 4); }

static nsresult CloseStreamThenRead(nsIInputStream* aInStr, void* aClosure,
                                    const char* aBuffer, uint32_t aOffset,
                                    uint32_t aCount, uint32_t* aCountWritten) {
  // Closing the stream will free the data
  nsresult rv = aInStr->Close();
  if (NS_FAILED(rv)) {
    return rv;
  }
  // This will likely be allocated in the same slot as what we have in aBuffer
  char* newAlloc = moz_xstrdup("abcd");

  char* toBuf = static_cast<char*>(aClosure);
  memcpy(&toBuf[aOffset], aBuffer, aCount);
  *aCountWritten = aCount;
  free(newAlloc);
  return NS_OK;
}

TEST(StringStream, CancelInReadSegments)
{
  char* buffer = moz_xstrdup("test");
  nsCOMPtr<nsIInputStream> stream;
  nsresult rv = NS_NewByteInputStream(
      getter_AddRefs(stream), mozilla::Span(buffer, 5), NS_ASSIGNMENT_ADOPT);
  ASSERT_NS_SUCCEEDED(rv);

  char buf[100];
  uint32_t count = 0;
  uint64_t available = 0;
  rv = stream->Available(&available);
  ASSERT_NS_SUCCEEDED(rv);
  rv = stream->ReadSegments(CloseStreamThenRead, buf, available, &count);
  ASSERT_NS_SUCCEEDED(rv);
  ASSERT_TRUE(count == 5);
  ASSERT_TRUE(!strcmp(buf, "test"));
}

quality90%

¤ Dauer der Verarbeitung: 0.12 Sekunden (vorverarbeitet) ¤

Wurzel

Suchen

Beweissystem der NASA

Beweissystem Isabelle

NIST Cobol Testsuite

Cephes Mathematical Library

Wiener Entwicklungsmethode

Haftungshinweis

Die Informationen auf dieser Webseite wurden nach bestem Wissen sorgfältig zusammengestellt. Es wird jedoch weder Vollständigkeit, noch Richtigkeit, noch Qualität der bereit gestellten Informationen zugesichert.

Bemerkung:

Die farbliche Syntaxdarstellung ist noch experimentell.