Files
carbon-lang/toolchain/lex/test_helpers.h
T
Jon Ross-Perkins 45ca3d28f5 Drop "diagnostic" from some filenames in the "diagnostics" folder (#6686)
Mainly because "sorting_diagnostic_consumer" is legacy, since
`SortingDiagnosticConsumer` became `SortingConsumer`. Also better
reflecting contents of these files.

Where I'm not renaming, I'm less positive about dropping "diagnostics"
from "file_diagnostics" and "null_diagnostics" (which contain both a
consumer and emitter, and "null.h" seems like poor naming), so not doing
that here. Also "diagnostic.h" contains `struct Diagnostic`, so is a
decent fit.

Assisted-by: Google Antigravity with Gemini 3 Flash
2026-02-04 17:24:55 +00:00

61 lines
2.2 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_TOOLCHAIN_LEX_TEST_HELPERS_H_
#define CARBON_TOOLCHAIN_LEX_TEST_HELPERS_H_
#include <gmock/gmock.h>
#include <array>
#include "common/check.h"
#include "common/string_helpers.h"
#include "toolchain/diagnostics/emitter.h"
namespace Carbon::Testing {
// A diagnostic converter for tests that lex a single token. Produces
// locations such as "`12.5`:1:3" to refer to the third character in the token.
class SingleTokenDiagnosticEmitter : public Diagnostics::Emitter<const char*> {
public:
// Form a converter for a given token. The string provided here must refer
// to the same character array that we are going to lex.
explicit SingleTokenDiagnosticEmitter(Diagnostics::Consumer* consumer,
llvm::StringRef token)
: Emitter(consumer), token_(token) {}
protected:
// Implements `DiagnosticConverter::ConvertLoc`.
auto ConvertLoc(const char* pos, ContextFnT /*context_fn*/) const
-> Diagnostics::ConvertedLoc override {
CARBON_CHECK(StringRefContainsPointer(token_, pos),
"invalid diagnostic location");
llvm::StringRef prefix = token_.take_front(pos - token_.begin());
auto [before_last_newline, this_line] = prefix.rsplit('\n');
if (before_last_newline.size() == prefix.size()) {
// On first line.
return {.loc = {.line_number = 1,
.column_number =
static_cast<int32_t>(pos - token_.begin() + 1)},
.last_byte_offset = -1};
} else {
// On second or subsequent lines. Note that the line number here is 2
// more than the number of newlines because `rsplit` removed one newline
// and `line_number` is 1-based.
return {
.loc = {.line_number =
static_cast<int32_t>(before_last_newline.count('\n') + 2),
.column_number = static_cast<int32_t>(this_line.size() + 1)},
.last_byte_offset = -1};
}
}
private:
llvm::StringRef token_;
};
} // namespace Carbon::Testing
#endif // CARBON_TOOLCHAIN_LEX_TEST_HELPERS_H_