1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
|
//===--- TestContext.cpp - Helper for setting up ASTContexts --------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 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
//
//===----------------------------------------------------------------------===//
#include "TestContext.h"
#include "swift/AST/GenericParamList.h"
#include "swift/AST/Module.h"
#include "swift/AST/ParseRequests.h"
#include "swift/Strings.h"
#include "swift/Subsystems.h"
using namespace swift;
using namespace swift::unittest;
static Decl *createOptionalType(ASTContext &ctx, SourceFile *fileForLookups,
Identifier name) {
auto *wrapped = GenericTypeParamDecl::createImplicit(
fileForLookups, ctx.getIdentifier("Wrapped"),
/*depth*/ 0, /*index*/ 0);
auto params = GenericParamList::create(ctx, SourceLoc(), wrapped,
SourceLoc());
auto decl = new (ctx) EnumDecl(SourceLoc(), name, SourceLoc(),
/*inherited*/{}, params, fileForLookups);
wrapped->setDeclContext(decl);
return decl;
}
TestContext::TestContext(ShouldDeclareOptionalTypes optionals)
: Ctx(*ASTContext::get(LangOpts, TypeCheckerOpts, SILOpts, SearchPathOpts,
ClangImporterOpts, SymbolGraphOpts, CASOpts,
SourceMgr, Diags)) {
registerParseRequestFunctions(Ctx.evaluator);
registerTypeCheckerRequestFunctions(Ctx.evaluator);
registerClangImporterRequestFunctions(Ctx.evaluator);
auto stdlibID = Ctx.getIdentifier(STDLIB_NAME);
auto *module = ModuleDecl::create(stdlibID, Ctx);
Ctx.addLoadedModule(module);
FileForLookups = new (Ctx) SourceFile(*module, SourceFileKind::Library,
/*buffer*/ std::nullopt);
module->addFile(*FileForLookups);
if (optionals == DeclareOptionalTypes) {
SmallVector<ASTNode, 2> optionalTypes;
optionalTypes.push_back(createOptionalType(
Ctx, FileForLookups, Ctx.getIdentifier("Optional")));
optionalTypes.push_back(createOptionalType(
Ctx, FileForLookups, Ctx.getIdentifier("ImplicitlyUnwrappedOptional")));
auto result = SourceFileParsingResult{Ctx.AllocateCopy(optionalTypes),
/*tokens*/ std::nullopt,
/*interfaceHash*/ std::nullopt};
Ctx.evaluator.cacheOutput(ParseSourceFileRequest{FileForLookups},
std::move(result));
}
}
|