Update providedCiphersStr_ in one place.
[folly.git] / folly / io / async / test / AsyncSignalHandlerTest.cpp
1 /*
2  * Copyright 2017 Facebook, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 #include <folly/io/async/AsyncSignalHandler.h>
17 #include <folly/io/async/EventBase.h>
18
19 #include <folly/portability/GTest.h>
20
21 using namespace folly;
22
23 namespace {
24 class TestSignalHandler : public AsyncSignalHandler {
25  public:
26   using AsyncSignalHandler::AsyncSignalHandler;
27
28   void signalReceived(int /* signum */) noexcept override {
29     called = true;
30   }
31
32   bool called{false};
33 };
34 }
35
36 TEST(AsyncSignalHandler, basic) {
37   EventBase evb;
38   TestSignalHandler handler{&evb};
39
40   handler.registerSignalHandler(SIGUSR1);
41   kill(getpid(), SIGUSR1);
42
43   EXPECT_FALSE(handler.called);
44   evb.loopOnce(EVLOOP_NONBLOCK);
45   EXPECT_TRUE(handler.called);
46 }
47
48 TEST(AsyncSignalHandler, attachEventBase) {
49   TestSignalHandler handler{nullptr};
50   EXPECT_FALSE(handler.getEventBase());
51   EventBase evb;
52
53   handler.attachEventBase(&evb);
54   EXPECT_EQ(&evb, handler.getEventBase());
55
56   handler.registerSignalHandler(SIGUSR1);
57   kill(getpid(), SIGUSR1);
58   EXPECT_FALSE(handler.called);
59   evb.loopOnce(EVLOOP_NONBLOCK);
60   EXPECT_TRUE(handler.called);
61
62   handler.unregisterSignalHandler(SIGUSR1);
63   handler.detachEventBase();
64   EXPECT_FALSE(handler.getEventBase());
65 }