861c5077f1
Summary: Its helpful to store all history of sent messages for testing purposes. Reviewed By: jknoxville Differential Revision: D14797191 fbshipit-source-id: 0e58166cb4ab79852165f2107b2f6bcd259f1096
40 lines
1010 B
C++
40 lines
1010 B
C++
/**
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* 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/FlipperConnection.h>
|
|
#include <map>
|
|
#include <queue>
|
|
#include <string>
|
|
|
|
namespace facebook {
|
|
namespace flipper {
|
|
|
|
class FlipperConnectionMock : public FlipperConnection {
|
|
public:
|
|
void send(const std::string& method, const folly::dynamic& params) override {
|
|
sent_[method] = params;
|
|
sent_message_history[method].push(params);
|
|
}
|
|
|
|
void receive(const std::string& method, const FlipperReceiver& receiver)
|
|
override {
|
|
receivers_[method] = receiver;
|
|
}
|
|
|
|
void error(const std::string& message, const std::string& stacktrace)
|
|
override {}
|
|
|
|
std::map<std::string, folly::dynamic> sent_;
|
|
std::map<std::string, FlipperReceiver> receivers_;
|
|
|
|
std::map<std::string, std::queue<folly::dynamic>> sent_message_history;
|
|
};
|
|
|
|
} // namespace flipper
|
|
} // namespace facebook
|