82 lines · cpp
1//===- GraphWriterTest.cpp - GraphWriter unit tests -----------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "llvm/Support/GraphWriter.h"10#include "llvm/Analysis/CFGPrinter.h"11#include "llvm/AsmParser/Parser.h"12#include "llvm/IR/Function.h"13#include "llvm/IR/LLVMContext.h"14#include "llvm/IR/Module.h"15#include "llvm/Testing/Support/SupportHelpers.h"16#include "llvm/Support/SourceMgr.h"17#include "llvm/Support/raw_ostream.h"18#include "gtest/gtest.h"19#include <string>20 21#define ASSERT_NO_ERROR(x) \22 if (std::error_code ASSERT_NO_ERROR_ec = x) { \23 SmallString<128> MessageStorage; \24 raw_svector_ostream Message(MessageStorage); \25 Message << #x ": did not return errc::success.\n" \26 << "error number: " << ASSERT_NO_ERROR_ec.value() << "\n" \27 << "error message: " << ASSERT_NO_ERROR_ec.message() << "\n"; \28 GTEST_FATAL_FAILURE_(MessageStorage.c_str()); \29 } else { \30 }31 32namespace llvm {33namespace {34 35class GraphWriterTest : public testing::Test {36protected:37 LLVMContext C;38 39 std::unique_ptr<Module> makeLLVMModule() {40 const char *ModuleStrig = "define i32 @f(i32 %x) {\n"41 "bb0:\n"42 " %y1 = icmp eq i32 %x, 0 \n"43 " br i1 %y1, label %bb1, label %bb2 \n"44 "bb1:\n"45 " br label %bb3\n"46 "bb2:\n"47 " br label %bb3\n"48 "bb3:\n"49 " %y2 = phi i32 [0, %bb1], [1, %bb2] \n"50 " ret i32 %y2\n"51 "}\n";52 SMDiagnostic Err;53 return parseAssemblyString(ModuleStrig, Err, C);54 }55};56 57static void writeCFGToDotFile(Function &F, std::string Name,58 bool CFGOnly = false) {59 std::error_code EC;60 llvm::unittest::TempDir Tmp("tmpdir", /*Unique=*/true);61 SmallString<128> FileName(Tmp.path().begin(), Tmp.path().end());62 sys::path::append(FileName, Name + ".dot");63 raw_fd_ostream File(FileName, EC, sys::fs::OpenFlags::OF_Text);64 65 DOTFuncInfo CFGInfo(&F);66 67 ASSERT_NO_ERROR(EC);68 // Test intentionally does not pass BPI, WriteGraph should work without it.69 WriteGraph(File, &CFGInfo, CFGOnly);70}71 72TEST_F(GraphWriterTest, WriteCFGDotFileTest) {73 auto M = makeLLVMModule();74 Function *F = M->getFunction("f");75 76 writeCFGToDotFile(*F, "test-full");77 writeCFGToDotFile(*F, "test-cfg-only", true);78}79 80} // end anonymous namespace81} // end namespace llvm82