mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 21:30:12 +01:00
Chandler Carruth
3c15882f4e
This is a convenient test helper for anything that can use injected streams. Extract this to where it can be used by other tests and add some basic tests, mostly documenting how it works.
42 lines
1.0 KiB
C++
42 lines
1.0 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
|
|
|
|
#ifndef CARBON_COMMON_TEST_RAW_OSTREAM_H_
|
|
#define CARBON_COMMON_TEST_RAW_OSTREAM_H_
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <string>
|
|
|
|
#include "common/ostream.h"
|
|
|
|
namespace Carbon::Testing {
|
|
|
|
// A raw_ostream that makes it easy to repeatedly check streamed output.
|
|
class TestRawOstream : public llvm::raw_string_ostream {
|
|
public:
|
|
explicit TestRawOstream() : llvm::raw_string_ostream(buffer_) {}
|
|
|
|
~TestRawOstream() override {
|
|
if (!buffer_.empty()) {
|
|
ADD_FAILURE() << "Unchecked output:\n" << buffer_;
|
|
}
|
|
}
|
|
|
|
// Flushes the stream and returns the contents so far, clearing the stream
|
|
// back to empty.
|
|
auto TakeStr() -> std::string {
|
|
std::string result = std::move(buffer_);
|
|
buffer_.clear();
|
|
return result;
|
|
}
|
|
|
|
private:
|
|
std::string buffer_;
|
|
};
|
|
|
|
} // namespace Carbon::Testing
|
|
|
|
#endif // CARBON_COMMON_TEST_RAW_OSTREAM_H_
|