Extended ConnectionMock functionality
Summary: Its helpful to store all history of sent messages for testing purposes. Reviewed By: jknoxville Differential Revision: D14797191 fbshipit-source-id: 0e58166cb4ab79852165f2107b2f6bcd259f1096
This commit is contained in:
committed by
Facebook Github Bot
parent
684ecd8a39
commit
861c5077f1
@@ -1,15 +1,14 @@
|
|||||||
/*
|
/**
|
||||||
* Copyright (c) 2018-present, Facebook, Inc.
|
* 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.
|
|
||||||
*
|
*
|
||||||
|
* This source code is licensed under the MIT license found in the LICENSE
|
||||||
|
* file in the root directory of this source tree.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <Flipper/FlipperConnection.h>
|
#include <Flipper/FlipperConnection.h>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
#include <queue>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
|
||||||
namespace facebook {
|
namespace facebook {
|
||||||
@@ -19,6 +18,7 @@ class FlipperConnectionMock : public FlipperConnection {
|
|||||||
public:
|
public:
|
||||||
void send(const std::string& method, const folly::dynamic& params) override {
|
void send(const std::string& method, const folly::dynamic& params) override {
|
||||||
sent_[method] = params;
|
sent_[method] = params;
|
||||||
|
sent_message_history[method].push(params);
|
||||||
}
|
}
|
||||||
|
|
||||||
void receive(const std::string& method, const FlipperReceiver& receiver)
|
void receive(const std::string& method, const FlipperReceiver& receiver)
|
||||||
@@ -31,6 +31,8 @@ class FlipperConnectionMock : public FlipperConnection {
|
|||||||
|
|
||||||
std::map<std::string, folly::dynamic> sent_;
|
std::map<std::string, folly::dynamic> sent_;
|
||||||
std::map<std::string, FlipperReceiver> receivers_;
|
std::map<std::string, FlipperReceiver> receivers_;
|
||||||
|
|
||||||
|
std::map<std::string, std::queue<folly::dynamic>> sent_message_history;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace flipper
|
} // namespace flipper
|
||||||
|
|||||||
Reference in New Issue
Block a user