1 /*
2  * Copyright (c) 2023 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 <gtest/gtest.h>
17 
18 #include "event_report.h"
19 #include "istorage_manager.h"
20 #include "storage_manager_proxy.h"
21 #include "system_ability_definition.h"
22 #include "tokenid_kit.h"
23 #define private public
24 #include "uri_permission_manager_stub_impl.h"
25 #undef private
26 
27 using namespace testing;
28 using namespace testing::ext;
29 
30 namespace OHOS {
31 namespace AAFwk {
32 class UriPermissionTest : public testing::Test {
33 public:
34     static void SetUpTestCase();
35     static void TearDownTestCase();
36     void SetUp() override;
37     void TearDown() override;
38 };
39 
SetUpTestCase()40 void UriPermissionTest::SetUpTestCase() {}
41 
TearDownTestCase()42 void UriPermissionTest::TearDownTestCase() {}
43 
SetUp()44 void UriPermissionTest::SetUp() {}
45 
TearDown()46 void UriPermissionTest::TearDown() {}
47 
48 /*
49  * Feature: URIPermissionManagerService
50  * Function: GrantUriPermission
51  * SubFunction: NA
52  * FunctionPoints: URIPermissionManagerService GrantUriPermission
53  */
54 HWTEST_F(UriPermissionTest, Upms_GrantUriPermission_001, TestSize.Level1)
55 {
56     auto upms = std::make_shared<UriPermissionManagerStubImpl>();
57     EXPECT_NE(upms, nullptr);
58     auto uriStr = "file://com.example.test/data/storage/el2/base/haps/entry/files/test_A.txt";
59     Uri uri(uriStr);
60     unsigned int flag = 1;
61     std::string targetBundleName = "name2";
62     upms->GrantUriPermission(uri, flag, targetBundleName);
63 }
64 
65 /*
66  * Feature: URIPermissionManagerService
67  * Function: ConnectManager
68  * SubFunction: NA
69  * FunctionPoints: URIPermissionManagerService ConnectManager
70  */
71 HWTEST_F(UriPermissionTest, Upms_ConnectManager_001, TestSize.Level1)
72 {
73     auto upms = std::make_unique<UriPermissionManagerStubImpl>();
74     ASSERT_NE(upms, nullptr);
75     sptr<StorageManager::IStorageManager> storageManager = nullptr;
76     upms->ConnectManager(storageManager, STORAGE_MANAGER_MANAGER_ID);
77     ASSERT_NE(storageManager, nullptr);
78 }
79 
80 /*
81  * Feature: URIPermissionManagerService
82  * Function: RevokeUriPermission
83  * SubFunction: NA
84  * FunctionPoints: URIPermissionManagerService RevokeUriPermission
85  */
86 HWTEST_F(UriPermissionTest, Upms_RevokeUriPermission_001, TestSize.Level1)
87 {
88     auto upms = std::make_shared<UriPermissionManagerStubImpl>();
89     EXPECT_NE(upms, nullptr);
90     unsigned int tmpFlag = 1;
91     uint32_t fromTokenId = 2;
92     uint32_t targetTokenId = 3;
93     int autoremove = 1;
94     GrantInfo info = { tmpFlag, fromTokenId, targetTokenId, autoremove };
95     std::list<GrantInfo> infoList = { info };
96     auto uriStr = "file://com.example.test/data/storage/el2/base/haps/entry/files/test_A.txt";
97     upms->uriMap_.emplace(uriStr, infoList);
98     upms->RevokeUriPermission(targetTokenId);
99 }
100 
101 /*
102  * Feature: URIPermissionManagerService
103  * Function: OnRemoteDied
104  * SubFunction: NA
105  * FunctionPoints: URIPermissionManagerService OnRemoteDied
106  */
107 HWTEST_F(UriPermissionTest, Upms_OnRemoteDied_001, TestSize.Level1)
108 {
__anoneb437e500102(const wptr<IRemoteObject>& remote) 109     const auto& onClearProxyCallback = [](const wptr<IRemoteObject>& remote) {};
110     sptr<UriPermissionManagerStubImpl::ProxyDeathRecipient> object =
111         new UriPermissionManagerStubImpl::ProxyDeathRecipient(onClearProxyCallback);
112     EXPECT_NE(object, nullptr);
113     object->OnRemoteDied(nullptr);
114 }
115 
116 /*
117  * Feature: URIPermissionManagerService
118  * Function: OnRemoteDied
119  * SubFunction: NA
120  * FunctionPoints: URIPermissionManagerService OnRemoteDied
121  */
122 HWTEST_F(UriPermissionTest, Upms_OnRemoteDied_002, TestSize.Level1)
123 {
124     sptr<UriPermissionManagerStubImpl::ProxyDeathRecipient> object =
125         new UriPermissionManagerStubImpl::ProxyDeathRecipient(nullptr);
126     EXPECT_NE(object, nullptr);
127     object->OnRemoteDied(nullptr);
128 }
129 }  // namespace AAFwk
130 }  // namespace OHOS
131