1 // Copyright 2022 gRPC authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "src/core/lib/event_engine/common_closures.h"
16
17 #include <memory>
18
19 #include "absl/functional/any_invocable.h"
20 #include "gtest/gtest.h"
21
22 #include <grpc/support/port_platform.h>
23
24 #include "src/core/lib/gprpp/notification.h"
25
26 using ::grpc_event_engine::experimental::AnyInvocableClosure;
27 using ::grpc_event_engine::experimental::SelfDeletingClosure;
28
29 class AnyInvocableClosureTest : public testing::Test {};
30
TEST_F(AnyInvocableClosureTest,CallsItsFunction)31 TEST_F(AnyInvocableClosureTest, CallsItsFunction) {
32 grpc_core::Notification signal;
33 AnyInvocableClosure closure([&signal] { signal.Notify(); });
34 closure.Run();
35 signal.WaitForNotification();
36 }
37
38 class SelfDeletingClosureTest : public testing::Test {};
39
TEST_F(SelfDeletingClosureTest,CallsItsFunction)40 TEST_F(SelfDeletingClosureTest, CallsItsFunction) {
41 grpc_core::Notification signal;
42 auto* closure = SelfDeletingClosure::Create([&signal] { signal.Notify(); });
43 closure->Run();
44 signal.WaitForNotification();
45 // ASAN should catch if this closure is not deleted
46 }
47
TEST_F(SelfDeletingClosureTest,CallsItsFunctionAndIsDestroyed)48 TEST_F(SelfDeletingClosureTest, CallsItsFunctionAndIsDestroyed) {
49 grpc_core::Notification fn_called;
50 grpc_core::Notification destroyed;
51 auto* closure =
52 SelfDeletingClosure::Create([&fn_called] { fn_called.Notify(); },
53 [&destroyed] { destroyed.Notify(); });
54 closure->Run();
55 fn_called.WaitForNotification();
56 destroyed.WaitForNotification();
57 }
58
main(int argc,char ** argv)59 int main(int argc, char** argv) {
60 testing::InitGoogleTest(&argc, argv);
61 auto result = RUN_ALL_TESTS();
62 return result;
63 }
64