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