1 /* 2 * Copyright (c) 2024 Huawei Device 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 #include "net_dns_result_callback_proxy.h" 17 #include <gmock/gmock.h> 18 #include <gtest/gtest.h> 19 #include "net_manager_constants.h" 20 21 namespace OHOS { 22 namespace NetsysNative { 23 using namespace testing::ext; 24 using ::testing::_; 25 using ::testing::Return; 26 27 class RemoteObjectMocker : public IRemoteObject { 28 public: RemoteObjectMocker()29 RemoteObjectMocker() : IRemoteObject{u"RemoteObjectMocker"} {} ~RemoteObjectMocker()30 ~RemoteObjectMocker() {} 31 32 MOCK_METHOD(int32_t, GetObjectRefCount, (), (override)); 33 MOCK_METHOD(int, SendRequest, (uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option), 34 (override)); 35 MOCK_METHOD(bool, IsProxyObject, (), (const, override)); 36 MOCK_METHOD(bool, IsObjectDead, (), (const, override)); 37 MOCK_METHOD(std::u16string, GetInterfaceDescriptor, (), (override)); 38 MOCK_METHOD(bool, CheckObjectLegality, (), (const, override)); 39 MOCK_METHOD(bool, AddDeathRecipient, (const sptr<DeathRecipient> &recipient), (override)); 40 MOCK_METHOD(bool, RemoveDeathRecipient, (const sptr<DeathRecipient> &recipient), (override)); 41 MOCK_METHOD(bool, Marshalling, (OHOS::Parcel & parcel), (const, override)); 42 MOCK_METHOD(sptr<IRemoteBroker>, AsInterface, (), (override)); 43 MOCK_METHOD(int, Dump, (int fd, const std::vector<std::u16string> &args), (override)); 44 }; 45 46 class NetDnsResultCallbackProxyTest : public testing::Test { 47 protected: 48 RemoteObjectMocker *remoteObjectMocker; 49 sptr<NetDnsResultCallbackProxy> proxy; 50 SetUp()51 void SetUp() override 52 { 53 remoteObjectMocker = new RemoteObjectMocker(); 54 sptr<IRemoteObject> impl(remoteObjectMocker); 55 proxy = new (std::nothrow) NetDnsResultCallbackProxy(impl); 56 } 57 TearDown()58 void TearDown() override 59 { 60 remoteObjectMocker = nullptr; 61 proxy = nullptr; 62 } 63 }; 64 65 HWTEST_F(NetDnsResultCallbackProxyTest, NetDnsResultCallbackProxyTest_OnDnsResultReport_001, TestSize.Level0) 66 { 67 uint32_t listsize = 1; 68 std::list<NetDnsResultReport> dnsResultReport; 69 int32_t ret = proxy->OnDnsResultReport(listsize, dnsResultReport); 70 EXPECT_EQ(ret, NetManagerStandard::NETMANAGER_SUCCESS); 71 } 72 73 HWTEST_F(NetDnsResultCallbackProxyTest, NetDnsResultCallbackProxyTest_OnDnsResultReport_002, TestSize.Level0) 74 { 75 uint32_t listsize = 0; 76 std::list<NetDnsResultReport> dnsResultReport; 77 EXPECT_CALL(*remoteObjectMocker, SendRequest(_, _, _, _)).WillOnce(Return(1)); 78 int32_t ret = proxy->OnDnsResultReport(listsize, dnsResultReport); 79 EXPECT_EQ(ret, NetManagerStandard::NETMANAGER_ERR_OPERATION_FAILED); 80 } 81 82 HWTEST_F(NetDnsResultCallbackProxyTest, NetDnsResultCallbackProxyTest_OnDnsResultReport_003, TestSize.Level0) 83 { 84 uint32_t listsize = 0; 85 std::list<NetDnsResultReport> dnsResultReport; 86 NetDnsResultReport report; 87 dnsResultReport.push_back(report); 88 int32_t ret = proxy->OnDnsResultReport(listsize, dnsResultReport); 89 EXPECT_EQ(ret, NetManagerStandard::NETMANAGER_SUCCESS); 90 } 91 92 HWTEST_F(NetDnsResultCallbackProxyTest, NetDnsResultCallbackProxyTest_OnDnsQueryResultReport_001, TestSize.Level0) 93 { 94 uint32_t listsize = 1; 95 std::list<NetDnsQueryResultReport> dnsResultReport; 96 int32_t ret = proxy->OnDnsQueryResultReport(listsize, dnsResultReport); 97 EXPECT_EQ(ret, NetManagerStandard::NETMANAGER_SUCCESS); 98 } 99 100 HWTEST_F(NetDnsResultCallbackProxyTest, NetDnsResultCallbackProxyTest_OnDnsQueryResultReport_002, TestSize.Level0) 101 { 102 uint32_t listsize = 0; 103 std::list<NetDnsQueryResultReport> dnsResultReport; 104 EXPECT_CALL(*remoteObjectMocker, SendRequest(_, _, _, _)).WillOnce(Return(1)); 105 int32_t ret = proxy->OnDnsQueryResultReport(listsize, dnsResultReport); 106 EXPECT_EQ(ret, NetManagerStandard::NETMANAGER_ERR_OPERATION_FAILED); 107 } 108 109 HWTEST_F(NetDnsResultCallbackProxyTest, NetDnsResultCallbackProxyTest_OnDnsQueryResultReport_003, TestSize.Level0) 110 { 111 uint32_t listsize = 1; 112 std::list<NetDnsQueryResultReport> dnsResultReport; 113 NetDnsQueryResultReport report; 114 dnsResultReport.push_back(report); 115 int32_t ret = proxy->OnDnsQueryResultReport(listsize, dnsResultReport); 116 EXPECT_EQ(ret, NetManagerStandard::NETMANAGER_SUCCESS); 117 } 118 119 HWTEST_F(NetDnsResultCallbackProxyTest, NetDnsResultCallbackProxyTest_OnDnsQueryAbnormalReport_001, TestSize.Level0) 120 { 121 uint32_t eventfailcause = 1; 122 NetDnsQueryResultReport report; 123 int32_t ret = proxy->OnDnsQueryAbnormalReport(eventfailcause, report); 124 EXPECT_EQ(ret, NetManagerStandard::NETMANAGER_SUCCESS); 125 } 126 127 } // namespace NetsysNative 128 } // namespace OHOS