mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:02:23 +01:00
Jon Ross-PerkinsandGeoff Romer
4c4c4a4d2c
This adds a RawStringOstream. Versus TestRawOstream, which is
consolidated over to RawStringOstream, it uses a string for storage
instead of a vector, mainly to support move-to-string semantics. Versus
llvm::raw_string_ostream, it owns the string and supports pwrite (which
is needed for driver and its fd_ostream compatibility requirement).
This converts most uses of llvm::raw_string_ostream, leaving behind a
few in InstNamer that explicitly cannot own the string, such as:
```
llvm::raw_string_ostream(name)
<< "_" << tree.tokens().GetColumnNumber(token);
```
I have this as its own library so that it can use CHECK.
Yes this doesn't save much code, but it's code we repeatedly write.
---------
Co-authored-by: Geoff Romer <gromer@google.com>
72 lines
1.6 KiB
C++
72 lines
1.6 KiB
C++
// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
// Exceptions. See /LICENSE for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
#include "common/raw_string_ostream.h"
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace Carbon::Testing {
|
|
namespace {
|
|
|
|
using ::testing::HasSubstr;
|
|
using ::testing::StrEq;
|
|
|
|
TEST(RawStringOstreamTest, Basics) {
|
|
RawStringOstream os;
|
|
|
|
os << "test 1";
|
|
EXPECT_THAT(os.TakeStr(), StrEq("test 1"));
|
|
|
|
os << "test"
|
|
<< " "
|
|
<< "2";
|
|
EXPECT_THAT(os.TakeStr(), StrEq("test 2"));
|
|
|
|
os << "test";
|
|
os << " ";
|
|
os << "3";
|
|
EXPECT_THAT(os.TakeStr(), StrEq("test 3"));
|
|
}
|
|
|
|
TEST(RawStringOstreamTest, MultipleStreams) {
|
|
RawStringOstream os1;
|
|
RawStringOstream os2;
|
|
|
|
os1 << "test ";
|
|
os2 << "test stream 2";
|
|
os1 << "stream 1";
|
|
EXPECT_THAT(os1.TakeStr(), StrEq("test stream 1"));
|
|
EXPECT_THAT(os2.TakeStr(), StrEq("test stream 2"));
|
|
}
|
|
|
|
TEST(RawStringOstreamTest, MultipleLines) {
|
|
RawStringOstream os;
|
|
|
|
os << "test line 1\n";
|
|
os << "test line 2\n";
|
|
os << "test line 3\n";
|
|
EXPECT_THAT(os.TakeStr(), StrEq("test line 1\ntest line 2\ntest line 3\n"));
|
|
}
|
|
|
|
TEST(RawStringOstreamTest, Substring) {
|
|
RawStringOstream os;
|
|
|
|
os << "test line 1\n";
|
|
os << "test line 2\n";
|
|
os << "test line 3\n";
|
|
EXPECT_THAT(os.TakeStr(), HasSubstr("test line 2"));
|
|
}
|
|
|
|
TEST(RawStringOstreamTest, Pwrite) {
|
|
RawStringOstream os;
|
|
os << "test line 1\n";
|
|
os.pwrite("splat", 5, 1);
|
|
os << "test line 2\n";
|
|
EXPECT_THAT(os.TakeStr(), HasSubstr("tsplatine 1\ntest line 2\n"));
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace Carbon::Testing
|