1 /* 2 * Copyright (c) 2017, 2018, Oracle and/or its affiliates. All rights reserved. 3 * 4 * This program is free software; you can redistribute it and/or modify 5 * it under the terms of the GNU General Public License, version 2.0, 6 * as published by the Free Software Foundation. 7 * 8 * This program is also distributed with certain software (including 9 * but not limited to OpenSSL) that is licensed under separate terms, 10 * as designated in a particular file or component or in included license 11 * documentation. The authors of MySQL hereby grant you an additional 12 * permission to link the program and your derivative works with the 13 * separately licensed software that they have included with MySQL. 14 * 15 * This program is distributed in the hope that it will be useful, 16 * but WITHOUT ANY WARRANTY; without even the implied warranty of 17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 18 * GNU General Public License, version 2.0, for more details. 19 * 20 * You should have received a copy of the GNU General Public License 21 * along with this program; if not, write to the Free Software 22 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 23 */ 24 25 #ifndef UNITTEST_GUNIT_XPLUGIN_XCL_MOCK_CONNECTION_H_ 26 #define UNITTEST_GUNIT_XPLUGIN_XCL_MOCK_CONNECTION_H_ 27 28 #include <gmock/gmock.h> 29 #include <cstdint> 30 #include <memory> 31 #include <string> 32 #include <utility> 33 34 #include "plugin/x/client/mysqlxclient/xconnection.h" 35 36 namespace xcl { 37 namespace test { 38 39 class Mock_connection : public XConnection { 40 public: 41 MOCK_METHOD1(connect_to_localhost, XError(const std::string &unix_socket)); 42 MOCK_METHOD3(connect, XError(const std::string &host, const uint16_t port, 43 const Internet_protocol ip_mode)); 44 MOCK_METHOD0(get_socket_fd, my_socket()); 45 MOCK_METHOD0(activate_tls, XError()); 46 MOCK_METHOD1(shutdown, XError(const Shutdown_type how_to_shutdown)); 47 MOCK_METHOD2(write, 48 XError(const uint8_t *data, const std::size_t data_length)); 49 MOCK_METHOD2(read, XError(uint8_t *data, const std::size_t data_length)); 50 MOCK_METHOD1(set_read_timeout, XError(const int deadline_milliseconds)); 51 MOCK_METHOD1(set_write_timeout, XError(const int deadline_milliseconds)); 52 MOCK_METHOD0(close, void()); 53 MOCK_METHOD0(state, const State &()); 54 }; 55 56 } // namespace test 57 } // namespace xcl 58 59 #endif // UNITTEST_GUNIT_XPLUGIN_XCL_MOCK_CONNECTION_H_ 60