1 /* 2 * Copyright (c) 2024 Shenzhen Kaihong Digital Industry Development Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef OHOS_SHARING_MOCK_NETWORK_SESSION_H 17 #define OHOS_SHARING_MOCK_NETWORK_SESSION_H 18 19 #include <gmock/gmock.h> 20 #include "inetwork_session.h" 21 22 namespace OHOS { 23 namespace Sharing { 24 25 class MockNetworkSession : public INetworkSession { 26 public: Send(const DataBuffer::Ptr & buf,int32_t nSize)27 bool Send(const DataBuffer::Ptr &buf, int32_t nSize) 28 { 29 (void)buf; 30 (void)nSize; 31 return false; 32 } 33 RegisterCallback(std::shared_ptr<INetworkSessionCallback> callback)34 void RegisterCallback(std::shared_ptr<INetworkSessionCallback> callback) 35 { 36 callback_ = std::move(callback); 37 } 38 GetCallback()39 std::shared_ptr<INetworkSessionCallback> &GetCallback() 40 { 41 return callback_; 42 } 43 44 private: 45 std::shared_ptr<INetworkSessionCallback> callback_ = nullptr; 46 47 public: 48 MOCK_METHOD(bool, Start, ()); 49 MOCK_METHOD(void, Shutdown, ()); 50 MOCK_METHOD(SocketInfo::Ptr, GetSocketInfo, ()); 51 MOCK_METHOD(bool, Send, (const char *buf, int32_t nSize)); 52 }; 53 54 } // namespace Sharing 55 } // namespace OHOS 56 #endif 57