1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465 |
- /*
- * Copyright 2017-present Facebook, Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include <folly/io/async/AsyncSignalHandler.h>
- #include <folly/io/async/EventBase.h>
- #include <folly/portability/GTest.h>
- using namespace folly;
- namespace {
- class TestSignalHandler : public AsyncSignalHandler {
- public:
- using AsyncSignalHandler::AsyncSignalHandler;
- void signalReceived(int /* signum */) noexcept override {
- called = true;
- }
- bool called{false};
- };
- } // namespace
- TEST(AsyncSignalHandler, basic) {
- EventBase evb;
- TestSignalHandler handler{&evb};
- handler.registerSignalHandler(SIGUSR1);
- kill(getpid(), SIGUSR1);
- EXPECT_FALSE(handler.called);
- evb.loopOnce(EVLOOP_NONBLOCK);
- EXPECT_TRUE(handler.called);
- }
- TEST(AsyncSignalHandler, attachEventBase) {
- TestSignalHandler handler{nullptr};
- EXPECT_FALSE(handler.getEventBase());
- EventBase evb;
- handler.attachEventBase(&evb);
- EXPECT_EQ(&evb, handler.getEventBase());
- handler.registerSignalHandler(SIGUSR1);
- kill(getpid(), SIGUSR1);
- EXPECT_FALSE(handler.called);
- evb.loopOnce(EVLOOP_NONBLOCK);
- EXPECT_TRUE(handler.called);
- handler.unregisterSignalHandler(SIGUSR1);
- handler.detachEventBase();
- EXPECT_FALSE(handler.getEventBase());
- }
|