carbon-language/carbon-lang
1
0
Fork 0
mirror of https://github.com/carbon-language/carbon-lang.git synced 2026-10-06 09:24:46 +01:00
Code Issues Packages Projects Releases 10 Wiki Activity

Extract a test helper to its own library. (#2828)

Browse Source 
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.
This commit is contained in:
Chandler Carruth
2023-05-18 08:03:57 -07:00
committed by GitHub
parent e42b48ecc5
commit 3c15882f4e
5 changed files with 138 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
toolchain/driver/driver_test.cpp
+13 -43
View File
@@ -7,6 +7,7 @@
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include "common/test_raw_ostream.h"
#include "llvm/ADT/SmallString.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/SourceMgr.h"
@@ -20,40 +21,9 @@ using ::testing::ElementsAre;
using ::testing::HasSubstr;
using ::testing::StrEq;
/// A raw_ostream that makes it easy to repeatedly check streamed output.
class RawTestOstream : public llvm::raw_ostream {
public:
~RawTestOstream() override {
flush();
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 {
flush();
std::string result = std::move(buffer_);
buffer_.clear();
return result;
}
private:
void write_impl(const char* ptr, size_t size) override {
buffer_.append(ptr, ptr + size);
}
[[nodiscard]] auto current_pos() const -> uint64_t override {
return buffer_.size();
}
std::string buffer_;
};
TEST(DriverTest, FullCommandErrors) {
RawTestOstream test_output_stream;
RawTestOstream test_error_stream;
TestRawOstream test_output_stream;
TestRawOstream test_error_stream;
Driver driver = Driver(test_output_stream, test_error_stream);
EXPECT_FALSE(driver.RunFullCommand({}));
@@ -67,8 +37,8 @@ TEST(DriverTest, FullCommandErrors) {
}
TEST(DriverTest, Help) {
RawTestOstream test_output_stream;
RawTestOstream test_error_stream;
TestRawOstream test_output_stream;
TestRawOstream test_error_stream;
Driver driver = Driver(test_output_stream, test_error_stream);
EXPECT_TRUE(driver.RunHelpSubcommand(ConsoleDiagnosticConsumer(), {}));
@@ -87,8 +57,8 @@ TEST(DriverTest, Help) {
}
TEST(DriverTest, HelpErrors) {
RawTestOstream test_output_stream;
RawTestOstream test_error_stream;
TestRawOstream test_output_stream;
TestRawOstream test_error_stream;
Driver driver = Driver(test_output_stream, test_error_stream);
EXPECT_FALSE(driver.RunHelpSubcommand(ConsoleDiagnosticConsumer(), {"foo"}));
@@ -122,8 +92,8 @@ auto CreateTestFile(llvm::StringRef text) -> std::string {
}
TEST(DriverTest, DumpTokens) {
RawTestOstream test_output_stream;
RawTestOstream test_error_stream;
TestRawOstream test_output_stream;
TestRawOstream test_error_stream;
Driver driver = Driver(test_output_stream, test_error_stream);
auto test_file_path = CreateTestFile("Hello World");
@@ -164,8 +134,8 @@ TEST(DriverTest, DumpTokens) {
}
TEST(DriverTest, DumpErrors) {
RawTestOstream test_output_stream;
RawTestOstream test_error_stream;
TestRawOstream test_output_stream;
TestRawOstream test_error_stream;
Driver driver = Driver(test_output_stream, test_error_stream);
EXPECT_FALSE(driver.RunDumpSubcommand(ConsoleDiagnosticConsumer(), {"foo"}));
@@ -189,8 +159,8 @@ TEST(DriverTest, DumpErrors) {
}
TEST(DriverTest, DumpParseTree) {
RawTestOstream test_output_stream;
RawTestOstream test_error_stream;
TestRawOstream test_output_stream;
TestRawOstream test_error_stream;
Driver driver = Driver(test_output_stream, test_error_stream);
auto test_file_path = CreateTestFile("var v: Int = 42;");