RpcServerMock.h 666 B

1234567891011121314151617181920212223242526272829
  1. #ifndef RPC_RPCSERVERMOCK_H
  2. #define RPC_RPCSERVERMOCK_H
  3. #include <functional>
  4. #include <unordered_set>
  5. #include <unordered_map>
  6. #include <boost/asio.hpp>
  7. #include <google/protobuf/service.h>
  8. #include <gmock/gmock.h>
  9. #include "Rpc/RpcServer.h"
  10. namespace Egametang {
  11. class RpcServerMock: public RpcServer
  12. {
  13. public:
  14. RpcServerMock(boost::asio::io_service& service, int port):
  15. RpcServer(service, port)
  16. {
  17. }
  18. MOCK_METHOD4(RunService, void(RpcSessionPtr, const RpcMetaPtr, const StringPtr, MessageHandler));
  19. MOCK_METHOD1(Register, void(ProtobufServicePtr));
  20. MOCK_METHOD1(Remove, void(RpcSessionPtr&));
  21. };
  22. } // namespace Egametang
  23. #endif // RPC_RPCSERVERMOCK_H