Files
flipper/xplat/FlipperTestLib/FlipperPluginMock.h
John Knox c4d12c21d9 Rename Mock classes
Summary: Part of sonar to flipper rename

Reviewed By: priteshrnandgaonkar

Differential Revision: D9929102

fbshipit-source-id: 88cdbd33aa5c76e386449f54f4eb8c6435865ed1
2018-09-24 07:00:45 -07:00

63 lines
1.6 KiB
C++

/*
* Copyright (c) 2018-present, Facebook, Inc.
*
* This source code is licensed under the MIT license found in the LICENSE
* file in the root directory of this source tree.
*
*/
#pragma once
#include <Flipper/FlipperPlugin.h>
namespace facebook {
namespace flipper {
namespace test {
class FlipperPluginMock : public FlipperPlugin {
using ConnectionCallback =
std::function<void(std::shared_ptr<FlipperConnection>)>;
using DisconnectionCallback = std::function<void()>;
public:
FlipperPluginMock(const std::string& identifier) : identifier_(identifier) {}
FlipperPluginMock(
const std::string& identifier,
const ConnectionCallback& connectionCallback)
: identifier_(identifier), connectionCallback_(connectionCallback) {}
FlipperPluginMock(
const std::string& identifier,
const ConnectionCallback& connectionCallback,
const DisconnectionCallback& disconnectionCallback)
: identifier_(identifier),
connectionCallback_(connectionCallback),
disconnectionCallback_(disconnectionCallback) {}
std::string identifier() const override {
return identifier_;
}
void didConnect(std::shared_ptr<FlipperConnection> conn) override {
if (connectionCallback_) {
connectionCallback_(conn);
}
}
void didDisconnect() override {
if (disconnectionCallback_) {
disconnectionCallback_();
}
}
private:
std::string identifier_;
ConnectionCallback connectionCallback_;
DisconnectionCallback disconnectionCallback_;
};
} // namespace test
} // namespace flipper
} // namespace facebook