From 80f329bcd0281c11062879025761d0657167fe8b Mon Sep 17 00:00:00 2001 From: Petr Hosek Date: Wed, 24 Feb 2021 10:26:11 -0800 Subject: [PATCH] [Profile] Include a few asserts in coverage mapping test These should catch any accidental use of the compilation directory. Differential Revision: https://reviews.llvm.org/D97402 --- llvm/unittests/ProfileData/CoverageMappingTest.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/llvm/unittests/ProfileData/CoverageMappingTest.cpp b/llvm/unittests/ProfileData/CoverageMappingTest.cpp index d1bb87a..3f9a00b 100644 --- a/llvm/unittests/ProfileData/CoverageMappingTest.cpp +++ b/llvm/unittests/ProfileData/CoverageMappingTest.cpp @@ -145,8 +145,10 @@ struct CoverageMappingTest : ::testing::TestWithParam> { unsigned getGlobalFileIndex(StringRef Name) { auto R = Files.find(Name); - if (R != Files.end()) + if (R != Files.end()) { + assert(R->second > 0 && "got index of compilation dir"); return R->second; + } unsigned Index = Files.size() + 1; Files.try_emplace(Name, Index); return Index; @@ -161,8 +163,10 @@ struct CoverageMappingTest : ::testing::TestWithParam> { auto &CurrentFunctionFileMapping = InputFunctions.back().ReverseVirtualFileMapping; auto R = CurrentFunctionFileMapping.find(GlobalIndex); - if (R != CurrentFunctionFileMapping.end()) + if (R != CurrentFunctionFileMapping.end()) { + assert(R->second > 0 && "got index of compilation dir"); return R->second; + } unsigned IndexInFunction = CurrentFunctionFileMapping.size(); CurrentFunctionFileMapping.insert( std::make_pair(GlobalIndex, IndexInFunction)); -- 2.7.4