BlackMATov/promise.hpp
1
0
Fork 0
mirror of https://github.com/BlackMATov/promise.hpp.git synced 2025-12-13 03:46:29 +07:00
Code Issues Packages Projects Releases Wiki Activity

add finally method

Browse Source 
https://github.com/BlackMATov/promise.hpp/issues/33
This commit is contained in:
BlackMATov
2020-01-01 03:26:41 +07:00
parent 00b102f82c
commit 661b30b395
2 changed files with 155 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
headers/promise.hpp/promise.hpp
View File

@@ -416,6 +416,21 @@ namespace promise_hpp
[](auto&& v) { return std::forward<decltype(v)>(v); },
std::forward<RejectF>(on_reject));
}
//
// finally
//
template < typename FinallyF >
promise<T> finally(FinallyF&& on_finally) {
return then([f = on_finally](auto&& v) {
std::invoke(std::move(f));
return std::forward<decltype(v)>(v);
}, [f = on_finally](std::exception_ptr e) -> T {
std::invoke(std::move(f));
std::rethrow_exception(e);
});
}
private:
class state;
std::shared_ptr<state> state_;
@@ -837,6 +852,20 @@ namespace promise_hpp
[](){},
std::forward<RejectF>(on_reject));
}
//
// finally
//
template < typename FinallyF >
promise<void> finally(FinallyF&& on_finally) {
return then([f = on_finally]() {
std::invoke(std::move(f));
}, [f = on_finally](std::exception_ptr e) {
std::invoke(std::move(f));
std::rethrow_exception(e);
});
}
private:
class state;
std::shared_ptr<state> state_;

126
untests/promise_tests.cpp
View File

@@ -310,6 +310,132 @@ TEST_CASE("promise") {
REQUIRE(call_fail_with_logic_error);
}
}
SECTION("finally") {
{
bool all_is_ok = false;
auto p = pr::promise<int>();
p.finally([&all_is_ok](){
all_is_ok = true;
});
REQUIRE_FALSE(all_is_ok);
p.resolve(1);
REQUIRE(all_is_ok);
}
{
bool all_is_ok = false;
auto p = pr::promise<int>();
p.finally([&all_is_ok](){
all_is_ok = true;
});
REQUIRE_FALSE(all_is_ok);
p.reject(std::make_exception_ptr(std::logic_error("hello fail")));
REQUIRE(all_is_ok);
}
{
bool all_is_ok = false;
pr::make_resolved_promise(1)
.finally([&all_is_ok](){
all_is_ok = true;
});
REQUIRE(all_is_ok);
}
{
bool all_is_ok = false;
pr::make_rejected_promise<int>(std::logic_error("hello fail"))
.finally([&all_is_ok](){
all_is_ok = true;
});
REQUIRE(all_is_ok);
}
}
SECTION("after_finally") {
{
int check_84_int = 0;
auto p = pr::promise<>();
p.finally([&check_84_int](){
check_84_int = 42;
return 100500;
}).then([&check_84_int](){
check_84_int *= 2;
});
REQUIRE(check_84_int == 0);
p.resolve();
REQUIRE(check_84_int == 84);
}
{
int check_84_int = 0;
auto p = pr::promise<>();
p.finally([&check_84_int](){
check_84_int = 42;
return 100500;
}).except([&check_84_int](std::exception_ptr){
check_84_int *= 2;
});
REQUIRE(check_84_int == 0);
p.reject(std::make_exception_ptr(std::logic_error("hello fail")));
REQUIRE(check_84_int == 84);
}
}
SECTION("failed_finally") {
{
int check_84_int = 0;
auto p = pr::promise<>();
p.finally([&check_84_int](){
check_84_int += 42;
throw std::logic_error("hello fail");
}).except([&check_84_int](std::exception_ptr e){
if ( check_hello_fail_exception(e) ) {
check_84_int += 42;
}
});
p.resolve();
REQUIRE(check_84_int == 84);
}
{
int check_84_int = 0;
auto p = pr::promise<>();
p.finally([&check_84_int](){
check_84_int += 42;
throw std::logic_error("hello fail");
}).except([&check_84_int](std::exception_ptr e){
if ( check_hello_fail_exception(e) ) {
check_84_int += 42;
}
});
p.reject(std::make_exception_ptr(std::logic_error("hello")));
REQUIRE(check_84_int == 84);
}
{
int check_84_int = 0;
auto p = pr::promise<int>();
p.finally([&check_84_int](){
check_84_int += 42;
throw std::logic_error("hello fail");
}).except([&check_84_int](std::exception_ptr e) -> int {
if ( check_hello_fail_exception(e) ) {
check_84_int += 42;
}
return 0;
});
p.resolve(1);
REQUIRE(check_84_int == 84);
}
{
int check_84_int = 0;
auto p = pr::promise<int>();
p.finally([&check_84_int](){
check_84_int += 42;
throw std::logic_error("hello fail");
}).except([&check_84_int](std::exception_ptr e) -> int {
if ( check_hello_fail_exception(e) ) {
check_84_int += 42;
}
return 0;
});
p.reject(std::make_exception_ptr(std::logic_error("hello")));
REQUIRE(check_84_int == 84);
}
}
SECTION("make_promise") {
{
int check_84_int = 0;