162 lines
4.4 KiB
C++
162 lines
4.4 KiB
C++
/* -*- C++ -*-
|
|
* Serene Programming Language
|
|
*
|
|
* Copyright (c) 2019-2022 Sameer Rahmani <lxsameer@gnu.org>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, version 2.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#ifndef SERENE_TEST_CONTEXT_H
|
|
#define SERENE_TEST_CONTEXT_H
|
|
|
|
#include "serene/context.h"
|
|
#include "serene/namespace.h"
|
|
#include "serene/reader/location.h"
|
|
#include "serene/serene.h"
|
|
|
|
#include "./test_helpers.cpp.inc"
|
|
#include <catch2/catch_all.hpp>
|
|
#include <catch2/catch_test_macros.hpp>
|
|
|
|
#include <llvm/ADT/None.h>
|
|
|
|
namespace serene {
|
|
|
|
TEST_CASE("Compiler options", "[context]") {
|
|
auto opts = Options();
|
|
opts.JITLazy = !opts.JITLazy;
|
|
opts.JITenableGDBNotificationListener =
|
|
!opts.JITenableGDBNotificationListener;
|
|
|
|
auto ctx = makeSereneContext(opts);
|
|
|
|
CHECK(ctx->opts.JITLazy == opts.JITLazy);
|
|
CHECK(ctx->opts.JITenableGDBNotificationListener ==
|
|
opts.JITenableGDBNotificationListener);
|
|
};
|
|
|
|
TEST_CASE("makeNamespace & getNS", "[context]") {
|
|
auto ctx = makeSereneContext();
|
|
auto *ns = ctx->getNS("blah");
|
|
|
|
REQUIRE_FALSE(ns);
|
|
|
|
auto userNs =
|
|
ctx->makeNamespace("user", llvm::Optional<llvm::StringRef>("/some/file"));
|
|
|
|
CHECK(userNs->name == "user");
|
|
REQUIRE(userNs->filename);
|
|
CHECK(userNs->filename.getValue() == "/some/file");
|
|
|
|
ns = ctx->getNS("user");
|
|
|
|
REQUIRE(ns);
|
|
CHECK(ns->name == userNs->name);
|
|
|
|
/// Creating new ns with the same name overrides the old one
|
|
auto userNs1 = ctx->makeNamespace(
|
|
"user", llvm::Optional<llvm::StringRef>("/some/other/file"));
|
|
|
|
ns = ctx->getNS("user");
|
|
|
|
REQUIRE(ns);
|
|
CHECK(ns->name == userNs1->name);
|
|
REQUIRE(ns->filename);
|
|
CHECK(ns->filename.getValue() == "/some/other/file");
|
|
};
|
|
|
|
TEST_CASE(
|
|
"withCurrentNS run a function with the given namespace as the current NS",
|
|
"[context]") {
|
|
auto ctx = makeSereneContext();
|
|
auto userNs =
|
|
ctx->makeNamespace("user", llvm::Optional<llvm::StringRef>("/some/file"));
|
|
|
|
ctx->withCurrentNS<int>("user", [&]() -> int {
|
|
CHECK(ctx->getCurrentNS().name == userNs->name);
|
|
return 0;
|
|
});
|
|
|
|
// Checking the `void` type specialization
|
|
ctx->withCurrentNS<void>("user", [&]() -> void {
|
|
CHECK(ctx->getCurrentNS().name == userNs->name);
|
|
});
|
|
|
|
CHECK(ctx->getCurrentNS().name == DEFAULT_NS_NAME);
|
|
};
|
|
|
|
TEST_CASE("getSharedPtrToNS returns a shared ptr to the NS", "[context]") {
|
|
auto ctx = makeSereneContext();
|
|
auto userNs =
|
|
ctx->makeNamespace("user", llvm::Optional<llvm::StringRef>("/some/file"));
|
|
|
|
auto userNs1 = ctx->makeNamespace(
|
|
"user1", llvm::Optional<llvm::StringRef>("/some/file1"));
|
|
|
|
CHECK(ctx->getCurrentNS().name == DEFAULT_NS_NAME);
|
|
|
|
NSPtr shouldBeUser1 = ctx->getSharedPtrToNS("user");
|
|
|
|
REQUIRE(shouldBeUser1);
|
|
CHECK(shouldBeUser1->name == userNs->name);
|
|
};
|
|
|
|
TEST_CASE("Compilation phase", "[context]") {
|
|
auto ctx = makeSereneContext();
|
|
auto cp = CompilationPhase::O3;
|
|
|
|
ctx->setOperationPhase(cp);
|
|
|
|
CHECK(ctx->getTargetPhase() == cp);
|
|
CHECK(ctx->getOptimizatioLevel() == 3);
|
|
|
|
// Anything below 0 is 0
|
|
cp = CompilationPhase::MLIR;
|
|
ctx->setOperationPhase(cp);
|
|
CHECK(ctx->getOptimizatioLevel() == 0);
|
|
};
|
|
|
|
TEST_CASE("makeNamespace", "[context]") {
|
|
auto ctx = makeSereneContext();
|
|
|
|
auto ns = ctx->makeNamespace("example.ns", llvm::None);
|
|
|
|
// Namespace has to be empty
|
|
CHECK(ns->name == "example.ns");
|
|
CHECK(ns->getTree().empty());
|
|
CHECK(ns->filename == llvm::None);
|
|
};
|
|
|
|
TEST_CASE("context and jit", "[context]") {
|
|
auto ctx = makeSereneContext();
|
|
|
|
auto ns = ctx->makeNamespace("example.ns", llvm::None);
|
|
|
|
REQUIRE(ctx->jit);
|
|
|
|
// no JITDylib should be defined for an empty NS
|
|
CHECK(ctx->getNumberOfJITDylibs(*ns) == 0);
|
|
|
|
SECTION("JITDylib management") {
|
|
auto unknown = reader::LocationRange::UnknownLocation(DEFAULT_NS_NAME);
|
|
REQUIRE(!ctx->jit->addNS(*ns, unknown));
|
|
|
|
CHECK(ctx->getNumberOfJITDylibs(*ns) == 1);
|
|
|
|
REQUIRE(ctx->getLatestJITDylib(*ns));
|
|
CHECK(ctx->getLatestJITDylib(*ns)->getName() == "example.ns#1");
|
|
};
|
|
};
|
|
|
|
} // namespace serene
|
|
#endif
|