49 lines
1.1 KiB
C++
49 lines
1.1 KiB
C++
|
//===- llvm/unittest/ADT/ScopeExit.cpp - Scope exit unit tests --*- C++ -*-===//
|
||
|
|
//
|
||
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
||
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
||
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#include "llvm/ADT/ScopeExit.h"
|
||
|
|
#include "gtest/gtest.h"
|
||
|
|
|
||
|
|
using namespace llvm;
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
TEST(ScopeExitTest, Basic) {
|
||
|
|
struct Callable {
|
||
|
|
bool &Called;
|
||
|
|
Callable(bool &Called) : Called(Called) {}
|
||
|
|
Callable(Callable &&RHS) : Called(RHS.Called) {}
|
||
|
|
void operator()() { Called = true; }
|
||
|
|
};
|
||
|
|
bool Called = false;
|
||
|
|
{
|
||
|
|
auto g = make_scope_exit(Callable(Called));
|
||
|
|
EXPECT_FALSE(Called);
|
||
|
|
}
|
||
|
|
EXPECT_TRUE(Called);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(ScopeExitTest, Release) {
|
||
|
|
int Count = 0;
|
||
|
|
auto Increment = [&] { ++Count; };
|
||
|
|
{
|
||
|
|
auto G = make_scope_exit(Increment);
|
||
|
|
auto H = std::move(G);
|
||
|
|
auto I = std::move(G);
|
||
|
|
EXPECT_EQ(0, Count);
|
||
|
|
}
|
||
|
|
EXPECT_EQ(1, Count);
|
||
|
|
{
|
||
|
|
auto G = make_scope_exit(Increment);
|
||
|
|
G.release();
|
||
|
|
}
|
||
|
|
EXPECT_EQ(1, Count);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // end anonymous namespace
|