e44cad5e99
Summary: This change isolates the usage of folly async from Flipper. Is now self-contained in Flipper Folly schedulers. Users of Flipper can decide not to use the types defined in that header and implement their own. NOTE: changes are minimal, we are just replacing direct calls to folly event base with a scheduler which simply relays this on to folly. Reviewed By: fabiomassimo Differential Revision: D36626483 fbshipit-source-id: add0241caf4af0aa5c3b5c2e7efc2e725f5400ab
100 lines
3.1 KiB
C++
100 lines
3.1 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#include <Flipper/FlipperConnectionManagerImpl.h>
|
|
#include <Flipper/FlipperFollyScheduler.h>
|
|
#include <FlipperTestLib/ConnectionContextStoreMock.h>
|
|
#include <folly/Singleton.h>
|
|
#include <gtest/gtest.h>
|
|
#include <memory>
|
|
|
|
namespace facebook {
|
|
namespace flipper {
|
|
namespace test {
|
|
|
|
using folly::EventBase;
|
|
|
|
class FlipperConnectionManagerImplTerminationTest : public ::testing::Test {
|
|
protected:
|
|
std::shared_ptr<FlipperState> state;
|
|
std::shared_ptr<ConnectionContextStore> contextStore;
|
|
std::unique_ptr<Scheduler> sonarScheduler;
|
|
std::unique_ptr<Scheduler> connectionScheduler;
|
|
void SetUp() override {
|
|
// Folly singletons must be registered before they are used.
|
|
// Without this, test fails in phabricator.
|
|
folly::SingletonVault::singleton()->registrationComplete();
|
|
state = std::make_shared<FlipperState>();
|
|
contextStore = std::make_shared<ConnectionContextStoreMock>();
|
|
sonarScheduler = std::make_unique<FollyScheduler>(new EventBase());
|
|
connectionScheduler = std::make_unique<FollyScheduler>(new EventBase());
|
|
}
|
|
};
|
|
|
|
TEST_F(
|
|
FlipperConnectionManagerImplTerminationTest,
|
|
testNullEventBaseGetsRejected) {
|
|
try {
|
|
auto instance = std::make_shared<FlipperConnectionManagerImpl>(
|
|
FlipperInitConfig{DeviceData{}, nullptr, connectionScheduler.get()},
|
|
state,
|
|
contextStore);
|
|
FAIL();
|
|
} catch (std::invalid_argument& e) {
|
|
// Pass test
|
|
}
|
|
try {
|
|
auto instance = std::make_shared<FlipperConnectionManagerImpl>(
|
|
FlipperInitConfig{DeviceData{}, sonarScheduler.get(), nullptr},
|
|
state,
|
|
contextStore);
|
|
FAIL();
|
|
} catch (std::invalid_argument& e) {
|
|
// Pass test
|
|
}
|
|
}
|
|
|
|
TEST_F(
|
|
FlipperConnectionManagerImplTerminationTest,
|
|
testNonStartedEventBaseDoesntHang) {
|
|
auto config = FlipperInitConfig{
|
|
DeviceData{}, sonarScheduler.get(), connectionScheduler.get()};
|
|
auto instance = std::make_shared<FlipperConnectionManagerImpl>(
|
|
config, state, contextStore);
|
|
instance->start();
|
|
}
|
|
|
|
TEST_F(
|
|
FlipperConnectionManagerImplTerminationTest,
|
|
testStartedEventBaseDoesntHang) {
|
|
auto flipperEventBase = new EventBase();
|
|
auto connectionEventBase = new EventBase();
|
|
auto flipperThread =
|
|
std::thread([flipperEventBase]() { flipperEventBase->loopForever(); });
|
|
auto connectionThread = std::thread(
|
|
[connectionEventBase]() { connectionEventBase->loopForever(); });
|
|
auto localSonarScheduler = std::make_unique<FollyScheduler>(flipperEventBase);
|
|
auto localConnectionScheduler =
|
|
std::make_unique<FollyScheduler>(connectionEventBase);
|
|
auto config = FlipperInitConfig{
|
|
DeviceData{}, localSonarScheduler.get(), localConnectionScheduler.get()};
|
|
auto instance = std::make_shared<FlipperConnectionManagerImpl>(
|
|
config, state, contextStore);
|
|
|
|
instance->start();
|
|
|
|
flipperEventBase->terminateLoopSoon();
|
|
connectionEventBase->terminateLoopSoon();
|
|
|
|
flipperThread.join();
|
|
connectionThread.join();
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace flipper
|
|
} // namespace facebook
|