Files
swift-mirror/unittests/Frontend/ArgParsingTest.h
2025-03-05 15:21:01 +00:00

56 lines
1.4 KiB
C++

//===-- ArgParsingTest.h ----------------------------------------*- C++ -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2025 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#ifndef ARG_PARSING_TEST_H
#define ARG_PARSING_TEST_H
#include "swift/Frontend/Frontend.h"
#include "gtest/gtest.h"
using Args = std::vector<std::string>;
class ArgParsingTest : public ::testing::Test {
swift::CompilerInvocation invocation;
swift::SourceManager sourceMgr;
swift::DiagnosticEngine diags;
protected:
std::optional<std::string> langMode;
public:
ArgParsingTest();
void parseArgs(const Args &args);
swift::LangOptions &getLangOptions();
};
template <typename T>
struct ArgParsingTestCase final {
Args args;
T expectedResult;
ArgParsingTestCase(Args args, T expectedResult)
: args(std::move(args)), expectedResult(std::move(expectedResult)) {}
};
// MARK: - Printers
void PrintTo(const Args &, std::ostream *);
template <typename T>
void PrintTo(const ArgParsingTestCase<T> &value, std::ostream *os) {
PrintTo(value.args, os);
}
#endif // ARG_PARSING_TEST_H