1 /*
2 * Copyright (c) 2025 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 <cstdio>
17 #include <cstring>
18 #include <gtest/gtest.h>
19
20 #include "usb_driver_manager.h"
21
22 using namespace std;
23 using namespace testing::ext;
24 using namespace OHOS::HDI::Usb::Ddk;
25
26 namespace {
27 class UsbDdkServiceTest : public testing::Test {
28 public:
29 static void SetUpTestCase();
30 static void TearDownTestCase();
31 void SetUp();
32 void TearDown();
33 };
34
SetUpTestCase()35 void UsbDdkServiceTest::SetUpTestCase()
36 {
37 }
38
TearDownTestCase()39 void UsbDdkServiceTest::TearDownTestCase()
40 {
41 }
42
SetUp()43 void UsbDdkServiceTest::SetUp()
44 {
45 }
46
TearDown()47 void UsbDdkServiceTest::TearDown()
48 {
49 }
50
51 HWTEST_F(UsbDdkServiceTest, UpdateDriverInfo001, TestSize.Level1)
52 {
53 V1_1::DriverAbilityInfo driverInfo;
54 driverInfo.driverUid = "driverUid12345";
55 bool ret = V1_1::UsbDriverManager::GetInstance().UpdateDriverInfo(driverInfo);
56 EXPECT_FALSE(ret);
57
58 driverInfo.driverUid = "driverUid-12345";
59 driverInfo.vids = { 1001 };
60 ret = V1_1::UsbDriverManager::GetInstance().UpdateDriverInfo(driverInfo);
61 EXPECT_TRUE(ret);
62
63 uint32_t tokenId = 12345;
64 V1_1::DriverAbilityInfo queriedDriverInfo;
65 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
66 EXPECT_TRUE(ret);
67 EXPECT_EQ(driverInfo.driverUid, queriedDriverInfo.driverUid);
68 auto iter = std::find(queriedDriverInfo.vids.begin(), queriedDriverInfo.vids.end(), 1001);
69 EXPECT_NE(queriedDriverInfo.vids.end(), iter);
70 }
71
72 HWTEST_F(UsbDdkServiceTest, UpdateDriverInfo002, TestSize.Level1)
73 {
74 V1_1::DriverAbilityInfo driverInfo;
75 driverInfo.driverUid = "driverUid-12345";
76 driverInfo.vids = { 1001 };
77 bool ret = V1_1::UsbDriverManager::GetInstance().UpdateDriverInfo(driverInfo);
78 EXPECT_TRUE(ret);
79
80 uint32_t tokenId = 12345;
81 V1_1::DriverAbilityInfo queriedDriverInfo;
82 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
83 EXPECT_TRUE(ret);
84 EXPECT_EQ(1, queriedDriverInfo.vids.size());
85
86 driverInfo.vids = { 1001, 1002 };
87 ret = V1_1::UsbDriverManager::GetInstance().UpdateDriverInfo(driverInfo);
88 EXPECT_TRUE(ret);
89
90 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
91 EXPECT_TRUE(ret);
92 EXPECT_EQ(2, queriedDriverInfo.vids.size());
93 auto iter = std::find(queriedDriverInfo.vids.begin(), queriedDriverInfo.vids.end(), 1002);
94 EXPECT_NE(queriedDriverInfo.vids.end(), iter);
95 }
96
97 HWTEST_F(UsbDdkServiceTest, RemoveDriverInfo001, TestSize.Level1)
98 {
99 V1_1::DriverAbilityInfo driverInfo;
100 driverInfo.driverUid = "driverUid-12345";
101 bool ret = V1_1::UsbDriverManager::GetInstance().UpdateDriverInfo(driverInfo);
102 EXPECT_TRUE(ret);
103 uint32_t tokenId = 12345;
104 V1_1::DriverAbilityInfo queriedDriverInfo;
105 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
106 EXPECT_TRUE(ret);
107
108 std::string driverUid = "driverUid12345";
109 ret = V1_1::UsbDriverManager::GetInstance().RemoveDriverInfo(driverUid);
110 EXPECT_FALSE(ret);
111 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
112 EXPECT_TRUE(ret);
113
114 driverUid = "driverUid-12345";
115 ret = V1_1::UsbDriverManager::GetInstance().RemoveDriverInfo(driverUid);
116 EXPECT_TRUE(ret);
117 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
118 EXPECT_FALSE(ret);
119 }
120
121 HWTEST_F(UsbDdkServiceTest, RemoveDriverInfo002, TestSize.Level1)
122 {
123 V1_1::DriverAbilityInfo driverInfo;
124 driverInfo.driverUid = "driverUid-12345";
125 bool ret = V1_1::UsbDriverManager::GetInstance().UpdateDriverInfo(driverInfo);
126 EXPECT_TRUE(ret);
127 uint32_t tokenId = 12345;
128 V1_1::DriverAbilityInfo queriedDriverInfo;
129 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
130 EXPECT_TRUE(ret);
131
132 std::string driverUid = "driverUid-11111";
133 ret = V1_1::UsbDriverManager::GetInstance().RemoveDriverInfo(driverUid);
134 EXPECT_TRUE(ret);
135 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
136 EXPECT_TRUE(ret);
137
138 driverUid = "driverUid-12345";
139 ret = V1_1::UsbDriverManager::GetInstance().RemoveDriverInfo(driverUid);
140 EXPECT_TRUE(ret);
141 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
142 EXPECT_FALSE(ret);
143 }
144
145 HWTEST_F(UsbDdkServiceTest, QueryDriverInfo001, TestSize.Level1)
146 {
147 V1_1::DriverAbilityInfo queriedDriverInfo;
148 bool ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(11111, queriedDriverInfo);
149 EXPECT_FALSE(ret);
150
151 V1_1::DriverAbilityInfo updateDriverInfo;
152 updateDriverInfo.driverUid = "driverUid-11111";
153 ret = V1_1::UsbDriverManager::GetInstance().UpdateDriverInfo(updateDriverInfo);
154 EXPECT_TRUE(ret);
155
156 uint32_t tokenId = 22222;
157 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
158 EXPECT_FALSE(ret);
159
160 tokenId = 11111;
161 ret = V1_1::UsbDriverManager::GetInstance().QueryDriverInfo(tokenId, queriedDriverInfo);
162 EXPECT_TRUE(ret);
163 EXPECT_EQ(updateDriverInfo.driverUid, queriedDriverInfo.driverUid);
164 }
165 }