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 66 67 68 69 70 71 72 73 74 75 76 77 78 79
|
#include <exception>
#include "scopeguard.h"
#include "gtest/gtest.h"
namespace bpftrace {
TEST(ScopeGuardTest, InnerScope)
{
int x = 5;
{
SCOPE_EXIT
{
x++;
};
}
EXPECT_EQ(x, 6);
}
TEST(ScopeGuardTest, FunctionReturn)
{
int x = 5;
[&]() {
SCOPE_EXIT
{
x++;
};
x++;
return;
}();
EXPECT_EQ(x, 7);
}
TEST(ScopeGuardTest, ExceptionContext)
{
int x = 5;
try {
[&]() {
SCOPE_EXIT
{
x++;
};
throw std::runtime_error("exception");
}();
} catch (const std::exception &) {
EXPECT_EQ(x, 6);
x++;
}
EXPECT_EQ(x, 7);
}
TEST(ScopeGuardTest, MultipleGuards)
{
int x = 5;
{
SCOPE_EXIT
{
x++;
};
SCOPE_EXIT
{
x++;
};
}
EXPECT_EQ(x, 7);
}
} // namespace bpftrace
|